skillbox/server/book/schema/mutations.py

63 lines
2.2 KiB
Python

import graphene
from django.core.exceptions import ValidationError
from api.utils import get_object, get_errors
from book.models import ContentBlock
from book.schema.queries import ContentBlockNode
class MutateContentBlock(graphene.relay.ClientIDMutation):
class Input:
id = graphene.ID()
type = graphene.String()
title = graphene.String()
contents = graphene.String()
errors = graphene.List(graphene.String)
content_block = graphene.Field(ContentBlockNode)
# updated_title = graphene.String()
# updated_type = graphene.String()
@classmethod
def mutate_and_get_payload(cls, *args, **kwargs):
try:
id_param = kwargs['id']
type_param = kwargs['type']
title = kwargs['title']
contents_data = kwargs['contents']
new_content_block = ContentBlock(type=type_param, title=title, contents=contents_data)
content_block = get_object(ContentBlock, id_param)
content_block.add_sibling(instance=new_content_block, pos='right')
# ContentBlock.objects.get()
# cb.add_sibling()
new_content_block.save()
# image_instance = get_object(Image, kwargs['id'])
# if image_instance:
# image_data = kwargs.get('image')
# updated_image = update_create_instance(image_instance, image_data, exception=['id', 'tags'])
# tag_slugs = image_data.get('tag_slugs', '')
# if tag_slugs is not None:
# tag_slugs = [t.strip() for t in tag_slugs.split(',') if t.strip()]
# tags = list(Tag.objects.filter(slug__in=tag_slugs))
# tag_slugs = [t for t in tag_slugs if t not in [e.slug for e in tags]]
# updated_image.tags.set(*(tags + tag_slugs))
return cls(content_block=new_content_block)
except ValidationError as e:
errors = get_errors(e)
except Exception as e:
errors = ['Error: {}'.format(e)]
return cls(content_block=None, errors=errors)
class BookMutations(object):
mutate_content_block = MutateContentBlock.Field()