from django.db import models
from ctrack.organisations.models import Organisation, Person
class Grading(models.Model):
GRADING_TYPE = [("CONFIDENCE", "Confidence"), ("QUALITY", "Quality"), ("MISC", "Misc")]
descriptor = models.CharField(max_length=2, help_text="Q1, C1, etc")
description = models.TextField(max_length=250)
type = models.CharField(max_length=20, choices=GRADING_TYPE, help_text="Type of grading")
def __str__(self):
return self.descriptor
class FileStore(models.Model):
descriptor = models.CharField(max_length=100)
virtual_location = models.CharField(
max_length=100, help_text="USB, Rosa, email, etc"
)
physical_location = models.CharField(
max_length=100, blank=True, help_text="Cupboard, room, building, etc"
) # cupboard, room, building, address
physical_location_organisation = models.ForeignKey(
Organisation, on_delete=models.CASCADE
)
def __str__(self):
return self.descriptor
class DocumentFile(models.Model):
FILETYPE_CHOICES = [(1, "Excel"), (2, "Word"), (3, "PDF"), (4, "Hard Copy")]
name = models.CharField(max_length=255)
type = models.IntegerField(choices=FILETYPE_CHOICES, default=1)
file_store_location = models.ForeignKey(FileStore, on_delete=models.CASCADE)
class EssentialService(models.Model):
def get_sentinel_org():
"""
We need this so that we can ensure models.SET() is applied with a callable
to handle when Users are deleted from the system, preventing the Organisation
objects related to them being deleted also.
"""
return Organisation.objects.get_or_create(name="DELETED ORGANISATION")[0]
name = models.CharField(max_length=256)
description = models.TextField(max_length=1000)
organisation = models.ForeignKey(
Organisation, on_delete=models.SET(get_sentinel_org)
)
caf = models.ForeignKey("CAF", on_delete=models.CASCADE, blank=True, null=True, related_name="essential_services")
class Meta:
verbose_name = "Essential Service"
def __str__(self):
return f"{self.organisation.name} | {self.name}"
class CAF(models.Model):
owner = models.ForeignKey(Organisation, on_delete=models.CASCADE)
quality_grading = models.ForeignKey(Grading, on_delete=models.CASCADE, blank=True, null=True,
related_name="quality_grading")
confidence_grading = models.ForeignKey(Grading, on_delete=models.CASCADE, blank=True, null=True,
related_name="confidence_grading")
file = models.ForeignKey(DocumentFile, on_delete=models.CASCADE, blank=True, null=True)
version = models.CharField(max_length=10, blank=True, null=True)
triage_review_date = models.DateField(blank=True, null=True)
triage_review_inspector = models.ForeignKey(Person, on_delete=models.CASCADE, blank=True, null=True)
comments = models.TextField(max_length=1000)
class Meta:
verbose_name = "CAF"
def __str__(self):
return f"CAF | {self.owner}"
