graphql-python/graphene-django
1
1
Fork 0
mirror of https://github.com/graphql-python/graphene-django.git synced 2024-11-22 17:47:12 +03:00
Code Issues Packages Projects Releases Wiki Activity

Update DjangoModelFormMutation to honor input_field_name

Browse Source 
Fixes: #1037
This commit is contained in:
Eugene Prikazchikov 2021-05-18 12:45:50 +03:00
parent 623d0f219e
commit c73ce86ef6
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
graphene_django/forms/mutation.py
View File

@ -111,9 +111,12 @@ class DjangoFormMutation(BaseDjangoFormMutation):
class DjangoModelDjangoFormMutationOptions(DjangoFormMutationOptions): class DjangoModelDjangoFormMutationOptions(DjangoFormMutationOptions):
model = None model = None
return_field_name = None return_field_name = None
input_field_name = None
class DjangoModelFormMutation(BaseDjangoFormMutation): class DjangoModelFormMutation(BaseDjangoFormMutation):
_DEFAULT_INPUT_FIELD_NAME = "input"
class Meta: class Meta:
abstract = True abstract = True
@ -127,6 +130,7 @@ class DjangoModelFormMutation(BaseDjangoFormMutation):
return_field_name=None, return_field_name=None,
only_fields=(), only_fields=(),
exclude_fields=(), exclude_fields=(),
input_field_name=_DEFAULT_INPUT_FIELD_NAME,
**options **options
): ):
@ -166,6 +170,17 @@ class DjangoModelFormMutation(BaseDjangoFormMutation):
super(DjangoModelFormMutation, cls).__init_subclass_with_meta__( super(DjangoModelFormMutation, cls).__init_subclass_with_meta__(
_meta=_meta, input_fields=input_fields, **options _meta=_meta, input_fields=input_fields, **options
) )
cls.input_field_name = input_field_name
if cls.input_field_name != cls._DEFAULT_INPUT_FIELD_NAME:
cls._meta.arguments[cls.input_field_name] = cls._meta.arguments.pop(
cls._DEFAULT_INPUT_FIELD_NAME
)
@classmethod
def mutate(cls, root, info, **kwargs):
if cls.input_field_name != cls._DEFAULT_INPUT_FIELD_NAME:
kwargs[cls._DEFAULT_INPUT_FIELD_NAME] = kwargs.pop(cls.input_field_name)
return super(DjangoModelFormMutation, cls).mutate(root, info, **kwargs)
@classmethod @classmethod
def mutate_and_get_payload(cls, root, info, **input): def mutate_and_get_payload(cls, root, info, **input):