Skip to content

Commit 13325d4

Browse files
authored
Fix stubtest for influxdb-client (#9808)
Fixes #9805
1 parent 4ca5ee9 commit 13325d4

File tree

1 file changed

+27
-21
lines changed

1 file changed

+27
-21
lines changed

stubs/influxdb-client/influxdb_client/client/write_api.pyi

Lines changed: 27 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,10 @@
1+
import logging
12
from _typeshed import Incomplete
23
from collections.abc import Iterable
34
from enum import Enum
5+
from types import TracebackType
46
from typing import Any
5-
from typing_extensions import TypeAlias
7+
from typing_extensions import Self, TypeAlias
68

79
from influxdb_client.client._base import _BaseWriteApi
810
from influxdb_client.client.write.point import Point
@@ -12,35 +14,37 @@ _DataClass: TypeAlias = Any # any dataclass
1214
_NamedTuple: TypeAlias = tuple[Any, ...] # any NamedTuple
1315
_Observable: TypeAlias = Any # reactivex.Observable
1416

15-
logger: Incomplete
17+
logger: logging.Logger
1618

1719
class WriteType(Enum):
1820
batching: int
1921
asynchronous: int
2022
synchronous: int
2123

2224
class WriteOptions:
23-
write_type: Incomplete
24-
batch_size: Incomplete
25-
flush_interval: Incomplete
26-
jitter_interval: Incomplete
27-
retry_interval: Incomplete
28-
max_retries: Incomplete
29-
max_retry_delay: Incomplete
30-
max_retry_time: Incomplete
31-
exponential_base: Incomplete
25+
write_type: WriteType
26+
batch_size: int
27+
flush_interval: int
28+
jitter_interval: int
29+
retry_interval: int
30+
max_retries: int
31+
max_retry_delay: int
32+
max_retry_time: int
33+
exponential_base: int
3234
write_scheduler: Incomplete
35+
max_close_wait: int
3336
def __init__(
3437
self,
3538
write_type: WriteType = ...,
36-
batch_size: int = ...,
37-
flush_interval: int = ...,
38-
jitter_interval: int = ...,
39-
retry_interval: int = ...,
40-
max_retries: int = ...,
41-
max_retry_delay: int = ...,
42-
max_retry_time: int = ...,
43-
exponential_base: int = ...,
39+
batch_size: int = 1_000,
40+
flush_interval: int = 1_000,
41+
jitter_interval: int = 0,
42+
retry_interval: int = 5_000,
43+
max_retries: int = 5,
44+
max_retry_delay: int = 125_000,
45+
max_retry_time: int = 180_000,
46+
exponential_base: int = 2,
47+
max_close_wait: int = 300_000,
4448
write_scheduler=...,
4549
) -> None: ...
4650
def to_retry_strategy(self, **kwargs): ...
@@ -99,6 +103,8 @@ class WriteApi(_BaseWriteApi):
99103
) -> Any: ...
100104
def flush(self) -> None: ...
101105
def close(self) -> None: ...
102-
def __enter__(self): ...
103-
def __exit__(self, exc_type, exc_val, exc_tb) -> None: ...
106+
def __enter__(self) -> Self: ...
107+
def __exit__(
108+
self, exc_type: type[BaseException] | None, exc_val: BaseException | None, exc_tb: TracebackType | None
109+
) -> None: ...
104110
def __del__(self) -> None: ...

0 commit comments

Comments
 (0)