84 lines
2.3 KiB
Python
84 lines
2.3 KiB
Python
from django.db import models
|
|
from django.utils import timezone
|
|
|
|
import logging
|
|
logger = logging.getLogger('django')
|
|
|
|
|
|
class CustomModel(models.Model):
|
|
""" Custom model class with default fields to use. """
|
|
created_date = models.DateTimeField(default=timezone.now)
|
|
|
|
class Meta:
|
|
abstract = True
|
|
|
|
|
|
class EquipmentType(CustomModel):
|
|
name = models.CharField(max_length=100)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
|
|
class State(CustomModel):
|
|
name = models.CharField(max_length=100)
|
|
EquipmentType = models.ForeignKey(
|
|
EquipmentType, on_delete=models.PROTECT, null=True, blank=True)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
|
|
class KegType(CustomModel):
|
|
name = models.CharField(max_length=100)
|
|
manufacturer = models.CharField(max_length=100, blank=True)
|
|
size_gal = models.DecimalField(max_digits=6, decimal_places=2)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
|
|
class Equipment(CustomModel):
|
|
name = models.CharField(max_length=100)
|
|
equipment_type = models.ForeignKey(EquipmentType, on_delete=models.PROTECT)
|
|
keg_type = models.ForeignKey(KegType, on_delete=models.PROTECT)
|
|
state = models.ForeignKey(State, on_delete=models.PROTECT)
|
|
|
|
def __str__(self):
|
|
return '{} {}'.format(self.name, self.id)
|
|
|
|
|
|
class TransactionType(CustomModel):
|
|
name = models.CharField(max_length=100)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
|
|
class MaintenanceType(CustomModel):
|
|
name = models.CharField(max_length=100)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
|
|
class EquipmentMaintenance(CustomModel):
|
|
equipment = models.ForeignKey(Equipment, on_delete=models.CASCADE)
|
|
notes = models.TextField(max_length=500, blank=True, null=True)
|
|
|
|
|
|
class EquipmentMaintenanceLine(CustomModel):
|
|
maintenance = models.ForeignKey(
|
|
EquipmentMaintenance, on_delete=models.CASCADE)
|
|
maintenance_type = models.ForeignKey(
|
|
MaintenanceType, on_delete=models.PROTECT)
|
|
notes = models.TextField(max_length=500, blank=True, null=True)
|
|
|
|
|
|
class EquipmentTransactions(CustomModel):
|
|
equipment = models.ForeignKey(Equipment, on_delete=models.CASCADE)
|
|
transaction_type = models.ForeignKey(
|
|
TransactionType, on_delete=models.PROTECT)
|
|
beer_batch = models.ForeignKey(
|
|
'beer.Batch', on_delete=models.PROTECT, blank=True, null=True)
|