100 lines
3.6 KiB
Python
100 lines
3.6 KiB
Python
import json
|
|
|
|
import graphene
|
|
from graphene import relay
|
|
from django.core.exceptions import ValidationError
|
|
from wagtail.core.fields import StreamField
|
|
|
|
from api.utils import get_object, get_errors
|
|
from book.models import ContentBlock, Chapter
|
|
from book.schema.inputs import ContentBlockInput
|
|
from book.schema.queries import ContentBlockNode
|
|
|
|
|
|
class MutateContentBlock(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 AddContentBlock(relay.ClientIDMutation):
|
|
class Input:
|
|
content_block = graphene.Argument(ContentBlockInput)
|
|
|
|
new_content_block = graphene.Field(ContentBlockNode)
|
|
|
|
@classmethod
|
|
def mutate_and_get_payload(cls, root, info, **args):
|
|
content_block_data = args.get('content_block')
|
|
title = content_block_data.get('title')
|
|
new_content_block = ContentBlock(title=title)
|
|
parent = Chapter.objects.get(pk=18).specific # atm just "1.1 Lehrbeginn" / todo: dynamic, refactor in above class
|
|
parent.add_child(instance=new_content_block)
|
|
revision = new_content_block.save_revision()
|
|
revision.publish()
|
|
new_content_block.save()
|
|
|
|
contents = content_block_data.get('contents')
|
|
for content in contents:
|
|
# todo: add all the content blocks
|
|
# todo: refactor this mess
|
|
if content['type'] == 'text_block':
|
|
new_content_block.contents = json.dumps([
|
|
{'type': 'text_block', 'value': {
|
|
'text': '<p>{}</p>'.format(content['text'])
|
|
}}
|
|
])
|
|
new_content_block.save()
|
|
return cls(new_content_block=new_content_block)
|
|
|
|
|
|
class BookMutations(object):
|
|
mutate_content_block = MutateContentBlock.Field()
|
|
add_content_block = AddContentBlock.Field()
|