Create path to save file

prefeitura-rio · Nov 19, 2023 · 57d9e11 · 57d9e11
1 parent 991bfa6
commit 57d9e11
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pipelines/rj_sms/whatsapp/clinica_scheduled_patients/tasks.py b/pipelines/rj_sms/whatsapp/clinica_scheduled_patients/tasks.py
@@ -1,4 +1,5 @@
 # -*- coding: utf-8 -*-
+import os
 import csv
 import shutil
 import pandas as pd
@@ -22,6 +23,9 @@ def get_patients():
 
 @task
 def save_patients(dataframe):
+    path = 'pipelines/rj_sms/whatsapp/clinica_scheduled_patients/data'
+    if not os.path.exists(path):
+        os.mkdir(path)
     data_futura = datetime.today() + timedelta(days=3)
     data_formatada = data_futura.strftime("%Y-%m-%d")
     filename = f"pipelines/rj_sms/whatsapp/clinica_scheduled_patients/data/{data_formatada}.csv"