graphql-python/graphene-django
1
1
Fork 0
mirror of https://github.com/graphql-python/graphene-django.git synced 2025-07-10 16:22:20 +03:00
Code Issues Packages Projects Releases Wiki Activity

Refactor validation function to reduce complexity

Browse Source
This commit is contained in:
lilac-supernova-2 2023-08-26 03:30:00 -04:00 committed by GitHub
parent 34db851230
commit 98fd498959
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 34 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
graphene_django/types.py
View File

@ -62,10 +62,7 @@ def construct_fields(
return fields
def validate_fields(type_, model, fields, only_fields, exclude_fields):
# Validate the given fields against the model's fields and custom fields
all_field_names = set(fields.keys())
only_fields = only_fields if only_fields is not ALL_FIELDS else ()
def validate_only_fields(only_fields, all_field_names, model, type_):
for name in only_fields or ():
if name in all_field_names:
continue
@ -83,8 +80,8 @@ def validate_fields(type_, model, fields, only_fields, exclude_fields):
type_=type_,
)
)
continue
else:
warnings.warn(
(
'Field name "{field_name}" doesn\'t exist on Django model "{app_label}.{object_name}". '
@ -97,6 +94,7 @@ def validate_fields(type_, model, fields, only_fields, exclude_fields):
)
)
def validate_exclude_fields(exclude_fields, all_field_names, model, type_):
# Validate exclude fields
for name in exclude_fields or ():
if name in all_field_names:
@ -107,7 +105,8 @@ def validate_fields(type_, model, fields, only_fields, exclude_fields):
'Either remove the custom field or remove the field from the "exclude" list.'
).format(field_name=name, type_=type_)
)
else:
continue
if not hasattr(model, name):
warnings.warn(
(
@ -121,6 +120,14 @@ def validate_fields(type_, model, fields, only_fields, exclude_fields):
)
)
def validate_fields(type_, model, fields, only_fields, exclude_fields):
# Validate the given fields against the model's fields and custom fields
all_field_names = set(fields.keys())
only_fields = only_fields if only_fields is not ALL_FIELDS else ()
validate_only_fields(only_fields, all_field_names, model, type_)
validate_exclude_fields(exclude_fields, all_field_names, model, type_)
class DjangoObjectTypeOptions(ObjectTypeOptions):
model = None # type: Type[Model]