zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-08-12 04:29:44 +02:00
Code Issues Projects Releases Packages Wiki Activity

trigger PR

Browse source
This commit is contained in:
Rachid Mrad 2024-11-15 16:56:15 -05:00
parent 13d61fa40b
commit 9565731a93
No known key found for this signature in database
1 changed files with 27 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

27
src/registrar/fixtures/fixtures_users.py
View file

@ -23,6 +23,13 @@ class UserFixture:
""" """
ADMINS = [ ADMINS = [
# {
# "username": "a6b5dd22-f54e-4d55-83ce-ca7b65b2dc1a",
# "first_name": "Rach test admin",
# "last_name": "Mrad test admin",
# "email": "rachid.mrad+2@gmail.com",
# "title": "Super admin tester",
# },
{ {
"username": "aad084c3-66cc-4632-80eb-41cdf5c5bcbf", "username": "aad084c3-66cc-4632-80eb-41cdf5c5bcbf",
"first_name": "Aditi", "first_name": "Aditi",
@ -154,6 +161,14 @@ class UserFixture:
] ]
STAFF = [ STAFF = [
# {
# "username": "994b7a90-f1d1-4140-a3d2-ff34183c7ee2",
# "first_name": "Rach test staff",
# "last_name": "Mrad test staff",
# "email": "rachid.mrad+3@gmail.com",
# "title": "Super staff tester",
# },
{ {
"username": "ffec5987-aa84-411b-a05a-a7ee5cbcde54", "username": "ffec5987-aa84-411b-a05a-a7ee5cbcde54",
"first_name": "Aditi-Analyst", "first_name": "Aditi-Analyst",
@ -309,6 +324,18 @@ class UserFixture:
# Get all users to be updated (both new and existing) # Get all users to be updated (both new and existing)
created_or_existing_users = User.objects.filter(username__in=[user.get("username") for user in users]) created_or_existing_users = User.objects.filter(username__in=[user.get("username") for user in users])
# Update `is_staff` for existing users if necessary
users_to_update = []
for user in created_or_existing_users:
if not user.is_staff:
user.is_staff = True
users_to_update.append(user)
# Save any users that were updated
if users_to_update:
User.objects.bulk_update(users_to_update, ["is_staff"])
logger.info(f"Updated {len(users_to_update)} existing users to have is_staff=True.")
# Filter out users who are already in the group # Filter out users who are already in the group
users_not_in_group = created_or_existing_users.exclude(groups__id=group.id) users_not_in_group = created_or_existing_users.exclude(groups__id=group.id)