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

Add custom function setting

Browse Source
This commit is contained in:
Jonathan Kim 2020-01-22 10:31:27 +00:00
parent f9252e3c4e
commit ecdbae3c8a
3 changed files with 15 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
graphene_django/converter.py
View File

@ -70,7 +70,11 @@ def convert_choices_to_named_enum_with_descriptions(name, choices):
def generate_enum_name(django_model_meta, field): def generate_enum_name(django_model_meta, field):
if graphene_settings.DJANGO_CHOICE_FIELD_ENUM_V3_NAMING is True: if graphene_settings.DJANGO_CHOICE_FIELD_ENUM_CUSTOM_NAME and callable(
graphene_settings.DJANGO_CHOICE_FIELD_ENUM_CUSTOM_NAME
):
name = graphene_settings.DJANGO_CHOICE_FIELD_ENUM_CUSTOM_NAME(field)
elif graphene_settings.DJANGO_CHOICE_FIELD_ENUM_V3_NAMING is True:
name = "DjangoModel{app_label}{object_name}{field_name}Choices".format( name = "DjangoModel{app_label}{object_name}{field_name}Choices".format(
app_label=to_camel_case(django_model_meta.app_label.title()), app_label=to_camel_case(django_model_meta.app_label.title()),
object_name=django_model_meta.object_name, object_name=django_model_meta.object_name,

1
graphene_django/settings.py
View File

@ -38,6 +38,7 @@ DEFAULTS = {
"CAMELCASE_ERRORS": False, "CAMELCASE_ERRORS": False,
# Set to True to enable v3 naming convention for choice field Enum's # Set to True to enable v3 naming convention for choice field Enum's
"DJANGO_CHOICE_FIELD_ENUM_V3_NAMING": False, "DJANGO_CHOICE_FIELD_ENUM_V3_NAMING": False,
"DJANGO_CHOICE_FIELD_ENUM_CUSTOM_NAME": None,
} }
if settings.DEBUG: if settings.DEBUG:

18
graphene_django/tests/test_types.py
View File

@ -514,14 +514,14 @@ class TestDjangoObjectType:
query: Query query: Query
} }
enum DjangoModelTestsPetModelKindChoices { enum CustomEnumKind {
CAT CAT
DOG DOG
} }
type PetModelKind { type PetModelKind {
id: ID! id: ID!
kind: DjangoModelTestsPetModelKindChoices! kind: CustomEnumKind!
} }
type Query { type Query {
@ -531,15 +531,13 @@ class TestDjangoObjectType:
) )
graphene_settings.DJANGO_CHOICE_FIELD_ENUM_V3_NAMING = False graphene_settings.DJANGO_CHOICE_FIELD_ENUM_V3_NAMING = False
def test_django_objecttype_choices_override_enum(self, PetModel): def test_django_objecttype_choices_custom_enum_name(self, PetModel):
def convert_choice(model, field_name, **kwargs): def custom_name(field):
return convert_choice_field_to_enum( return f"CustomEnum{field.name.title()}"
model._meta.get_field(field_name), **kwargs
) graphene_settings.DJANGO_CHOICE_FIELD_ENUM_CUSTOM_NAME = custom_name
class PetModelKind(DjangoObjectType): class PetModelKind(DjangoObjectType):
kind = Field(convert_choice(PetModel, "kind", name="CustomEnumName"))
class Meta: class Meta:
model = PetModel model = PetModel
fields = ["id", "kind"] fields = ["id", "kind"]
@ -570,3 +568,5 @@ class TestDjangoObjectType:
} }
""" """
) )
graphene_settings.DJANGO_CHOICE_FIELD_ENUM_CUSTOM_NAME = None