1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
|
from django.db import models
from ctrack.caf.managers import ApplicableSystemManager
from ctrack.organisations.models import Organisation, Person
from django.urls.base import reverse
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 ApplicableSystem(models.Model):
CRITICAL = "CR"
IMPORTANT = "IM"
SYSTEM_CATEGORISATION = (
(CRITICAL, "Critical"),
(IMPORTANT, "Important (Legacy use only)"),
)
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, help_text="System name assigned by OES")
function = models.TextField(
max_length=1000,
blank=True,
null=True,
help_text="How the system is relevant to delivering or supporting the "
"essential service",
)
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="applicable_systems",
)
dft_categorisation = models.CharField(
max_length=2,
choices=SYSTEM_CATEGORISATION,
default=CRITICAL,
verbose_name="DfT Categorisation",
help_text="Refer to documentation for description of these criteria",
)
class Meta:
verbose_name = "NIS System"
def get_primary_contact(self):
return self.organisation.person_set.filter(primary_nis_contact=True)
def __str__(self):
return f"{self.organisation.name} | {self.name}"
objects = ApplicableSystemManager()
class CAF(models.Model):
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 get_absolute_url(self):
return reverse("caf:detail", kwargs={"pk": self.pk})
def applicable_systems(self):
"""
Returns a Queryset of objects we can use in our templates.
"""
return ApplicableSystem.objects.filter(caf=self)
def organisation(self):
first_ass = ApplicableSystem.objects.filter(caf=self).first()
return first_ass.organisation
def sub_mode(self):
return self.organisation().submode
def __str__(self):
# Get the organisation and applicable system
ass = ApplicableSystem.objects.filter(caf=self).first()
return f"CAF | {ass.organisation.name}_v{self.version}"
|
