zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-08-23 09:41:05 +02:00
Code Issues Projects Releases Packages Wiki Activity

Add endpoint for report

Browse source
This commit is contained in:
zandercymatics 2024-07-16 10:42:34 -06:00
parent b331f61d0d
commit f347ff9c7f
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
4 changed files with 43 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -108,7 +108,7 @@ class BaseExport(ABC):
return Q()
@classmethod
def get_filter_conditions(cls, start_date=None, end_date=None):
def get_filter_conditions(cls, request=None, start_date=None, end_date=None):
"""
Get a Q object of filter conditions to filter when building queryset.
"""
@ -191,7 +191,7 @@ class BaseExport(ABC):
return cls.update_queryset(queryset, **kwargs)
@classmethod
def export_data_to_csv(cls, csv_file, start_date=None, end_date=None):
def export_data_to_csv(cls, csv_file, request=None, start_date=None, end_date=None):
"""
All domain metadata:
Exports domains of all statuses plus domain managers.
@ -204,7 +204,7 @@ class BaseExport(ABC):
prefetch_related = cls.get_prefetch_related()
exclusions = cls.get_exclusions()
annotations_for_sort = cls.get_annotations_for_sort()
filter_conditions = cls.get_filter_conditions(start_date, end_date)
filter_conditions = cls.get_filter_conditions(request, start_date, end_date)
computed_fields = cls.get_computed_fields()
related_table_fields = cls.get_related_table_fields()
@ -543,6 +543,21 @@ class DomainDataType(DomainExport):
"federal_agency__agency",
]
class DomainDataTypeUser(DomainDataType):
"""
The DomainDataType report, but sliced on the current request user
"""
@classmethod
def get_filter_conditions(cls, request=None, start_date=None, end_date=None):
"""
Get a Q object of filter conditions to filter when building queryset.
"""
user_domain_roles = UserDomainRole.objects.filter(user=request.user)
domain_ids = user_domain_roles.values_list("domain_id", flat=True)
return Q(id__in=domain_ids)
class DomainDataFull(DomainExport):
"""
@ -601,7 +616,7 @@ class DomainDataFull(DomainExport):
return ["domain"]
@classmethod
def get_filter_conditions(cls, start_date=None, end_date=None):
def get_filter_conditions(cls, request=None, start_date=None, end_date=None):
"""
Get a Q object of filter conditions to filter when building queryset.
"""
@ -696,7 +711,7 @@ class DomainDataFederal(DomainExport):
return ["domain"]
@classmethod
def get_filter_conditions(cls, start_date=None, end_date=None):
def get_filter_conditions(cls, request=None, start_date=None, end_date=None):
"""
Get a Q object of filter conditions to filter when building queryset.
"""
@ -794,7 +809,7 @@ class DomainGrowth(DomainExport):
return ["domain"]
@classmethod
def get_filter_conditions(cls, start_date=None, end_date=None):
def get_filter_conditions(cls, request=None, start_date=None, end_date=None):
"""
Get a Q object of filter conditions to filter when building queryset.
"""
@ -866,7 +881,7 @@ class DomainManaged(DomainExport):
return ["permissions"]
@classmethod
def get_filter_conditions(cls, start_date=None, end_date=None):
def get_filter_conditions(cls, request=None, start_date=None, end_date=None):
"""
Get a Q object of filter conditions to filter when building queryset.
"""
@ -1001,7 +1016,7 @@ class DomainUnmanaged(DomainExport):
return ["permissions"]
@classmethod
def get_filter_conditions(cls, start_date=None, end_date=None):
def get_filter_conditions(cls, request=None, start_date=None, end_date=None):
"""
Get a Q object of filter conditions to filter when building queryset.
"""
@ -1231,7 +1246,7 @@ class DomainRequestGrowth(DomainRequestExport):
]
@classmethod
def get_filter_conditions(cls, start_date=None, end_date=None):
def get_filter_conditions(cls, request=None, start_date=None, end_date=None):
"""
Get a Q object of filter conditions to filter when building queryset.
"""