[Jobs][AN] Amendements : ajout des auteurs et cosignataires

parlapi/jobs/an_amendements.py

@@ -2,7 +2,8 @@
 
 from .base import BaseANJob
 from .utils import ijson_items
-from ..models import Acteur, Amendement, Document, Legislature, Organe
+from ..models import Acteur, Amendement, ActeurAmendement,\
+                    Document, Legislature, Organe
 
 
 class ImportAmendementsJob(BaseANJob):
@@ -76,6 +77,31 @@ class ImportAmendementsJob(BaseANJob):
             data['amendement_parent'] = \
                 self.save(Amendement, json['amendementParent'])
 
+        acteurs = []
+
+        auteur = json['signataires']['auteur']
+        if 'acteurRef' in auteur:
+            aa = ActeurAmendement(relation=u'auteur')
+            aa.acteur_id = auteur['acteurRef']
+            acteurs.append(aa)
+
+        cosignataires = json['signataires']['cosignataires']
+        if cosignataires:
+            if not 'acteurRef' in cosignataires:
+                cosignataires = []
+            else:
+                cosignataires = cosignataires['acteurRef']
+
+            if isinstance(cosignataires, basestring):
+                cosignataires = [cosignataires]
+
+            for cosignataire in cosignataires:
+                acs = ActeurAmendement(relation=u'cosignataire')
+                acs.acteur_id = cosignataire
+                acteurs.append(acs)
+
+        data['acteurs'] = acteurs
+
         so = json['sort']
         if so:
             data['sort'] = so['sortEnSeance']