mirror of
https://github.com/daniviga/django-ram.git
synced 2025-08-04 13:17:50 +02:00
97 lines
2.5 KiB
Python
97 lines
2.5 KiB
Python
from django.db import models
|
|
from django.conf import settings
|
|
from django.dispatch.dispatcher import receiver
|
|
from django_countries.fields import CountryField
|
|
|
|
from dcc.utils import get_image_preview, slugify
|
|
|
|
|
|
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"
|
|
|
|
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"
|
|
|
|
|
|
class Tag(models.Model):
|
|
name = models.CharField(max_length=128, unique=True)
|
|
slug = models.CharField(max_length=128, unique=True)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
|
|
@receiver(models.signals.pre_save, sender=Tag)
|
|
def tag_pre_save(sender, instance, **kwargs):
|
|
instance.slug = slugify(instance.name)
|
|
|
|
|
|
class RollingStockType(models.Model):
|
|
type = models.CharField(max_length=64)
|
|
category = models.CharField(
|
|
max_length=64, choices=settings.ROLLING_STOCK_TYPES)
|
|
|
|
class Meta(object):
|
|
unique_together = ('category', 'type')
|
|
|
|
def __str__(self):
|
|
return "{0}".format(self.type)
|