92 lines
3 KiB
Python
92 lines
3 KiB
Python
from django.db import models
|
|
from django.db.models import Max
|
|
from django.utils.translation import gettext_lazy as _
|
|
from utils.constants import (
|
|
STR_SM_SIZE,
|
|
STR_SIZE,
|
|
STR_EXTEND_SIZE,
|
|
)
|
|
|
|
from user.models import User, Institution
|
|
from evidence.models import Property
|
|
# from device.models import Device
|
|
|
|
|
|
class LotTag(models.Model):
|
|
name = models.CharField(max_length=STR_SIZE, blank=False, null=False)
|
|
owner = models.ForeignKey(Institution, on_delete=models.CASCADE)
|
|
user = models.ForeignKey(User, on_delete=models.SET_NULL, null=True, blank=True)
|
|
inbox = models.BooleanField(default=False)
|
|
order = models.PositiveIntegerField(default=0)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
|
|
def save(self, *args, **kwargs):
|
|
if not self.pk:
|
|
# set the order to be last
|
|
max_order = LotTag.objects.filter(owner=self.owner).aggregate(Max('order'))['order__max']
|
|
self.order = (max_order or 0) + 1
|
|
super().save(*args, **kwargs)
|
|
|
|
|
|
def delete(self, *args, **kwargs):
|
|
institution = self.owner
|
|
order = self.order
|
|
super().delete(*args, **kwargs)
|
|
# Adjust the order of other instances
|
|
LotTag.objects.filter(owner=institution, order__gt=order).update(order=models.F('order') - 1)
|
|
|
|
|
|
class DeviceLot(models.Model):
|
|
lot = models.ForeignKey("Lot", on_delete=models.CASCADE)
|
|
device_id = models.CharField(max_length=STR_EXTEND_SIZE, blank=False, null=False)
|
|
|
|
|
|
class Lot(models.Model):
|
|
created = models.DateTimeField(auto_now_add=True)
|
|
updated = models.DateTimeField(auto_now=True)
|
|
name = models.CharField(max_length=STR_SIZE, blank=True, null=True)
|
|
code = models.CharField(max_length=STR_SIZE, blank=True, null=True)
|
|
description = models.CharField(max_length=STR_SIZE, blank=True, null=True)
|
|
closed = models.BooleanField(default=False)
|
|
owner = models.ForeignKey(Institution, on_delete=models.CASCADE)
|
|
user = models.ForeignKey(User, on_delete=models.SET_NULL, null=True, blank=True)
|
|
type = models.ForeignKey(LotTag, on_delete=models.CASCADE)
|
|
|
|
class Meta:
|
|
constraints = [
|
|
models.UniqueConstraint(fields=['owner', 'name', 'type'], name='unique_institution_and_name')
|
|
]
|
|
|
|
|
|
def add(self, v):
|
|
if DeviceLot.objects.filter(lot=self, device_id=v).exists():
|
|
return
|
|
DeviceLot.objects.create(lot=self, device_id=v)
|
|
|
|
def remove(self, v):
|
|
for d in DeviceLot.objects.filter(lot=self, device_id=v):
|
|
d.delete()
|
|
|
|
@property
|
|
def devices(self):
|
|
return DeviceLot.objects.filter(lot=self)
|
|
|
|
|
|
class LotProperty(Property):
|
|
lot = models.ForeignKey(Lot, on_delete=models.CASCADE)
|
|
|
|
class Type(models.IntegerChoices):
|
|
SYSTEM = 0, "System"
|
|
USER = 1, "User"
|
|
|
|
type = models.SmallIntegerField(choices=Type.choices, default=Type.USER)
|
|
|
|
class Meta:
|
|
constraints = [
|
|
models.UniqueConstraint(
|
|
fields=["key", "lot"], name="property_unique_type_key_lot")
|
|
]
|