Create output directory if required

6 years ago · 769a5d4275
parent af6f51f331
commit 769a5d4275
1 changed files with 6 additions and 0 deletions
--- a/survey_tools.py
+++ b/survey_tools.py
@ -162,5 +162,11 @@ def update_survey_output(df, output_dir):
        current_survey_col = current_profile.columns[-1]
        master[current_survey_col] = current_profile[current_survey_col].values

+        # Prepare output directory
+        try:
+            os.makedirs(output_dir)
+        except FileExistsError:
+            pass
+
        # Export updated results
        master.to_csv(csv_name, index=False)