Extend metadata

dcc/metadata/models.py

@@ -1,17 +1,31 @@
 from django.db import models
 from django_countries.fields import CountryField
 
+from dcc.utils import get_image_preview
+
 
 class Manufacturer(models.Model):
     name = models.CharField(max_length=128, unique=True)
+    logo = models.ImageField(
+        upload_to='images/',
+        null=True,
+        blank=True)
 
     def __str__(self):
         return self.name
 
+    def logo_thumbnail(self):
+        return get_image_preview(self.logo.url)
+    logo_thumbnail.short_description = "Preview"
+
 
 class Company(models.Model):
     name = models.CharField(max_length=128, unique=True)
     country = CountryField()
+    logo = models.ImageField(
+        upload_to='images/',
+        null=True,
+        blank=True)
 
     class Meta:
         verbose_name_plural = "Companies"
@@ -19,13 +33,37 @@ class Company(models.Model):
     def __str__(self):
         return self.name
 
+    def logo_thumbnail(self):
+        return get_image_preview(self.logo.url)
+    logo_thumbnail.short_description = "Preview"
+
 
 class Decoder(models.Model):
+    class Interface(models.IntegerChoices):
+        NEM651 = 1, "NEM651"
+        NEM652 = 2, "NEM652"
+        NEM658 = 3, "PluX"
+        NEM660 = 4, "21MTC"
+        NEM662 = 5, "Next18/Next18S"
+
     name = models.CharField(max_length=128, unique=True)
     manufacturer = models.ForeignKey(
         Manufacturer,
         on_delete=models.CASCADE)
     version = models.CharField(max_length=64, blank=True)
+    interface = models.PositiveSmallIntegerField(
+        choices=Interface.choices,
+        null=True,
+        blank=True
+    )
+    image = models.ImageField(
+        upload_to='images/',
+        null=True,
+        blank=True)
 
     def __str__(self):
         return "{0} - {1}".format(self.manufacturer, self.name)
+
+    def image_thumbnail(self):
+        return get_image_preview(self.image.url)
+    image_thumbnail.short_description = "Preview"