blob: 0e6b03e840217e77bc757c8f5f5cf2355fa3b36c (
plain) (
tree)
|
|
from datetime import datetime
import factory
from django.core.management.base import BaseCommand, CommandError
from factory import DjangoModelFactory, Faker, SubFactory, post_generation
from ctrack.organisation.models import Organisation, Person
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) # TODO we need to create OrganisationFactory
role = Faker("job")
email = Faker("ascii_company_email")
secondary_email = "ascii_company_email"
mobile = Faker("msisdn", locale="en_GB")
landline = Faker("phone_number", locale="en_GB")
date_updated = factory.LazyFunction(datetime.now)
updated_by = # TODO we need to create a fake User
clearance =
clearance_sponsor =
clearance_start_date =
clearance_last_checked =
clearance_expiry =
active = True
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}"
# )
# )
|
