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

Update types.py. Add permission_raise_exception meta (#33)

Browse Source 
* Update types.py

* Update types.py
This commit is contained in:
arielRevo 2024-03-05 17:33:16 +01:00 committed by GitHub
parent 396c47f98b
commit 7e073e11e4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
graphene_django/types.py
View File

@ -7,7 +7,7 @@ from django.db.models import Model
from django.utils.functional import SimpleLazyObject from django.utils.functional import SimpleLazyObject
import graphene import graphene
from graphene import Field, NonNull from graphene import Field, NonNull, Boolean
from graphene.relay import Connection, Node from graphene.relay import Connection, Node
from graphene.types.objecttype import ObjectType, ObjectTypeOptions from graphene.types.objecttype import ObjectType, ObjectTypeOptions
from graphene.types.utils import yank_fields_from_attrs from graphene.types.utils import yank_fields_from_attrs
@ -30,7 +30,7 @@ ALL_FIELDS = "__all__"
def construct_fields( def construct_fields(
model, registry, only_fields, exclude_fields, convert_choices_to_enum model, registry, only_fields, exclude_fields, convert_choices_to_enum, permission_raise_exception=True
): ):
_model_fields = get_model_fields(model) _model_fields = get_model_fields(model)
@ -55,9 +55,16 @@ def construct_fields(
else: else:
_convert_choices_to_enum = False _convert_choices_to_enum = False
if name != 'id' and not permission_raise_exception:
field.null = True
converted = convert_django_field_with_choices( converted = convert_django_field_with_choices(
field, registry, convert_choices_to_enum=_convert_choices_to_enum field, registry, convert_choices_to_enum=_convert_choices_to_enum
) )
if isinstance(converted, NonNull) and getattr(converted, '_of_type', None) == Boolean and not permission_raise_exception:
converted = Boolean(description=field.help_text, required=False)
fields[name] = converted fields[name] = converted
return fields return fields
@ -184,6 +191,7 @@ class DjangoObjectType(ObjectType):
field_to_permission=None, field_to_permission=None,
permission_to_field=None, permission_to_field=None,
permission_to_all_fields=None, permission_to_all_fields=None,
permission_raise_exception=True,
_meta=None, _meta=None,
**options **options
): ):
@ -251,7 +259,7 @@ class DjangoObjectType(ObjectType):
) )
django_fields = yank_fields_from_attrs( django_fields = yank_fields_from_attrs(
construct_fields(model, registry, fields, exclude, convert_choices_to_enum), construct_fields(model, registry, fields, exclude, convert_choices_to_enum, permission_raise_exception),
_as=Field, _as=Field,
) )
@ -295,6 +303,7 @@ class DjangoObjectType(ObjectType):
permission_to_field, permission_to_field,
permission_to_all_fields, permission_to_all_fields,
permission_classes, permission_classes,
permission_raise_exception,
) )
# Validate fields # Validate fields
@ -317,6 +326,7 @@ class DjangoObjectType(ObjectType):
permission_to_field, permission_to_field,
permission_to_all_fields, permission_to_all_fields,
permission_classes, permission_classes,
permission_raise_exception,
): ):
"""Combines permissions from meta""" """Combines permissions from meta"""
permissions = field_to_permission if field_to_permission else {} permissions = field_to_permission if field_to_permission else {}
@ -343,7 +353,7 @@ class DjangoObjectType(ObjectType):
set(permissions.get(name, ()) + permission_to_all_fields) set(permissions.get(name, ()) + permission_to_all_fields)
) )
if name in permissions: if name in permissions and permission_raise_exception:
fields_raise_exception[name] = hasattr(field, "_type") and isinstance( fields_raise_exception[name] = hasattr(field, "_type") and isinstance(
field._type, NonNull field._type, NonNull
) )