Refactor: add constant for magic auteur value

betagouv · Apr 21, 2020 · c3562f6 · c3562f6
1 parent 17532af
commit c3562f6
Show file tree

Hide file tree

Showing 5 changed files with 9 additions and 5 deletions.
diff --git a/repondeur/zam_repondeur/models/amendement.py b/repondeur/zam_repondeur/models/amendement.py
@@ -180,7 +180,11 @@ class AmendementLocation(Base):
 
 
 class Amendement(Base):
+
+    AUTEUR_GOUVERNEMENT = "LE GOUVERNEMENT"
+
     VERY_BIG_NUMBER = 999_999_999
+
     __tablename__ = "amendements"
     __table_args__ = (
         Index("ix_amendements__lecture_pk", "lecture_pk"),
@@ -465,7 +469,7 @@ def parse_num(text: str) -> Tuple[int, int]:
 
     @reify
     def gouvernemental(self) -> bool:
-        return self.auteur == "LE GOUVERNEMENT"
+        return self.auteur == Amendement.AUTEUR_GOUVERNEMENT
 
     @reify
     def is_withdrawn(self) -> bool:

diff --git a/repondeur/zam_repondeur/scripts/enfoncer_colonnes.py b/repondeur/zam_repondeur/scripts/enfoncer_colonnes.py
@@ -125,7 +125,7 @@ def extract_data_from_csv_file(
 
                 # Extraction de l'auteur
                 if mo.group(2) == "Gouvernement":
-                    auteur = "LE GOUVERNEMENT"
+                    auteur = Amendement.AUTEUR_GOUVERNEMENT
                 else:
                     groupe = mo.group(2)
 

diff --git a/repondeur/zam_repondeur/services/fetch/an/amendements.py b/repondeur/zam_repondeur/services/fetch/an/amendements.py
@@ -758,7 +758,7 @@ def get_auteur(self) -> str:
         if self.raw_auteur is None:
             return "Non trouvé"
         if self.raw_auteur.get("estGouvernement", "") == "1":
-            return "LE GOUVERNEMENT"
+            return Amendement.AUTEUR_GOUVERNEMENT
         nom = self.raw_auteur.get("nom", "")
         prenom = self.raw_auteur.get("prenom", "")
         return f"{nom} {prenom}"

diff --git a/repondeur/zam_repondeur/services/import_export/liasse_xml.py b/repondeur/zam_repondeur/services/import_export/liasse_xml.py
@@ -110,7 +110,7 @@ def _make_amendement(
     is_gouvernemental = extract("signataires", "auteur", "typeAuteur") == "Gouvernement"
 
     if is_gouvernemental:
-        auteur_name = "LE GOUVERNEMENT"
+        auteur_name = Amendement.AUTEUR_GOUVERNEMENT
         groupe_name = None
         matricule = None
     else:

diff --git a/repondeur/zam_repondeur/visam/views/amendements_add.py b/repondeur/zam_repondeur/visam/views/amendements_add.py
@@ -63,7 +63,7 @@ def post(self) -> Response:
             .one_or_none()
         )
 
-        auteur = "LE GOUVERNEMENT" if organisation.is_gouvernement else ""
+        auteur = Amendement.AUTEUR_GOUVERNEMENT if organisation.is_gouvernement else ""
         groupe = organisation.name
 
         max_position = max(