Update 07-08-23. Update Generation.py

dags/components/Generation.py

@@ -137,13 +137,10 @@ def get_generate_from_xcom(**kwargs):
     tasks_with_save = []
     definition = task.xcom_pull(task_ids="SCRIPTS-EXTRACTOR", key="EXTRACTION-DEFINITION-JSON")
 
-    #LOGICA - TABLAS QUE SI TIENEN EL PARAMETRO SAVE_OUTPUT
-    for i in range(len(definition)):
-        if definition[i]["save_output"] == True:
-            tasks_with_save.append(definition[i]["identifier"])
-
-    #print(tasks_with_save)
-
+    # LOGICA - TABLAS QUE SI TIENEN EL PARAMETRO SAVE_OUTPUT
+    for item in definition:
+        if "save_output" in item.keys() and item["save_output"]:
+            tasks_with_save.append(item["identifier"])
     success_tasks = tasks["tasks"]
     success_tasks = [item[1] for item in success_tasks]
     logger.info(f"GENERADORES QUE FUERON EXITOSOS (TABLAS): {success_tasks}")
@@ -153,14 +150,11 @@ def get_generate_from_xcom(**kwargs):
         if not key.startswith(OperationTypeEnum.SELECT.value) and not key.startswith(OperationTypeEnum.PROCEDURE.value):
             continue
         xcom_outputs = task.xcom_pull(task_ids="SCRIPTS-EXTRACTOR", key=key)
-
         logger.info(f"Trayendo tablas {xcom_outputs}")
         for select in xcom_outputs:
             tablename = select_multiple(select)["tablename"]
-            #print("IMPRIMIR", tablename)
             if (tasks["reset"] or tasks["status"] == ProcessStatusEnum.SUCCESS.value or select not in success_tasks) and tablename in tasks_with_save:
                 final_outputs.append(select)
-    #print(final_outputs)
     logger.info(f"Final outputs: {final_outputs}")
     Variable.set(key='GENERATES', value=final_outputs, serialize_json=True)
     if len(final_outputs) > 0: