Skip to content

Commit f243f7f

Browse files
chezouchezou
committed
Fix override signatures
1 parent faa56c4 commit f243f7f

File tree

11 files changed

+107
-21
lines changed

11 files changed

+107
-21
lines changed

‎tdclient/bulk_import_api.py‎

Lines changed: 16 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -30,13 +30,26 @@ class BulkImportAPI:
3030

3131
# Methods from API class
3232
def get(
33-
self, url: str, params: dict[str, Any] | None = None
33+
self,
34+
path: str,
35+
params: dict[str, Any] | None = None,
36+
headers: dict[str, str] | None = None,
37+
**kwargs: Any,
3438
) -> AbstractContextManager[urllib3.BaseHTTPResponse]: ...
3539
def post(
36-
self, url: str, params: dict[str, Any] | None = None
40+
self,
41+
path: str,
42+
params: dict[str, Any] | None = None,
43+
headers: dict[str, str] | None = None,
44+
**kwargs: Any,
3745
) -> AbstractContextManager[urllib3.BaseHTTPResponse]: ...
3846
def put(
39-
self, url: str, stream: BytesOrStream, size: int
47+
self,
48+
path: str,
49+
bytes_or_stream: BytesOrStream,
50+
size: int,
51+
headers: dict[str, str] | None = None,
52+
**kwargs: Any,
4053
) -> AbstractContextManager[urllib3.BaseHTTPResponse]: ...
4154
def raise_error(
4255
self, msg: str, res: urllib3.BaseHTTPResponse, body: bytes | str

‎tdclient/connector_api.py‎

Lines changed: 19 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010

1111
import urllib3
1212

13+
from tdclient.types import BytesOrStream
1314
from tdclient.util import create_url, normalize_connector_config
1415

1516

@@ -21,15 +22,30 @@ class ConnectorAPI:
2122

2223
# Methods from API class
2324
def get(
24-
self, url: str, params: dict[str, Any] | None = None
25+
self,
26+
path: str,
27+
params: dict[str, Any] | None = None,
28+
headers: dict[str, str] | None = None,
29+
**kwargs: Any,
2530
) -> AbstractContextManager[urllib3.BaseHTTPResponse]: ...
2631
def post(
2732
self, url: str, params: Any, headers: dict[str, str] | None = None
2833
) -> AbstractContextManager[urllib3.BaseHTTPResponse]: ...
2934
def put(
30-
self, url: str, params: Any, size: int, headers: dict[str, str] | None = None
35+
self,
36+
path: str,
37+
bytes_or_stream: BytesOrStream,
38+
size: int,
39+
headers: dict[str, str] | None = None,
40+
**kwargs: Any,
41+
) -> AbstractContextManager[urllib3.BaseHTTPResponse]: ...
42+
def delete(
43+
self,
44+
path: str,
45+
params: dict[str, Any] | None = None,
46+
headers: dict[str, str] | None = None,
47+
**kwargs: Any,
3148
) -> AbstractContextManager[urllib3.BaseHTTPResponse]: ...
32-
def delete(self, url: str) -> AbstractContextManager[urllib3.BaseHTTPResponse]: ...
3349
def raise_error(
3450
self, msg: str, res: urllib3.BaseHTTPResponse, body: bytes
3551
) -> None: ...

‎tdclient/database_api.py‎

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,10 +20,18 @@ class DatabaseAPI:
2020

2121
# Methods from API class
2222
def get(
23-
self, url: str, params: dict[str, Any] | None = None
23+
self,
24+
path: str,
25+
params: dict[str, Any] | None = None,
26+
headers: dict[str, str] | None = None,
27+
**kwargs: Any,
2428
) -> AbstractContextManager[urllib3.BaseHTTPResponse]: ...
2529
def post(
26-
self, url: str, params: dict[str, Any] | None = None
30+
self,
31+
path: str,
32+
params: dict[str, Any] | None = None,
33+
headers: dict[str, str] | None = None,
34+
**kwargs: Any,
2735
) -> AbstractContextManager[urllib3.BaseHTTPResponse]: ...
2836
def raise_error(
2937
self, msg: str, res: urllib3.BaseHTTPResponse, body: bytes

‎tdclient/export_api.py‎

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,10 +21,14 @@ class ExportAPI:
2121

2222
# Methods from API class
2323
def post(
24-
self, url: str, params: dict[str, Any] | None = None
24+
self,
25+
path: str,
26+
params: dict[str, Any] | None = None,
27+
headers: dict[str, str] | None = None,
28+
**kwargs: Any,
2529
) -> AbstractContextManager[urllib3.BaseHTTPResponse]: ...
2630
def raise_error(
27-
self, msg: str, res: urllib3.BaseHTTPResponse, body: bytes
31+
self, msg: str, res: urllib3.BaseHTTPResponse, body: bytes | str
2832
) -> None: ...
2933
def checked_json(self, body: bytes, required: list[str]) -> dict[str, Any]: ...
3034

‎tdclient/import_api.py‎

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,12 @@ class ImportAPI:
2424

2525
# Methods from API class
2626
def put(
27-
self, url: str, stream: BytesOrStream, size: int, **kwargs: Any
27+
self,
28+
path: str,
29+
bytes_or_stream: BytesOrStream,
30+
size: int,
31+
headers: dict[str, str] | None = None,
32+
**kwargs: Any,
2833
) -> AbstractContextManager[urllib3.BaseHTTPResponse]: ...
2934
def raise_error(
3035
self, msg: str, res: urllib3.BaseHTTPResponse, body: bytes

‎tdclient/job_api.py‎

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,11 @@ def get(
4040
headers: dict[str, str] | None = None,
4141
) -> AbstractContextManager[urllib3.BaseHTTPResponse]: ...
4242
def post(
43-
self, url: str, params: dict[str, Any] | None = None
43+
self,
44+
path: str,
45+
params: dict[str, Any] | None = None,
46+
headers: dict[str, str] | None = None,
47+
**kwargs: Any,
4448
) -> AbstractContextManager[urllib3.BaseHTTPResponse]: ...
4549
def raise_error(
4650
self, msg: str, res: urllib3.BaseHTTPResponse, body: bytes | str

‎tdclient/result_api.py‎

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,10 +21,18 @@ class ResultAPI:
2121

2222
# Methods from API class
2323
def get(
24-
self, url: str, params: dict[str, Any] | None = None
24+
self,
25+
path: str,
26+
params: dict[str, Any] | None = None,
27+
headers: dict[str, str] | None = None,
28+
**kwargs: Any,
2529
) -> AbstractContextManager[urllib3.BaseHTTPResponse]: ...
2630
def post(
27-
self, url: str, params: dict[str, Any] | None = None
31+
self,
32+
path: str,
33+
params: dict[str, Any] | None = None,
34+
headers: dict[str, str] | None = None,
35+
**kwargs: Any,
2836
) -> AbstractContextManager[urllib3.BaseHTTPResponse]: ...
2937
def raise_error(
3038
self, msg: str, res: urllib3.BaseHTTPResponse, body: bytes

‎tdclient/schedule_api.py‎

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,10 +22,18 @@ class ScheduleAPI:
2222

2323
# Methods from API class
2424
def get(
25-
self, url: str, params: dict[str, Any] | None = None
25+
self,
26+
path: str,
27+
params: dict[str, Any] | None = None,
28+
headers: dict[str, str] | None = None,
29+
**kwargs: Any,
2630
) -> AbstractContextManager[urllib3.BaseHTTPResponse]: ...
2731
def post(
28-
self, url: str, params: dict[str, Any] | None = None
32+
self,
33+
path: str,
34+
params: dict[str, Any] | None = None,
35+
headers: dict[str, str] | None = None,
36+
**kwargs: Any,
2937
) -> AbstractContextManager[urllib3.BaseHTTPResponse]: ...
3038
def raise_error(
3139
self, msg: str, res: urllib3.BaseHTTPResponse, body: bytes

‎tdclient/server_status_api.py‎

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,11 @@ class ServerStatusAPI:
1818

1919
# Methods from API class
2020
def get(
21-
self, url: str, params: dict[str, Any] | None = None
21+
self,
22+
path: str,
23+
params: dict[str, Any] | None = None,
24+
headers: dict[str, str] | None = None,
25+
**kwargs: Any,
2226
) -> AbstractContextManager[urllib3.BaseHTTPResponse]: ...
2327
def checked_json(self, body: bytes, required: list[str]) -> dict[str, Any]: ...
2428

‎tdclient/table_api.py‎

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,10 +23,18 @@ class TableAPI:
2323

2424
# Methods from API class
2525
def get(
26-
self, url: str, params: dict[str, Any] | None = None
26+
self,
27+
path: str,
28+
params: dict[str, Any] | None = None,
29+
headers: dict[str, str] | None = None,
30+
**kwargs: Any,
2731
) -> AbstractContextManager[urllib3.BaseHTTPResponse]: ...
2832
def post(
29-
self, url: str, params: dict[str, Any] | None = None
33+
self,
34+
path: str,
35+
params: dict[str, Any] | None = None,
36+
headers: dict[str, str] | None = None,
37+
**kwargs: Any,
3038
) -> AbstractContextManager[urllib3.BaseHTTPResponse]: ...
3139
def raise_error(
3240
self, msg: str, res: urllib3.BaseHTTPResponse, body: bytes | str

0 commit comments

Comments
 (0)