diff options
| -rw-r--r-- | ctrack/organisations/management/commands/populate_db.py | 32 | ||||
| -rwxr-xr-x | utility/trunc_and_populate.sh | 9 |
2 files changed, 27 insertions, 14 deletions
diff --git a/ctrack/organisations/management/commands/populate_db.py b/ctrack/organisations/management/commands/populate_db.py index e994b41..6f51a22 100644 --- a/ctrack/organisations/management/commands/populate_db.py +++ b/ctrack/organisations/management/commands/populate_db.py @@ -160,23 +160,12 @@ class Command(BaseCommand): # Every org gets on CAF for now for org in orgs: - # create a CAF for it - 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, - ) - # create an ApplicableSystem for it - app_s = ApplicableSystemFactory.create( - name=random.choice(fnames), - organisation=org, - caf=caf, - ) + # 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 + _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 ) @@ -187,3 +176,18 @@ class Command(BaseCommand): 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, + ) diff --git a/utility/trunc_and_populate.sh b/utility/trunc_and_populate.sh new file mode 100755 index 0000000..8ac3a4f --- /dev/null +++ b/utility/trunc_and_populate.sh @@ -0,0 +1,9 @@ +#!/usr/bin/env bash + +echo "Truncating the database." +sudo -u postgres psql -f utility/truncate_script.sql ctrack + +sleep 1 + +echo "Populating the database with test data." +python manage.py populate_db 20 |