38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
from django.db import models
|
|
from django.contrib.auth.models import User
|
|
|
|
|
|
class Activity(models.Model):
|
|
class Category(models.TextChoices):
|
|
SOCIAL = 'Social'
|
|
HEALTH = 'Health'
|
|
HOBBIES = 'Hobbies'
|
|
OBLIGATION = 'Obligation'
|
|
|
|
name = models.CharField(max_length=255)
|
|
icon = models.CharField(max_length=50)
|
|
category = models.CharField(max_length=50, choices=Category.choices)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
class Mission(models.Model):
|
|
title = models.CharField(max_length=255)
|
|
description = models.TextField()
|
|
trigger_mood_score = models.IntegerField()
|
|
is_active = models.BooleanField(default=True)
|
|
|
|
def __str__(self):
|
|
return self.title
|
|
|
|
class MoodEntry(models.Model):
|
|
user = models.ForeignKey(User, on_delete=models.CASCADE)
|
|
date_time = models.DateTimeField(auto_now_add=True)
|
|
mood_score = models.IntegerField()
|
|
note = models.TextField(blank=True, null=True)
|
|
color_code = models.CharField(max_length=7)
|
|
activities = models.ManyToManyField(Activity)
|
|
mission = models.ForeignKey(Mission, on_delete=models.SET_NULL, null=True, blank=True)
|
|
|
|
def __str__(self):
|
|
return f'{self.user.username} - {self.date_time.strftime("%Y-%m-%d")}' |