from django.contrib.auth import get_user_model
from django.db import models
from django_extensions.db.models import TitleDescriptionModel, TitleSlugDescriptionModel
from wagtail.core.fields import StreamField
from wagtail.images.blocks import ImageChooserBlock

from book.models import ContentBlock, TextBlock
from user.models import UserGroup


class Room(TitleSlugDescriptionModel):
    class Meta:
        verbose_name = 'Raum'
        verbose_name_plural = 'Räume'

    user_group = models.ForeignKey(UserGroup, blank=False, null=False, on_delete=models.CASCADE)
    appearance = models.CharField(blank=True, null=False, max_length=255)

    def __str__(self):
        return 'Room {}-{}-{}'.format(self.id, self.title, self.user_group)


class RoomEntry(TitleSlugDescriptionModel):
    class Meta:
        verbose_name = 'Raumeintrag'
        verbose_name_plural = 'Raumeinträge'

    room = models.ForeignKey(Room, blank=False, null=False, on_delete=models.CASCADE, related_name='room_entries')
    author = models.ForeignKey(get_user_model(), null=True, on_delete=models.CASCADE)
    subtitle = models.CharField(blank=True, null=False, max_length=255)

    contents = StreamField([
        ('text_block', TextBlock(icon='doc-full')),
        ('image_block', ImageChooserBlock(icon='image'))
    ], null=True, blank=True)

    def __str__(self):
        return 'RoomEntry {}-{}-{}'.format(self.id, self.title, self.author)