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
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
|
import random
from random import randint, choice
from django.core.management import BaseCommand
from django.core.management import CommandParser
from ctrack.caf.models import CAF
from ctrack.caf.tests.factories import (
GradingFactory,
FileStoreFactory,
CAFFactory,
ApplicableSystemFactory,
)
from ctrack.organisations.models import AddressType, Organisation
from ctrack.organisations.models import Mode
from ctrack.organisations.models import Submode
from ctrack.organisations.tests.factories import AddressFactory
from ctrack.organisations.tests.factories import OrganisationFactory
from ctrack.organisations.tests.factories import PersonFactory
from ctrack.organisations.tests.factories import RoleFactory
from ctrack.organisations.tests.factories import UserFactory
from ctrack.register.tests.factories import EngagementEventFactory
from ctrack.register.tests.factories import EngagementTypeFactory
fnames = [
"Clock Pylon Systems",
"Ultramarine Hanglider Navigator",
"Membranous Floor Heaters",
"Alan's Wardrobe Hinge Circuits",
"Marine Sluicegate Extension Pulleys",
"Ironway Prob Modelling Area",
"Bufferage Clippers",
"Slow Gauze Thread Manipulator",
"Terratoast Piling",
"Accounting and Warehouse Conducer",
"Able Hopscotch Mirrors",
"Jolly Main Legacy Circuitry",
]
class Command(BaseCommand):
help = """
Creates a bunch of people and organisations for them to work in.
Also creates users and roles as these are required fields.
"""
def add_arguments(self, parser: CommandParser) -> None:
parser.add_argument("number", nargs=1, type=int)
def handle(self, *args, **options):
number = options["number"][0]
# Set up some reasonable Modes and SubModes
m1 = Mode.objects.create(descriptor="Rail")
m2 = Mode.objects.create(descriptor="Maritime")
sb1 = Submode.objects.create(descriptor="Light Rail", mode=m1)
sb2 = Submode.objects.create(descriptor="Rail Maintenance", mode=m1)
sb3 = Submode.objects.create(descriptor="Rail Infrastructure", mode=m1)
sb4 = Submode.objects.create(descriptor="International Rail", mode=m1)
sb5 = Submode.objects.create(descriptor="Passenger Port", mode=m2)
sb6 = Submode.objects.create(descriptor="Freight Port", mode=m2)
sb7 = Submode.objects.create(descriptor="Shipping Infrastructure", mode=m2)
submodes = [sb1, sb2, sb3, sb4, sb5, sb6, sb7]
# we need a User object to completed the updated_by fields in Organisation and Person
user = (
UserFactory.create()
) # we need to have at least one user for the updated_by field
# Create 40 Organisation objects
orgs = [
OrganisationFactory.create(submode=submodes[randint(0, len(submodes) - 1)])
for org in range(40)
]
# Create 40 Address objects
addr_type = AddressType.objects.create(descriptor="Primary Address")
for org in orgs:
AddressFactory.create(type=addr_type, organisation=org)
roles = [
RoleFactory.create() for x in range(10)
] # because we have a many-to-many relationship with Role, we need to create one and pass it in
for org in orgs:
PersonFactory.create(
role=choice(roles),
updated_by=user,
predecessor=None,
organisation__submode=choice(submodes),
organisation=org,
)
inspector_role = RoleFactory.create(name="Compliance Inspector")
# set up some EngagementEvents
p1 = PersonFactory.create(
role=choice(roles),
updated_by=user,
predecessor=None,
organisation__submode=choice(submodes),
organisation=org,
)
p2 = PersonFactory.create(
role=choice(roles),
updated_by=user,
predecessor=None,
organisation__submode=choice(submodes),
organisation=org,
)
p3 = PersonFactory.create(
role=choice(roles),
updated_by=user,
predecessor=None,
organisation__submode=choice(submodes),
organisation=org,
)
regulator_org = OrganisationFactory.create(
submode=None,
name="The Regulator",
designation_type=3,
registered_company_name="The Regulator - HMG",
comments="This is the real regulator.",
)
inspectors = [
PersonFactory.create(
role=inspector_role,
updated_by=user,
job_title="Compliance Inspector",
predecessor=None,
organisation__submode=None,
organisation=regulator_org,
)
for _ in range(5)
]
etf1 = EngagementTypeFactory(descriptor="Information Notice")
etf2 = EngagementTypeFactory(descriptor="Designation Letter")
etf3 = EngagementTypeFactory(descriptor="CAF - Initial Submission", enforcement_instrument=False)
ee1 = EngagementEventFactory.create(type=etf1, user=user, participants=[p1, p2])
ee2 = EngagementEventFactory.create(type=etf2, user=user, participants=[p3])
# Quality gradings
q_descriptors = ["Q1", "Q2", "Q3", "Q4", "Q5"]
for g in q_descriptors:
GradingFactory.create(descriptor=g, type="QUALITY")
# Confidence gradings
c_descriptors = ["C1", "C2", "C3", "C4", "C5"]
for g in c_descriptors:
GradingFactory.create(descriptor=g, type="CONFIDENCE")
# File store
fs = FileStoreFactory.create(physical_location_organisation=orgs[1])
# Every org gets on CAF for now
for org in orgs:
# create a CAF and ApplicableService for it
self._create_caf_app_service(c_descriptors, org, q_descriptors)
# CAF submissions - they create EngagementEvents
# Get a random CAF
_caf = CAF.objects.get(pk=1) # we should have one by now
ee3 = EngagementEventFactory.create(
type=etf3, user=user, participants=[inspectors[1], p2], related_caf=_caf
)
# TODO - adapt this so that it records more than just Persons created
self.stdout.write(
self.style.SUCCESS(
f"Created {number} Person object[s]! Go forth and multiply."
)
)
def _create_caf_app_service(self, c_descriptors, org, q_descriptors):
caf = CAFFactory.create(
quality_grading__descriptor=random.choice(q_descriptors),
confidence_grading__descriptor=random.choice(c_descriptors),
triage_review_date=None,
triage_review_inspector=None,
)
# Each CAF can have up to three systems associated with it
for s in range(random.randint(1,3)):
ApplicableSystemFactory.create(
name=random.choice(fnames),
organisation=org,
caf=caf,
)
|
