41 lines
1.5 KiB
Python
41 lines
1.5 KiB
Python
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.blocks import ImageUrlBlock, BasicKnowledgeBlock
|
|
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_url', ImageUrlBlock(icon='image')),
|
|
('image_block', ImageChooserBlock(icon='image'))
|
|
], null=True, blank=True)
|
|
|
|
def __str__(self):
|
|
return 'RoomEntry {}-{}-{}'.format(self.id, self.title, self.author)
|