1+ import logging
12from _typeshed import Incomplete
23from collections .abc import Iterable
34from enum import Enum
5+ from types import TracebackType
46from typing import Any
5- from typing_extensions import TypeAlias
7+ from typing_extensions import Self , TypeAlias
68
79from influxdb_client .client ._base import _BaseWriteApi
810from 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
1719class WriteType (Enum ):
1820 batching : int
1921 asynchronous : int
2022 synchronous : int
2123
2224class 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