import graphene
from graphene import relay

from api.graphene_wagtail import GenericStreamFieldType

class ContentBlockType(graphene.Enum):
    NORMAL = 'normal'
    BASE_COMMUNICATION = 'base_communication'
    TASK = 'task'
    BASE_SOCIETY = 'base_society'
    BASE_INTERDISCIPLINARY = 'base_interdisciplinary'


class ContentBlockInterface(relay.Node):
    title = graphene.String()
    contents = GenericStreamFieldType()
    type = ContentBlockType()