darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2025-12-20 15:31:15 -05:00
Code Issues Packages Projects Releases Wiki Activity

Swapped random for secrets

Browse Source
This commit is contained in:
Anastasia Gradova
2024-03-01 14:43:36 -07:00
parent c0e3a68a07
commit 980e32c86c
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
app/commands.py
View File

@@ -1,7 +1,7 @@
import csv import csv
import functools import functools
import itertools import itertools
import random import secrets
import uuid import uuid
from datetime import datetime, timedelta from datetime import datetime, timedelta
from os import getenv from os import getenv
@@ -900,15 +900,15 @@ def add_test_organization_to_db(generate):
"Science and Technology", "Science and Technology",
] ]
agency = random.choice(agency_names) agency = secrets.choice(agency_names)
speciality = random.choice(government_sectors) speciality = secrets.choice(government_sectors)
return f"{fake.word().capitalize()} {speciality} {agency}" return f"{fake.word().capitalize()} {speciality} {agency}"
for num in range(1, int(generate) + 1): for num in range(1, int(generate) + 1):
org = create_organization( org = create_organization(
name=generate_gov_agency(), name=generate_gov_agency(),
organization_type=random.choice(["federal", "state", "other"]), organization_type=secrets.choice(["federal", "state", "other"]),
) )
print(f"{num} {org.name} created") print(f"{num} {org.name} created")