formatting for readability

This commit is contained in:
David Kennedy 2023-10-27 08:11:13 -04:00
parent 47251d9edb
commit b72919a892
No known key found for this signature in database
GPG key ID: 6528A5386E66B96B
2 changed files with 93 additions and 65 deletions

View file

@ -751,39 +751,55 @@ class DomainAdmin(ListHeaderAdmin):
change_form_template = "django/admin/domain_change_form.html"
change_list_template = "django/admin/domain_change_list.html"
readonly_fields = ["state"]
def export_data_type(self, request):
# match the CSV example with all the fields
response = HttpResponse(content_type='text/csv')
response['Content-Disposition'] = 'attachment; filename="domains-by-type.csv"'
response = HttpResponse(content_type="text/csv")
response["Content-Disposition"] = 'attachment; filename="domains-by-type.csv"'
csv_export.export_data_type_to_csv(response)
return response
def export_data_full(self, request):
# Smaller export based on 1
response = HttpResponse(content_type='text/csv')
response['Content-Disposition'] = 'attachment; filename="domains-current-full.csv"'
response = HttpResponse(content_type="text/csv")
response[
"Content-Disposition"
] = 'attachment; filename="domains-current-full.csv"'
csv_export.export_data_full_to_csv(response)
return response
def export_data_federal(self, request):
# Federal only
response = HttpResponse(content_type='text/csv')
response['Content-Disposition'] = 'attachment; filename="domains-current-federal.csv"'
response = HttpResponse(content_type="text/csv")
response[
"Content-Disposition"
] = 'attachment; filename="domains-current-federal.csv"'
csv_export.export_data_federal_to_csv(response)
return response
def get_urls(self):
from django.urls import path
urlpatterns = super().get_urls()
info = self.model._meta.app_label, self.model._meta.model_name
my_url = [
path('export_data_type/', self.export_data_type, name='%s_%s_export_data_type' % info),
path('export_data_full/', self.export_data_full, name='%s_%s_export_data_full' % info),
path('export_data_federal/', self.export_data_federal, name='%s_%s_export_data_federal' % info),
path(
"export_data_type/",
self.export_data_type,
name="%s_%s_export_data_type" % info,
),
path(
"export_data_full/",
self.export_data_full,
name="%s_%s_export_data_full" % info,
),
path(
"export_data_federal/",
self.export_data_federal,
name="%s_%s_export_data_federal" % info,
),
]
return my_url + urlpatterns