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

Add logging / additional validation

Browse source
This commit is contained in:
zandercymatics 2023-11-27 09:35:35 -07:00
parent 1b1a241101
commit dca3eb5676
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
2 changed files with 16 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/registrar/management/commands/generate_current_federal_report.py
View file

@ -20,6 +20,11 @@ class Command(BaseCommand):
# Ensures a slash is added
directory = os.path.join(options.get("directory"), "")
logger.info("Generating report...")
# TODO - Delete
current_directory = os.getcwd()
logger.info(f"Current working directory: {current_directory}")
self.generate_current_federal_report(directory)
logger.info(f"Success! Created {directory}current-federal.csv")
@ -28,3 +33,6 @@ class Command(BaseCommand):
file_path = os.path.join(directory, "current-federal.csv")
with open(file_path, "w") as file:
csv_export.export_data_federal_to_csv(file)
if not os.path.exists(file_path):
raise FileNotFoundError(f"Could not find newly created file at '{file_path}'")

8
src/registrar/management/commands/generate_current_full_report.py
View file

@ -20,6 +20,11 @@ class Command(BaseCommand):
# Ensures a slash is added
directory = os.path.join(options.get("directory"), "")
logger.info("Generating report...")
# TODO - Delete
current_directory = os.getcwd()
logger.info(f"Current working directory: {current_directory}")
self.generate_current_full_report(directory)
logger.info(f"Success! Created {directory}current-full.csv")
@ -28,3 +33,6 @@ class Command(BaseCommand):
file_path = os.path.join(directory, "current-full.csv")
with open(file_path, "w") as file:
csv_export.export_data_full_to_csv(file)
if not os.path.exists(file_path):
raise FileNotFoundError(f"Could not find newly created file at '{file_path}'")