73 lines
2 KiB
Python
73 lines
2 KiB
Python
from django.db import models
|
|
from datetime import datetime
|
|
from django.contrib.auth.models import AbstractUser
|
|
|
|
# Create your models here.
|
|
class User(AbstractUser):
|
|
@property
|
|
def display_name(self):
|
|
name = '%s %s' % (self.first_name, self.last_name)
|
|
if name:
|
|
return name
|
|
return self.username
|
|
|
|
|
|
class TipoRadio(models.Model):
|
|
class Meta:
|
|
verbose_name = 'tipo radio'
|
|
verbose_name_plural = 'tipi radio'
|
|
nome = models.CharField(max_length=64, unique=True)
|
|
descrizione = models.CharField(max_length=1000)
|
|
|
|
def __str__(self):
|
|
return self.nome
|
|
|
|
|
|
class RapportoRicezione(models.Model):
|
|
class Meta:
|
|
verbose_name = "rapporto ricezione"
|
|
verbose_name_plural = 'rapporti ricezione'
|
|
|
|
author = models.ForeignKey(User, related_name="rapporti", on_delete=models.PROTECT)
|
|
tipo_radio = models.ForeignKey(
|
|
TipoRadio, on_delete=models.PROTECT, related_name="rapporti"
|
|
)
|
|
|
|
created = models.DateField(default=datetime.now)
|
|
|
|
lat = models.FloatField()
|
|
lng = models.FloatField()
|
|
|
|
comprensibile = models.IntegerField() # TODO: >= 0 <= 5
|
|
stabilita = models.IntegerField() # TODO: >= 0 <= 5
|
|
|
|
def serialize(self):
|
|
d = {k: v for k, v in self.__dict__.items() if not k.startswith("_")}
|
|
d["colore"] = self.colore
|
|
d["radius"] = self.radius
|
|
d["explaination"] = self.explaination
|
|
return d
|
|
|
|
@property
|
|
def colore(self):
|
|
colors = ['red', 'yellow', 'orange', '#90c90d', 'green']
|
|
c = int(self.comprensibile)
|
|
return colors[c-1]
|
|
|
|
@property
|
|
def radius(self):
|
|
return int(self.stabilita) * 70
|
|
|
|
@property
|
|
def explaination(self):
|
|
return """
|
|
Ricevuto con %s
|
|
Aggiunto il %s da %s
|
|
""" % (
|
|
self.tipo_radio.nome,
|
|
self.created,
|
|
self.author.display_name,
|
|
)
|
|
|
|
def __str__(self):
|
|
return "<Rapporto %s del %s>" % (self.pk, self.created)
|