50 lines
1.8 KiB
Python
50 lines
1.8 KiB
Python
|
from django.db import models
|
||
|
|
from django.core.validators import MaxValueValidator, MinValueValidator
|
||
|
|
from django.utils import timezone
|
||
|
|
from django.contrib.auth import get_user_model
|
||
|
|
|
||
|
|
from mood.models import Mood
|
||
|
|
from common.helpers import get_upload_path
|
||
|
|
|
||
|
|
from colorfield.fields import ColorField
|
||
|
|
|
||
|
|
class Theme(models.Model):
|
||
|
|
user = models.ForeignKey(get_user_model(), models.CASCADE)
|
||
|
|
name = models.CharField(max_length=64)
|
||
|
|
icon = models.CharField(default="fas fa-bed", max_length=64)
|
||
|
|
color = ColorField(default="#000000")
|
||
|
|
|
||
|
|
class Dream(models.Model):
|
||
|
|
class DreamTypes(models.IntegerChoices):
|
||
|
|
NIGHT = 0, 'Night (main) sleep'
|
||
|
|
DAY = 1, 'Daydream'
|
||
|
|
NAP = 2, 'Napping'
|
||
|
|
|
||
|
|
user = models.ForeignKey(get_user_model(), models.CASCADE)
|
||
|
|
title = models.CharField(max_length=64)
|
||
|
|
content = models.TextField()
|
||
|
|
type = models.IntegerField(choices=DreamTypes.choices)
|
||
|
|
mood = models.ForeignKey(Mood, models.SET_NULL, null=True)
|
||
|
|
lucid = models.BooleanField(default=False)
|
||
|
|
wet = models.BooleanField(default=False)
|
||
|
|
|
||
|
|
class DreamTheme(models.Model):
|
||
|
|
dream = models.ForeignKey(Dream, models.CASCADE)
|
||
|
|
theme = models.ForeignKey(Theme, models.CASCADE)
|
||
|
|
|
||
|
|
class DreamMedia(models.Model):
|
||
|
|
dream = models.ForeignKey(Dream, models.CASCADE)
|
||
|
|
media = models.FileField(get_upload_path)
|
||
|
|
|
||
|
|
class ThemeRating(models.Model):
|
||
|
|
theme = models.ForeignKey(Theme, models.CASCADE)
|
||
|
|
name = models.CharField(max_length=64)
|
||
|
|
icon = models.CharField(default="fas fa-star", max_length=64)
|
||
|
|
color = ColorField(default="#000000")
|
||
|
|
value = models.IntegerField(validators=[MinValueValidator(0), MaxValueValidator(255)])
|
||
|
|
|
||
|
|
class DreamThemeRating(models.Model):
|
||
|
|
dream = models.ForeignKey(Dream, models.CASCADE)
|
||
|
|
theme_rating = models.ForeignKey(ThemeRating, models.SET_NULL, null=True)
|
||
|
|
comment = models.TextField(null=True, blank=True)
|