mirror of
https://github.com/graphql-python/graphene.git
synced 2025-10-22 19:54:17 +03:00
73 lines
2.4 KiB
Python
73 lines
2.4 KiB
Python
import re
|
|
from collections import OrderedDict
|
|
|
|
from promise import Promise, is_thenable
|
|
|
|
from ..types import Field, InputObjectType, String
|
|
from ..types.mutation import Mutation
|
|
|
|
|
|
class ClientIDMutation(Mutation):
|
|
|
|
class Meta:
|
|
abstract = True
|
|
|
|
@classmethod
|
|
def __init_subclass_with_meta__(cls, output=None, input_fields=None,
|
|
arguments=None, name=None, **options):
|
|
input_class = getattr(cls, 'Input', None)
|
|
base_name = re.sub('Payload$', '', name or cls.__name__)
|
|
|
|
assert not output, "Can't specify any output"
|
|
assert not arguments, "Can't specify any arguments"
|
|
|
|
bases = (InputObjectType, )
|
|
if input_class:
|
|
bases += (input_class, )
|
|
|
|
if not input_fields:
|
|
input_fields = {}
|
|
|
|
cls.Input = type(
|
|
'{}Input'.format(base_name),
|
|
bases,
|
|
OrderedDict(input_fields, client_mutation_id=String(
|
|
name='clientMutationId'))
|
|
)
|
|
|
|
arguments = OrderedDict(
|
|
input=cls.Input(required=True)
|
|
# 'client_mutation_id': String(name='clientMutationId')
|
|
)
|
|
mutate_and_get_payload = getattr(cls, 'mutate_and_get_payload', None)
|
|
if cls.mutate and cls.mutate.__func__ == ClientIDMutation.mutate.__func__:
|
|
assert mutate_and_get_payload, (
|
|
"{name}.mutate_and_get_payload method is required"
|
|
" in a ClientIDMutation.").format(name=name or cls.__name__)
|
|
|
|
if not name:
|
|
name = '{}Payload'.format(base_name)
|
|
|
|
super(ClientIDMutation, cls).__init_subclass_with_meta__(
|
|
output=None, arguments=arguments, name=name, **options)
|
|
cls._meta.fields['client_mutation_id'] = (
|
|
Field(String, name='clientMutationId')
|
|
)
|
|
|
|
@classmethod
|
|
def mutate(cls, root, info, input):
|
|
def on_resolve(payload):
|
|
try:
|
|
payload.client_mutation_id = input.get('client_mutation_id')
|
|
except:
|
|
raise Exception(
|
|
('Cannot set client_mutation_id in the payload object {}'
|
|
).format(repr(payload)))
|
|
return payload
|
|
|
|
result = cls.mutate_and_get_payload(root, info, **input)
|
|
if is_thenable(result):
|
|
return Promise.resolve(result).then(on_resolve)
|
|
|
|
return on_resolve(result)
|