Bläddra i källkod

added http retries

vnc-console 10 månader sedan
förälder
incheckning
5e1f91b09c
1 ändrade filer med 7 tillägg och 0 borttagningar
  1. 7 0
      anafapi/anafapi.py

+ 7 - 0
anafapi/anafapi.py

@@ -11,6 +11,9 @@ from collections import OrderedDict
 
 import requests
 import xlsxwriter as xlsw
+from urllib3.util.retry import Retry
+from requests.adapters import HTTPAdapter
+
 from .writer import Writer
 
 def insert_keys(dict_, obj, pos):
@@ -35,6 +38,10 @@ class anafAPI(object):
         '''Constructor
         '''
         self.session = requests.Session()
+        retry = Retry(connect=3, backoff_factor=1)
+        adapter = HTTPAdapter(max_retries=retry)
+        self.session.mount('http://', adapter)
+        self.session.mount('https://', adapter)
 
     def parse_list(self, file_name=None, list_=None, excl_=[], ):
         if file_name: