import yangilandi

core/apps/api/models/category.py

@@ -21,6 +21,10 @@ class FilialModel(AbstractBaseModel):
 
 
 class CategoryModel(AbstractBaseModel):
+    class CategoryType(models.TextChoices):
+        RESTAURANT = "RESTAURANT", _("Restaurant")
+        BAR = "BAR", _("Bar")
+
     filial = models.ForeignKey(
         FilialModel,
         verbose_name=_("filial"),
@@ -30,10 +34,16 @@ class CategoryModel(AbstractBaseModel):
         blank=True,
     )
     name = models.CharField(verbose_name=_("name"), max_length=255)
+    type = models.CharField(
+        verbose_name=_("type"),
+        max_length=20,
+        choices=CategoryType.choices,
+        default=CategoryType.RESTAURANT,
+    )
     image = models.ImageField(verbose_name=_("image"), upload_to="categories/", null=True, blank=True)
 
     def __str__(self):
-        return self.name
+        return f"{self.name} ({self.type})"
 
     @classmethod
     def _baker(cls):