brainsqueeze commited on
Commit
5d08d86
·
verified ·
1 Parent(s): d59cd77

Delete tools/base

Browse files
Files changed (2) hide show
  1. tools/base/__init__.py +0 -0
  2. tools/base/api_base.py +0 -42
tools/base/__init__.py DELETED
File without changes
tools/base/api_base.py DELETED
@@ -1,42 +0,0 @@
1
- from typing import Dict, Optional, Any
2
-
3
- from urllib3.util.retry import Retry
4
- from requests.adapters import HTTPAdapter
5
- import requests
6
-
7
-
8
- class BaseAPI:
9
-
10
- def __init__(
11
- self,
12
- url: str,
13
- headers: Optional[Dict[str, Any]] = None,
14
- total_retries: int = 3,
15
- backoff_factor: int = 2
16
- ) -> None:
17
- total_retries = max(total_retries, 10)
18
-
19
- adapter = HTTPAdapter(
20
- max_retries=Retry(
21
- total=total_retries,
22
- status_forcelist=[429, 500, 502, 503, 504],
23
- allowed_methods=frozenset({"HEAD", "GET", "POST", "OPTIONS"}),
24
- backoff_factor=backoff_factor,
25
- )
26
- )
27
- self.session = requests.Session()
28
- self.session.mount("https://", adapter)
29
- self.session.mount("http://", adapter)
30
-
31
- self.__url = url
32
- self.__headers = headers
33
-
34
- def get(self, **request_kwargs):
35
- r = self.session.get(url=self.__url, headers=self.__headers, params=request_kwargs, timeout=30)
36
- r.raise_for_status()
37
- return r.json()
38
-
39
- def post(self, payload: Dict[str, Any]):
40
- r = self.session.post(url=self.__url, headers=self.__headers, json=payload, timeout=30)
41
- r.raise_for_status()
42
- return r.json()