Benjamin_Loison/waybackpy
1
0
Fork 0
Code Issues 1 Pull Requests Packages Projects Releases Wiki Activity

fix: Retry (ref: https://github.com/python/typeshed/issues/6893)

Browse Source
This commit is contained in:
eggplants
2022-02-04 05:00:36 +09:00
parent e0fb863761
commit 639159e304
2 changed files with 4 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
waybackpy/cdx_utils.py
View File

@@ -4,9 +4,7 @@ from urllib.parse import quote
import requests import requests
from requests.adapters import HTTPAdapter from requests.adapters import HTTPAdapter
from urllib3.util.retry import Retry
# from urllib3.util.retry import Retry
from requests.packages.urllib3.util.retry import Retry
from .exceptions import WaybackError from .exceptions import WaybackError
from .utils import DEFAULT_USER_AGENT from .utils import DEFAULT_USER_AGENT

8
waybackpy/save_api.py
View File

@@ -5,9 +5,7 @@ from typing import Dict, Optional
import requests import requests
from requests.adapters import HTTPAdapter from requests.adapters import HTTPAdapter
from urllib3.util.retry import Retry
# from urllib3.util.retry import Retry
from requests.packages.urllib3.util.retry import Retry
from .exceptions import MaximumSaveRetriesExceeded from .exceptions import MaximumSaveRetriesExceeded
from .utils import DEFAULT_USER_AGENT from .utils import DEFAULT_USER_AGENT
@@ -65,12 +63,12 @@ class WaybackMachineSaveAPI(object):
the response URL yourself in the browser. the response URL yourself in the browser.
""" """
session = requests.Session() session = requests.Session()
retries_ = Retry( retries = Retry(
total=self.total_save_retries, total=self.total_save_retries,
backoff_factor=self.backoff_factor, backoff_factor=self.backoff_factor,
status_forcelist=self.status_forcelist, status_forcelist=self.status_forcelist,
) )
session.mount("https://", HTTPAdapter(max_retries=retries_)) session.mount("https://", HTTPAdapter(max_retries=retries))
self.response = session.get(self.request_url, headers=self.request_headers) self.response = session.get(self.request_url, headers=self.request_headers)
# requests.response.headers is requests.structures.CaseInsensitiveDict # requests.response.headers is requests.structures.CaseInsensitiveDict
self.headers = self.response.headers self.headers = self.response.headers