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
|
import factory
from django.core.management.base import BaseCommand, CommandError
from factory import DjangoModelFactory, Faker, post_generation, SubFactory
from ctrack.organisation.models import Person, Organisation
class OrganisationFactory(DjangoModelFactory):
pass
class PersonFactory(DjangoModelFactory):
class Meta:
model = Person
primary_nis_contact = True
voluntary_point_of_contact = True
has_egress = False
title = Faker("prefix")
job_title = Faker("job")
first_name = Faker("first_name")
last_name = Faker("last_name")
organisation = SubFactory(OrganisationFactory)
role =
email =
secondary_email =
mobile =
landline =
date_updated =
updated_by =
clearance =
clearance_sponsor =
clearance_start_date =
clearance_last_checked =
clearance_expiry =
active =
date_ended =
predecessor =
comments =
class Command(BaseCommand):
help = """
Creates a bunch of people and organisations for them to work in.
python manage.py generate_people
"""
# def add_arguments(self, parser):
# parser.add_argument("year", nargs="+", type=int)
#
# def handle(self, *args, **options):
# for opt in options["year"]:
# FinancialQuarter.objects.create(quarter=1, year=opt)
# FinancialQuarter.objects.create(quarter=2, year=opt)
# FinancialQuarter.objects.create(quarter=3, year=opt)
# FinancialQuarter.objects.create(quarter=4, year=opt)
# self.stdout.write(
# self.style.SUCCESS(
# f"Created FinancialQuarter objects for the years: {opt}"
# )
# )
|
