zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-08-11 20:19:38 +02:00
Code Issues Projects Releases Packages Wiki Activity

Linting

Browse source
This commit is contained in:
zandercymatics 2024-04-10 10:13:45 -06:00
parent 4f632111b8
commit 82f94600b7
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
1 changed files with 18 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

32
src/registrar/utility/csv_export.py
View file

@ -569,23 +569,25 @@ def get_sliced_domains(filter_condition):
""" """
return get_org_type_counts(DomainInformation, filter_condition) return get_org_type_counts(DomainInformation, filter_condition)
def get_sliced_requests(filter_condition): def get_sliced_requests(filter_condition):
"""Get filtered requests counts sliced by org type and election office.""" """Get filtered requests counts sliced by org type and election office."""
return get_org_type_counts(DomainRequest, filter_condition) return get_org_type_counts(DomainRequest, filter_condition)
def get_org_type_counts(model_class, filter_condition): def get_org_type_counts(model_class, filter_condition):
"""Returns a list of counts for each org type""" """Returns a list of counts for each org type"""
# Count all org types, such as federal # Count all org types, such as federal
dynamic_count_dict = {} dynamic_count_dict = {}
for choice in DomainRequest.OrganizationChoices: for choice in DomainRequest.OrganizationChoices:
choice_name = f"{choice}_count" choice_name = f"{choice}_count"
dynamic_count_dict[choice_name] = _org_type_count_query_builder(choice) dynamic_count_dict[choice_name] = _org_type_count_query_builder(choice)
# Static counts # Static counts
static_count_dict = { static_count_dict = {
# Count all distinct records # Count all distinct records
"total_count": Count('id'), "total_count": Count("id"),
# Count all election boards # Count all election boards
"election_board_count": Count(Case(When(is_election_board=True, then=1))), "election_board_count": Count(Case(When(is_election_board=True, then=1))),
} }
@ -598,32 +600,33 @@ def get_org_type_counts(model_class, filter_condition):
aggregates = model_queryset.aggregate(**merged_count_dict) aggregates = model_queryset.aggregate(**merged_count_dict)
# This can be automated but for the sake of readability, this is fixed for now. # This can be automated but for the sake of readability, this is fixed for now.
# To automate this would also mean the added benefit of # To automate this would also mean the added benefit of
# auto-updating (automatically adds new org types) charts, # auto-updating (automatically adds new org types) charts,
# but that requires an upstream refactor. # but that requires an upstream refactor.
return [ return [
# Total number of records # Total number of records
aggregates['total_count'], aggregates["total_count"],
# Number of records with org type FEDERAL # Number of records with org type FEDERAL
aggregates['federal_count'], aggregates["federal_count"],
# Number of records with org type INTERSTATE # Number of records with org type INTERSTATE
aggregates['interstate_count'], aggregates["interstate_count"],
# Number of records with org type STATE_OR_TERRITORY # Number of records with org type STATE_OR_TERRITORY
aggregates['state_or_territory_count'], aggregates["state_or_territory_count"],
# Number of records for TRIBAL # Number of records for TRIBAL
aggregates['tribal_count'], aggregates["tribal_count"],
# Number of records for COUNTY # Number of records for COUNTY
aggregates['county_count'], aggregates["county_count"],
# Number of records for CITY # Number of records for CITY
aggregates['city_count'], aggregates["city_count"],
# Number of records for SPECIAL_DISTRICT # Number of records for SPECIAL_DISTRICT
aggregates['special_district_count'], aggregates["special_district_count"],
# Number of records for SCHOOL_DISTRICT # Number of records for SCHOOL_DISTRICT
aggregates['school_district_count'], aggregates["school_district_count"],
# Number of records for ELECTION_BOARD # Number of records for ELECTION_BOARD
aggregates['election_board_count'], aggregates["election_board_count"],
] ]
def _org_type_count_query_builder(generic_org_type): def _org_type_count_query_builder(generic_org_type):
""" """
Returns an expression that counts the number of a given generic_org_type. Returns an expression that counts the number of a given generic_org_type.
@ -635,6 +638,7 @@ def _org_type_count_query_builder(generic_org_type):
""" """
return Count(Case(When(generic_org_type=generic_org_type, then=1))) return Count(Case(When(generic_org_type=generic_org_type, then=1)))
def export_data_managed_domains_to_csv(csv_file, start_date, end_date): def export_data_managed_domains_to_csv(csv_file, start_date, end_date):
"""Get counts for domains that have domain managers for two different dates, """Get counts for domains that have domain managers for two different dates,
get list of managed domains at end_date.""" get list of managed domains at end_date."""