Merge pull request #2795 from GSA-TTS/main

GSA-TTS · Nov 14, 2023 · fe85c06 · fe85c06
2 parents af86e78 + 4ff2ff3
commit fe85c06
Show file tree

Hide file tree

Showing 3 changed files with 1,112 additions and 97 deletions.
diff --git a/backend/census_historical_migration/management/commands/csv_to_postgres.py b/backend/census_historical_migration/management/commands/csv_to_postgres.py
@@ -52,16 +52,16 @@ def add_arguments(self, parser):
         )
 
     def handle(self, *args, **options):
-        folder = options.get("folder")
-        if not folder:
-            print("Please specify a folder name")
-            return
         if options.get("clean"):
             self.delete_data()
             return
         if options.get("sample"):
             self.sample_data()
             return
+        folder = options.get("folder")
+        if not folder:
+            print("Please specify a folder name")
+            return
         chunk_size = options.get("chunksize")
         self.process_csv_files(folder, chunk_size)