From ba025f2c6ac8db0f612b2bc8f3d84cf9c8606db6 Mon Sep 17 00:00:00 2001 From: zandercymatics <141044360+zandercymatics@users.noreply.github.com> Date: Thu, 6 Jun 2024 15:17:10 -0600 Subject: [PATCH] Linting --- src/registrar/assets/sass/_theme/_admin.scss | 2 +- src/registrar/tests/test_reports.py | 13 ++++-------- src/registrar/utility/csv_export.py | 22 ++++++++++---------- src/registrar/views/admin_views.py | 3 ++- 4 files changed, 18 insertions(+), 22 deletions(-) diff --git a/src/registrar/assets/sass/_theme/_admin.scss b/src/registrar/assets/sass/_theme/_admin.scss index 09c76fd3a..eec72c978 100644 --- a/src/registrar/assets/sass/_theme/_admin.scss +++ b/src/registrar/assets/sass/_theme/_admin.scss @@ -767,4 +767,4 @@ div.dja__model-description{ a { min-width: 160px !important; } -} \ No newline at end of file +} diff --git a/src/registrar/tests/test_reports.py b/src/registrar/tests/test_reports.py index 493cee430..e0b238098 100644 --- a/src/registrar/tests/test_reports.py +++ b/src/registrar/tests/test_reports.py @@ -681,13 +681,10 @@ class ExportDataTest(MockDb, MockEppLib): } all_requests = DomainRequest.objects.filter(**filter_condition).order_by(*sort_fields).distinct() - all_requests_dict = all_requests.values( - "requested_domain_name", - "generic_org_type", - "federal_type", - "submission_date" + extra = all_requests.values("requested_domain_name", "generic_org_type", "federal_type", "submission_date") + DomainRequestExport.write_csv_for_requests( + writer, columns, all_requests, extra_request_fields=extra, should_write_header=True ) - DomainRequestExport.write_csv_for_requests(writer, columns, requests=all_requests_dict, should_write_header=True) # Reset the CSV file's position to the beginning csv_file.seek(0) # Read the content into a variable @@ -695,9 +692,7 @@ class ExportDataTest(MockDb, MockEppLib): # We expect READY domains first, created between today-2 and today+2, sorted by created_at then name # and DELETED domains deleted between today-2 and today+2, sorted by deleted then name expected_content = ( - "Domain request,Domain type\n" - "city3.gov,Federal - Executive\n" - "city4.gov,Federal - Executive\n" + "Domain request,Domain type\n" "city3.gov,Federal - Executive\n" "city4.gov,Federal - Executive\n" ) # Normalize line endings and remove commas, diff --git a/src/registrar/utility/csv_export.py b/src/registrar/utility/csv_export.py index 4593d74d8..db5b581cd 100644 --- a/src/registrar/utility/csv_export.py +++ b/src/registrar/utility/csv_export.py @@ -6,11 +6,10 @@ from registrar.models import ( DomainInvitation, DomainRequest, DomainInformation, - User, PublicContact, UserDomainRole, ) -from django.db.models import QuerySet, Value, Case, When, CharField, Count, Q, F, Value, CharField +from django.db.models import QuerySet, Value, Case, When, CharField, Count, Q, F from django.utils import timezone from django.core.paginator import Paginator from django.db.models.functions import Concat, Coalesce @@ -751,7 +750,7 @@ class DomainRequestExport: @staticmethod def parse_row_for_requests(columns, request: DomainRequest, extra_fields: QuerySet): """ - Given a set of columns and a request dictionary, + Given a set of columns and a request dictionary, generate a new row from cleaned column data """ @@ -779,7 +778,7 @@ class DomainRequestExport: human_readable_election_board = "N/A" if request.is_election_board is not None: human_readable_election_board = "Yes" if request.is_election_board else "No" - + requested_domain = request.requested_domain if extra_fields.get("requested_domain_name") is None: extra_fields["requested_domain_name"] = getattr(requested_domain, "name", "No requested domain") @@ -909,13 +908,14 @@ class DomainRequestExport: "status", "requested_domain__name", ] - requests = DomainRequest.objects.select_related( - "creator", "authorizing_official", "federal_agency", "investigator", "requested_domain" - ).exclude( - status__in=excluded_statuses - ).order_by( - *order_by - ).distinct() + requests = ( + DomainRequest.objects.select_related( + "creator", "authorizing_official", "federal_agency", "investigator", "requested_domain" + ) + .exclude(status__in=excluded_statuses) + .order_by(*order_by) + .distinct() + ) extra_fields = DomainRequestExport.annotate_and_prepare_domain_request_data(requests) DomainRequestExport.write_csv_for_requests(writer, columns, requests, extra_fields, should_write_header=True) diff --git a/src/registrar/views/admin_views.py b/src/registrar/views/admin_views.py index a5274fb70..88a0d7de0 100644 --- a/src/registrar/views/admin_views.py +++ b/src/registrar/views/admin_views.py @@ -165,8 +165,9 @@ class ExportDataFederal(View): class ExportDomainRequestDataFull(View): + """Generates a downloaded report containing all Domain Requests (except started)""" def get(self, request, *args, **kwargs): - # Smaller export based on 1 + """Returns a content disposition response for current-full-domain-request.csv""" response = HttpResponse(content_type="text/csv") response["Content-Disposition"] = 'attachment; filename="current-full-domain-request.csv"' csv_export.DomainRequestExport.export_full_domain_request_report(response)