fixed error that caused test failure

proyectos_de_ley/stats/management/commands/create_stats.py

@@ -113,37 +113,41 @@ def get_with_dictamen_but_not_voted(self):
 
     def update_iniciativas_agrupadas_with_title_of_law(self):
         projects_with_law = Proyecto.objects.all().exclude(
-            titulo_de_ley='').values('codigo', 'titulo_de_ley', 'iniciativas_agrupadas')
-
-        iniciativas_con_ley = self.get_iniciativas_con_ley(projects_with_law)
-        project_code_as_dict = self.get_project_code_as_dict(projects_with_law)
-
-        for iniciativa in iniciativas_con_ley:
-            try:
-                p = Proyecto.objects.get(codigo=iniciativa)
-            except Proyecto.DoesNotExist:
-                continue
-            if p.titulo_de_ley == '':
-                p.titulo_de_ley = project_code_as_dict[iniciativa]
-                p.save()
+            titulo_de_ley='').exclude(
+            titulo_de_ley__isnull=True).values('codigo', 'titulo_de_ley', 'iniciativas_agrupadas')
+
+        iniciativas_deben_tener_ley = self.get_iniciativas_con_ley(projects_with_law)
+
+        for iniciativa in iniciativas_deben_tener_ley.items():
+            for i in iniciativa[1]['iniciativas']:
+                if i not in iniciativas_deben_tener_ley.keys():
+                    try:
+                        p = Proyecto.objects.get(codigo=i)
+                    except Proyecto.DoesNotExist:
+                        continue
+                    if p.titulo_de_ley == '':
+                        p.titulo_de_ley = iniciativa[1]['titulo_de_ley']
+                        p.save()
 
     def get_iniciativas_con_ley(self, projects_with_law):
-        iniciativas_con_ley = []
+        iniciativas_con_ley = {}
         for i in projects_with_law:
             iniciativas = i['iniciativas_agrupadas']
             if iniciativas != '' and iniciativas is not None:
                 iniciativas = iniciativas.replace('{', '')
                 iniciativas = iniciativas.replace('}', '')
                 iniciativas = iniciativas.split(',')
-                iniciativas_con_ley += iniciativas
-        return list(set(iniciativas_con_ley))
 
-    def get_project_code_as_dict(self, projects_with_law):
-        my_dict = {}
-        for i in projects_with_law:
-            code = i['codigo']
-            my_dict[code] = i['titulo_de_ley']
-        return my_dict
+                if i['codigo'] not in iniciativas_con_ley:
+                    iniciativas_con_ley[i['codigo']] = {
+                        'iniciativas': [],
+                        'titulo_de_ley': '',
+                    }
+
+                iniciativas_con_ley[i['codigo']]['iniciativas'] += iniciativas
+                iniciativas_con_ley[i['codigo']]['titulo_de_ley'] = i['titulo_de_ley']
+
+        return iniciativas_con_ley
 
     def get_proyect_ids(self, queryset):
         proyect_ids = set()