Update csv_export.py

This commit is contained in:
zandercymatics 2023-11-13 14:53:19 -07:00
parent 86dbe6b6a0
commit 0d64d9f040
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7

View file

@ -13,9 +13,12 @@ def export_domains_to_writer(writer, columns, sort_fields, filter_condition):
domainInfos = DomainInformation.objects.filter(**filter_condition).order_by(*sort_fields)
for domainInfo in domainInfos:
security_contacts = domainInfo.domain.contacts.filter(contact_type=PublicContact.ContactTypeChoices.SECURITY)
ao = (
(domainInfo.authorizing_official.first_name or "") + " " + (domainInfo.authorizing_official.last_name or "")
)
# For linter
ao = " "
if domainInfo.authorizing_official:
first_name = domainInfo.authorizing_official.first_name or ""
last_name = domainInfo.authorizing_official.last_name or ""
ao = first_name + " " + last_name
# create a dictionary of fields which can be included in output
FIELDS = {
"Domain name": domainInfo.domain.name,
@ -26,7 +29,7 @@ def export_domains_to_writer(writer, columns, sort_fields, filter_condition):
"Organization name": domainInfo.organization_name,
"City": domainInfo.city,
"State": domainInfo.state_territory,
"AO": ao if domainInfo.authorizing_official else " ",
"AO": ao,
"AO email": domainInfo.authorizing_official.email if domainInfo.authorizing_official else " ",
"Security Contact Email": security_contacts[0].email if security_contacts else " ",
"Status": domainInfo.domain.state,