refait une tentative API après 1s en cas de dépassement du quota auto…

…risé
betagouv · May 24, 2024 · 1a871ce · 1a871ce
1 parent 668f3d0
commit 1a871ce
Showing 1 changed file with 15 additions and 5 deletions.
diff --git a/impact/entreprises/management/commands/force_code_naf.py b/impact/entreprises/management/commands/force_code_naf.py
@@ -1,3 +1,5 @@
+import time
+
 from django.core.management.base import BaseCommand
 
 import api.exceptions
@@ -14,13 +16,21 @@ def handle(self, *args, **options):
                 )
             else:
                 try:
-                    infos_entreprise = api.recherche_entreprises.recherche(
-                        entreprise.siren
-                    )
-                    entreprise.code_NAF = infos_entreprise["code_NAF"]
-                    entreprise.save()
+                    maj(entreprise)
                     print(
                         f"OK: {entreprise.siren} {entreprise.denomination} {entreprise.code_NAF}"
                     )
+                except api.exceptions.TooManyRequestError as e:
+                    time.sleep(1)
+                    maj(entreprise)
+                    print(
+                        f"OK (2): {entreprise.siren} {entreprise.denomination} {entreprise.code_NAF}"
+                    )
                 except api.exceptions.APIError as e:
                     print(f"ERREUR {e}: {entreprise.siren}")
+
+
+def maj(entreprise):
+    infos_entreprise = api.recherche_entreprises.recherche(entreprise.siren)
+    entreprise.code_NAF = infos_entreprise["code_NAF"]
+    entreprise.save()