diff --git a/ofu_app/apps/food/models.py b/ofu_app/apps/food/models.py index 89b7841..1259c4a 100644 --- a/ofu_app/apps/food/models.py +++ b/ofu_app/apps/food/models.py @@ -31,6 +31,14 @@ class SingleFood(models.Model): fourth_star = models.SmallIntegerField(default=0) fifth_star = models.SmallIntegerField(default=0) + def __str__(self): + return "%s Rating: %f" % (self.name, self.rating) + + +class Allergene(models.Model): + id = models.AutoField(primary_key=True) + name = models.CharField(unique=True, max_length=MAX_LENGTH) + def __str__(self): return self.name