encuestado handling with Encuestado class

1 files changed, 18 insertions, 20 deletions

diff --git a/CeibalEncuesta/gtk2/CeibalEncuesta/CeibalEncuesta.py b/CeibalEncuesta/gtk2/CeibalEncuesta/CeibalEncuesta.py
index 514da50..8f898c7 100644
--- a/CeibalEncuesta/gtk2/CeibalEncuesta/CeibalEncuesta.py
+++ b/CeibalEncuesta/gtk2/CeibalEncuesta/CeibalEncuesta.py
@@ -281,21 +281,18 @@ class CeibalEncuesta(gtk.Window):
 
         encuestado = user.strip()
 
-        # Entrada para encuestado
-        if not encuestado in self.out_dict.keys():
-            self.out_dict[encuestado] = {}
-
         # Entrada para grupo
-        if not self.out_dict[encuestado].get(indice_grupo, False):
-            self.out_dict[encuestado][indice_grupo] = {}
+        encuestado_id = Encuestado(encuestado).get_id()
+        if not self.out_dict[encuestado_id].get(indice_grupo, False):
+            self.out_dict[encuestado_id][indice_grupo] = {}
 
         # Entrada para pregunta
-        if not self.out_dict[encuestado][indice_grupo].get("fields", False):
-            self.out_dict[encuestado][indice_grupo]["fields"] = {}
+        if not self.out_dict[encuestado_id][indice_grupo].get("fields", False):
+            self.out_dict[encuestado_id][indice_grupo]["fields"] = {}
 
-        self.out_dict[encuestado][indice_grupo]["name"] = grupo_name
+        self.out_dict[encuestado_id][indice_grupo]["name"] = grupo_name
 
-        self.out_dict[encuestado][indice_grupo]["fields"][indice_pregunta] = dict_pregunta
+        self.out_dict[encuestado_id][indice_grupo]["fields"][indice_pregunta] = dict_pregunta
 
     def save_text(self):
 
@@ -315,21 +312,22 @@ class CeibalEncuesta(gtk.Window):
 
         encuestado = user.strip()
 
+        encuestado_id = Encuestado(encuestado).get_id()
         # Entrada para encuestado
-        if not encuestado in self.out_dict.keys():
-            self.out_dict[encuestado] = {}
+        if not encuestado_id in self.out_dict.keys():
+            self.out_dict[encuestado_id] = {}
 
         # Entrada para grupo
-        if not self.out_dict[encuestado].get(indice_grupo, False):
-            self.out_dict[encuestado][indice_grupo] = {}
+        if not self.out_dict[encuestado_id].get(indice_grupo, False):
+            self.out_dict[encuestado_id][indice_grupo] = {}
 
         # Entrada para pregunta
-        if not self.out_dict[encuestado][indice_grupo].get("fields", False):
-            self.out_dict[encuestado][indice_grupo]["fields"] = {}
+        if not self.out_dict[encuestado_id][indice_grupo].get("fields", False):
+            self.out_dict[encuestado_id][indice_grupo]["fields"] = {}
 
-        self.out_dict[encuestado][indice_grupo]["name"] = grupo_name
+        self.out_dict[encuestado_id][indice_grupo]["name"] = grupo_name
 
-        self.out_dict[encuestado][indice_grupo]["fields"][indice_pregunta] = dict_pregunta
+        self.out_dict[encuestado_id][indice_grupo]["fields"][indice_pregunta] = dict_pregunta
 
         self.__save_json()
 
@@ -361,14 +359,14 @@ class CeibalEncuesta(gtk.Window):
             self.panel.update(self.out_dict[encuestado_id])
 
         else:
-            self.out_dict[encuestado_str] = {}
+            self.out_dict[encuestado_id] = {}
             self.panel.update({})
 
         self.panel.toolbar_encuesta.encuesta_actual = self.encuesta_actual
 
         self.menu_bar.hide()
         self.infowidget.hide()
-        self.infowidget.set_encuestado(encuestado_str)
+        self.infowidget.set_encuestado(encuestado_id)
 
         combo_box = getattr(self.panel.lista, "combo", None)
         if combo_box: