95 lines
2.6 KiB
Python
95 lines
2.6 KiB
Python
from django.contrib.auth import get_user_model
|
|
from django.db import models
|
|
from django_extensions.db.models import TitleSlugDescriptionModel
|
|
from wagtail.fields import StreamField
|
|
|
|
from books.blocks import (
|
|
DocumentBlock,
|
|
ImageUrlBlock,
|
|
LinkBlock,
|
|
SubtitleBlock,
|
|
VideoBlock,
|
|
)
|
|
from books.models import TextBlock
|
|
from core.mixins import GraphqlNodeMixin
|
|
from users.models import SchoolClass
|
|
|
|
|
|
class Room(TitleSlugDescriptionModel, GraphqlNodeMixin):
|
|
class Meta:
|
|
verbose_name = "Raum"
|
|
verbose_name_plural = "Räume"
|
|
|
|
school_class = models.ForeignKey(
|
|
SchoolClass,
|
|
blank=False,
|
|
null=False,
|
|
on_delete=models.CASCADE,
|
|
related_name="rooms",
|
|
)
|
|
appearance = models.CharField(blank=True, null=False, max_length=255)
|
|
user_created = models.BooleanField(blank=False, null=False, default=True)
|
|
restricted = models.BooleanField(default=False)
|
|
|
|
def __str__(self):
|
|
return f"{self.title} - {self.school_class}"
|
|
|
|
|
|
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)
|
|
|
|
contents = StreamField(
|
|
[
|
|
("text_block", TextBlock()),
|
|
("image_url_block", ImageUrlBlock()),
|
|
("link_block", LinkBlock()),
|
|
("document_block", DocumentBlock()),
|
|
("subtitle", SubtitleBlock()),
|
|
("video_block", VideoBlock()),
|
|
],
|
|
null=True,
|
|
blank=True,
|
|
use_json_field=True,
|
|
)
|
|
|
|
def __str__(self):
|
|
return f"RoomEntry {self.id}-{self.title}-{self.author}"
|
|
|
|
def can_user_see_entry(self, user):
|
|
return user.is_superuser or self.room.school_class.is_user_in_schoolclass(user)
|
|
|
|
|
|
class ModuleRoomSlug(TitleSlugDescriptionModel):
|
|
class Meta:
|
|
verbose_name = "Slug für Modul-Raum"
|
|
verbose_name_plural = "Slugs für Modul-Raum"
|
|
|
|
def __str__(self):
|
|
return f"ModuleRoomSlug {self.id}-{self.title}"
|
|
|
|
|
|
class Comment(models.Model):
|
|
text = models.TextField()
|
|
room_entry = models.ForeignKey(
|
|
RoomEntry, related_name="comments", on_delete=models.CASCADE
|
|
)
|
|
owner = models.ForeignKey(
|
|
get_user_model(), related_name="comments", on_delete=models.PROTECT
|
|
)
|
|
created = models.DateTimeField(auto_now_add=True)
|
|
|
|
def __str__(self):
|
|
return self.text
|