Spaces:
Running
Running
Delete tools/base
Browse files- tools/base/__init__.py +0 -0
- 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()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|