BugFix: emphasis in departamento key from csv files

3 files changed, 19 insertions, 5 deletions

diff --git a/CeibalEncuesta/gtk2/CeibalEncuesta/CeibalEncuesta.py b/CeibalEncuesta/gtk2/CeibalEncuesta/CeibalEncuesta.py
index b0595fe..94c2c0c 100644
--- a/CeibalEncuesta/gtk2/CeibalEncuesta/CeibalEncuesta.py
+++ b/CeibalEncuesta/gtk2/CeibalEncuesta/CeibalEncuesta.py
@@ -340,7 +340,8 @@ class CeibalEncuesta(gtk.Window):
 
         grupo = GrupoDeEncuestados()
         encuestado_str = encuestado
-        tipo = Encuestado(encuestado_str).get_tipo_grupo()
+        encuestado = Encuestado(encuestado_str)
+        tipo = encuestado.get_tipo_grupo()
         grupo.set_tipo(int(tipo))
         encuesta_actual = Encuesta()
         encuesta_actual.set_grupo(grupo)
@@ -355,18 +356,19 @@ class CeibalEncuesta(gtk.Window):
         self.out_dict = {}
         self.out_dict = enc["groups"]
 
-        if self.out_dict.get(encuestado, False):
-            self.panel.update(self.out_dict[encuestado])
+        encuestado_id = encuestado.get_id()
+        if self.out_dict.get(encuestado_id, False):
+            self.panel.update(self.out_dict[encuestado_id])
 
         else:
-            self.out_dict[encuestado] = {}
+            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)
+        self.infowidget.set_encuestado(encuestado_id)
 
         combo_box = getattr(self.panel.lista, "combo", None)
         if combo_box:
diff --git a/CeibalEncuesta/gtk2/CeibalEncuesta/Globales.py b/CeibalEncuesta/gtk2/CeibalEncuesta/Globales.py
index 6fe899e..a679c01 100644
--- a/CeibalEncuesta/gtk2/CeibalEncuesta/Globales.py
+++ b/CeibalEncuesta/gtk2/CeibalEncuesta/Globales.py
@@ -184,6 +184,9 @@ class Encuestado(object):
         tipo_grupo = second.split()[3]
         return int(tipo_grupo)
 
+    def get_id(self):
+        return self.encuestado_str.decode("utf-8")
+
 
 class GrupoDeEncuestados(object):
 
diff --git a/CeibalEncuesta/gtk2/CeibalEncuesta/tests.py b/CeibalEncuesta/gtk2/CeibalEncuesta/tests.py
index a589fe9..8b1c972 100644
--- a/CeibalEncuesta/gtk2/CeibalEncuesta/tests.py
+++ b/CeibalEncuesta/gtk2/CeibalEncuesta/tests.py
@@ -1,3 +1,4 @@
+# -*- coding: utf-8 -*-
 import unittest
 import os
 import shutil
@@ -131,6 +132,14 @@ class EncuestadoTests(unittest.TestCase):
         tipo_grupo = encuestado.get_tipo_grupo()
         self.assertEqual(1, tipo_grupo)
 
+    def test_get_id(self):
+        encuestado_str = "1101236 Río negro A 236 2 A 1 0"
+
+        encuestado = Encuestado(encuestado_str)
+        encuestado_id = encuestado.get_id()
+
+        expected = encuestado_str
+        self.assertEqual(expected, encuestado_id)
 
 if __name__ == '__main__':
     unittest.main()