From ec8e8b1f65093d90d018744cf175bfc56320cbc8 Mon Sep 17 00:00:00 2001 From: Rachid Mrad Date: Wed, 20 Nov 2024 13:15:27 -0500 Subject: [PATCH] lint --- src/registrar/fixtures/fixtures_requests.py | 10 +++++----- src/registrar/fixtures/fixtures_users.py | 3 +-- 2 files changed, 6 insertions(+), 7 deletions(-) diff --git a/src/registrar/fixtures/fixtures_requests.py b/src/registrar/fixtures/fixtures_requests.py index 9f736d0d1..c413f4b62 100644 --- a/src/registrar/fixtures/fixtures_requests.py +++ b/src/registrar/fixtures/fixtures_requests.py @@ -194,8 +194,8 @@ class DomainRequestFixture: if not request.requested_domain: if "requested_domain" in request_dict and request_dict["requested_domain"] is not None: return DraftDomain.objects.get_or_create(name=request_dict["requested_domain"])[0] - - # Generate a unique fake domain + + # Generate a unique fake domain return cls.fake_dot_gov() return request.requested_domain @@ -233,19 +233,19 @@ class DomainRequestFixture: except Exception as e: logger.warning(f"Expected fixture portfolio, did not find it: {e}") return None - + @classmethod def _get_random_sub_organization(cls, request): try: # Filter Suborganizations by the request's portfolio portfolio_suborganizations = Suborganization.objects.filter(portfolio=request.portfolio) - + # Assuming SuborganizationFixture.SUBORGS is a list of dictionaries with a "name" key suborganization_names = [suborg["name"] for suborg in SuborganizationFixture.SUBORGS] # Further filter by names in suborganization_names suborganization_options = portfolio_suborganizations.filter(name__in=suborganization_names) - + # Randomly choose one if any exist return random.choice(suborganization_options) if suborganization_options.exists() else None # nosec except Exception as e: diff --git a/src/registrar/fixtures/fixtures_users.py b/src/registrar/fixtures/fixtures_users.py index 0c4da9bac..b3ab530c3 100644 --- a/src/registrar/fixtures/fixtures_users.py +++ b/src/registrar/fixtures/fixtures_users.py @@ -161,7 +161,6 @@ class UserFixture: ] STAFF = [ - # { # "username": "994b7a90-f1d1-4140-a3d2-ff34183c7ee2", # "first_name": "Rach test staff", @@ -344,7 +343,7 @@ class UserFixture: # Only append the user if any of the fields were updated if updated: users_to_update.append(user) - + # Save any users that were updated if users_to_update: User.objects.bulk_update(users_to_update, ["is_staff"])