Update formatting

This commit is contained in:
Jonathan Kim 2020-06-10 17:49:13 +01:00
parent bf61399be3
commit cea771c493
2 changed files with 13 additions and 3 deletions

View File

@ -257,7 +257,11 @@ def convert_field_to_djangomodel(field, registry=None):
def convert_postgres_array_to_list(field, registry=None):
inner_type = convert_django_field(field.base_field)
if not isinstance(inner_type, (List, NonNull)):
inner_type = NonNull(type(inner_type)) if inner_type.kwargs['required'] else type(inner_type)
inner_type = (
NonNull(type(inner_type))
if inner_type.kwargs["required"]
else type(inner_type)
)
return List(inner_type, description=field.help_text, required=not field.null)
@ -271,5 +275,9 @@ def convert_postgres_field_to_string(field, registry=None):
def convert_postgres_range_to_string(field, registry=None):
inner_type = convert_django_field(field.base_field)
if not isinstance(inner_type, (List, NonNull)):
inner_type = NonNull(type(inner_type)) if inner_type.kwargs['required'] else type(inner_type)
inner_type = (
NonNull(type(inner_type))
if inner_type.kwargs["required"]
else type(inner_type)
)
return List(inner_type, description=field.help_text, required=not field.null)

View File

@ -333,7 +333,9 @@ def test_should_postgres_array_multiple_convert_list():
assert field.type.of_type.of_type.of_type.of_type == graphene.String
field = assert_conversion(
ArrayField, graphene.List, ArrayField(models.CharField(max_length=100, null=True))
ArrayField,
graphene.List,
ArrayField(models.CharField(max_length=100, null=True)),
)
assert isinstance(field.type, graphene.NonNull)
assert isinstance(field.type.of_type, graphene.List)