network_inventory/devices/models/warranty.py

50 lines
1.4 KiB
Python
Raw Normal View History

2019-06-10 21:56:21 +02:00
from django.db import models
from core.models import Category
from core.utils import td_format
from customers.models import Customer
2020-01-12 16:21:12 +01:00
from .device import Device
2019-06-10 21:56:21 +02:00
class WarrantyType(Category):
description = models.TextField()
class Meta:
2020-07-07 22:12:57 +02:00
ordering = ['name']
2019-06-10 21:56:21 +02:00
verbose_name_plural = "Warranty Types"
class Warranty(models.Model):
customer = models.ForeignKey(Customer,
on_delete=models.CASCADE,
blank=True)
2019-06-10 21:56:21 +02:00
device = models.ForeignKey(Device, on_delete=models.CASCADE)
valid_from = models.DateField()
valid_until = models.DateField()
warranty_type = models.ForeignKey(WarrantyType,
models.SET_NULL,
blank=True,
null=True)
def save(self, *args, **kwargs):
self.customer = self.device.customer
super(Warranty, self).save(*args, **kwargs)
2019-06-10 21:56:21 +02:00
def __str__(self):
return str(self.device)
2019-06-10 21:56:21 +02:00
class Meta:
2020-07-07 22:12:57 +02:00
ordering = ['customer']
2019-06-10 21:56:21 +02:00
verbose_name_plural = "Warranties"
constraints = [
models.CheckConstraint(check=models.Q(
valid_from__lte=models.F("valid_until")),
name='valid_from_lt_valid_until'),
]
@property
def duration_in_years(self):
delta = self.valid_until - self.valid_from
return td_format(delta)