1 files changed, 39 insertions, 0 deletions
diff --git a/app/taxonomy/models.py b/app/taxonomy/models.py
new file mode 100644
index 0000000..38f1bf7
--- /dev/null
+++ b/app/taxonomy/models.py
@@ -0,0 +1,39 @@
+from django.contrib.gis.db import models
+from django.urls import reverse
+from django.utils.translation import ugettext_lazy as _
+from django.utils.functional import cached_property
+
+from taggit.models import TagBase, GenericTaggedItemBase
+
+
+class LuxTag(TagBase):
+    ''' override the default taggit model to add some color '''
+    color_rgb = models.CharField(max_length=20, blank=True)
+
+    class Meta:
+        verbose_name = _("Tag")
+        verbose_name_plural = _("Tags")
+
+    @cached_property
+    def get_absolute_url(self):
+        return reverse("taxonomy:tags", kwargs={"slug": self.slug})
+
+
+class TaggedItems(GenericTaggedItemBase):
+    ''' necessary with custom tag model, lets you still use TaggableManager'''
+    tag = models.ForeignKey(LuxTag, related_name="%(app_label)s_%(class)s_items", on_delete=models.CASCADE)
+
+
+class Category(models.Model):
+    """ Generic model for Categories """
+    name = models.CharField(max_length=250)
+    color_rgb = models.CharField(max_length=20, blank=True)
+    slug = models.SlugField(blank=True)
+    date_created = models.DateTimeField(blank=True, auto_now_add=True, editable=False)
+    date_updated = models.DateTimeField(blank=True, auto_now=True, editable=False)
+
+    def __str__(self):
+        return self.name
+
+    def get_absolute_url(self):
+        return reverse("taxonomy:categories", kwargs={"slug": self.slug})