move constants in dedicatd file

gip-inclusion · Jan 16, 2024 · 8069f3e · 8069f3e
1 parent f761b77
commit 8069f3e
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/lemarche/tenders/constants.py b/lemarche/tenders/constants.py
@@ -133,3 +133,15 @@
     (SOURCE_API, "API"),
     (SOURCE_TALLY, "TALLY"),
 )
+
+TENDER_SIAE_SOURCE_EMAIL = "EMAIL"
+TENDER_SIAE_SOURCE_DASHBOARD = "DASHBOARD"
+TENDER_SIAE_SOURCE_LINK = "LINK"
+TENDER_SIAE_SOURCE_AI = "AI"
+
+TENDER_SIAE_SOURCE_CHOICES = (
+    (TENDER_SIAE_SOURCE_EMAIL, "E-mail"),
+    (TENDER_SIAE_SOURCE_DASHBOARD, "Dashboard"),
+    (TENDER_SIAE_SOURCE_LINK, "Lien"),
+    (TENDER_SIAE_SOURCE_AI, "IA"),
+)
diff --git a/lemarche/www/tenders/views.py b/lemarche/www/tenders/views.py
@@ -349,7 +349,7 @@ def get(self, request, *args, **kwargs):
                 if not user.has_tender_siae(self.object):
                     for siae in user.siaes.all():
                         TenderSiae.objects.create(
-                            tender=self.object, siae=siae, source=TenderSiae.TENDER_SIAE_SOURCE_LINK
+                            tender=self.object, siae=siae, source=tender_constants.TENDER_SIAE_SOURCE_LINK
                         )
                 # update stats
                 TenderSiae.objects.filter(