48 lines
1.9 KiB
Python
48 lines
1.9 KiB
Python
from django.db import models
|
|
from django.contrib.contenttypes.models import ContentType
|
|
from django.contrib.contenttypes.fields import GenericForeignKey, GenericRelation
|
|
|
|
# Create your models here.
|
|
class Ingredient(models.Model):
|
|
limit = models.Q(app_label="api", model="recipe") | models.Q(app_label="api", model="list")
|
|
content_type = models.ForeignKey(ContentType, limit_choices_to=limit, on_delete=models.CASCADE)
|
|
object_id = models.PositiveBigIntegerField()
|
|
content_object = GenericForeignKey()
|
|
|
|
|
|
name = models.CharField(max_length=50)
|
|
in_stock = models.BooleanField(default=False)
|
|
|
|
class HomegroupInvite(models.Model):
|
|
homegroup = models.ForeignKey("api.Homegroup", on_delete=models.CASCADE, related_name="invites", blank=True)
|
|
user = models.ForeignKey("users.User", on_delete=models.CASCADE, related_name="invites", blank=True)
|
|
|
|
class Meta:
|
|
unique_together = ("homegroup", "user")
|
|
|
|
class Homegroup(models.Model):
|
|
# Foreign Key Recipe -> Homegroup [as recipes]
|
|
# Foreign Key User -> Homegroup [as users]
|
|
name = models.CharField(max_length=50)
|
|
invited_users = models.ManyToManyField("users.User", related_name="homegroup_invites", through=HomegroupInvite)
|
|
|
|
def __repr__(self):
|
|
return f"{self.id}: {self.name}"
|
|
|
|
def __str__(self):
|
|
return f"{self.id}: {self.name}"
|
|
|
|
class List(models.Model):
|
|
# Foreign Key Recipe -> List [as recipes]
|
|
extra_ingredients = GenericRelation(Ingredient, related_query_name="extra_ingredients")
|
|
homegroup = models.OneToOneField(Homegroup, on_delete=models.CASCADE, primary_key=True)
|
|
|
|
|
|
class Recipe(models.Model):
|
|
name = models.CharField(max_length=50)
|
|
homegroup = models.ForeignKey(Homegroup, related_name="recipes", on_delete=models.CASCADE)
|
|
list = models.ForeignKey(List, related_name="recipes", on_delete=models.SET_NULL, blank=True, null=True)
|
|
ingredients = GenericRelation(Ingredient, related_query_name="ingredients")
|
|
|
|
|