Update: Change input forms

action_flow/fase1_extract.py

@@ -1,80 +1,61 @@
-#!/usr/bin/env python
-# coding: utf-8
-
-# # Imports
-
-# In[29]:
-
-
 import pandas as pd
 import os
 import sys
+from extraccion import agentes_entidades
 
-
-# In[30]:
-
-
-module_path = os.path.abspath(os.path.join('..'))
+module_path = os.path.abspath(os.path.join(".."))
 
 # Add to sys.path if not already present
 if module_path not in sys.path:
     sys.path.append(module_path)
     print(module_path)
 
-from extraccion import agentes_entidades
-
-
-# # Variables
-
-# In[31]:
-
 
 INPUT_FOLDER = f"{module_path}/input/Preguntas Categoricas/"
 OUTPUT_FOLDER = f"{module_path}/output/fase1"
 FILES_TO_PROCESS = os.listdir(INPUT_FOLDER)
 DELIMITER = "|^"
 DIC_QUESTIONS = {
-    "Encuesta_MediaG01Q02.csv":agentes_entidades.extractor_pre_1
+    "Encuesta_MediaG01Q02.csv": agentes_entidades.extractor_pre_1,
+    "Encuesta_MediaG01Q04.csv": agentes_entidades.extractor_pre_2,
+    "Encuesta_MediaG01Q10.csv": agentes_entidades.extractor_pre_3,
+    "Encuesta_MediaG03Q17.csv": agentes_entidades.extractor_pre_4,
+    "Encuesta_MediaG03Q18.csv": agentes_entidades.extractor_pre_5,
+    # "Encuesta_MediaG03Q19.csv": agentes_entidades.extractor_pre_6,
+    "Encuesta_MediaG04Q22.csv": agentes_entidades.extractor_pre_7,
+    "Encuesta_MediaG04Q23.csv": agentes_entidades.extractor_pre_8,
+    "Encuesta_MediaG04Q25.csv": agentes_entidades.extractor_pre_9,
+    "Encuesta_MediaG05Q30.csv": agentes_entidades.extractor_pre_10,
+    "Encuesta_MediaG05Q34.csv": agentes_entidades.extractor_pre_11,
+    # "Encuesta_MediaG05Q27.csv": agentes_entidades.extractor_pre_12,
+    "Encuesta_MediaG06Q35.csv": agentes_entidades.extractor_pre_13,
+    # "Encuesta_MediaG06Q37.csv": agentes_entidades.extractor_pre_14,
+    # "Encuesta_MediaG06Q39.csv": agentes_entidades.extractor_pre_15
     # COMPLETAR RESTO
 }
 
 
-# # Functions
-
-# In[32]:
-
-
 def extract_answers(answers):
     answer_formated = ""
     iterator_answers = answers.acciones
     for item in iterator_answers:
-        answer_formated+=f"{item.accion}{DELIMITER}"
+        answer_formated += f"{item.accion}{DELIMITER}"
     return answer_formated
 
 
-# In[33]:
-
-
-def format_answer(dataframe,function):
+def format_answer(dataframe, function):
     dataframe["respuestas_formato"] = None
     for index, row in dataframe.iterrows():
-        answers = function (row['respuesta'])
+        answers = function(row["respuesta"])
         answer_to_insert = extract_answers(answers)
-        dataframe.loc[index,'respuestas_formato'] = answer_to_insert
-
-
-# In[34]:
+        dataframe.loc[index, "respuestas_formato"] = answer_to_insert
 
 
 def format_all_answers(Dic_questions):
-    for key,value in Dic_questions.items():
-        question_dataframe = pd.read_csv(INPUT_FOLDER+"/"+key)
+    for key, value in Dic_questions.items():
+        question_dataframe = pd.read_csv(INPUT_FOLDER + "/" + key)
         format_answer(question_dataframe, value)
-        question_dataframe.to_csv(OUTPUT_FOLDER+"/"+key)
-
-
-# In[35]:
+        question_dataframe.to_csv(OUTPUT_FOLDER + "/" + key)
 
 
 format_all_answers(DIC_QUESTIONS)
-