diff --git a/stubs/influxdb-client/METADATA.toml b/stubs/influxdb-client/METADATA.toml index 73304a797..a97ad1fa7 100644 --- a/stubs/influxdb-client/METADATA.toml +++ b/stubs/influxdb-client/METADATA.toml @@ -1,4 +1,4 @@ -version = "1.40.*" +version = "1.41.*" upstream_repository = "https://github.com/influxdata/influxdb-client-python" # requires a version of urllib3 with a py.typed file requires = ["urllib3>=2"] diff --git a/stubs/influxdb-client/influxdb_client/client/flux_csv_parser.pyi b/stubs/influxdb-client/influxdb_client/client/flux_csv_parser.pyi index 9e3d15f9b..dada44e53 100644 --- a/stubs/influxdb-client/influxdb_client/client/flux_csv_parser.pyi +++ b/stubs/influxdb-client/influxdb_client/client/flux_csv_parser.pyi @@ -45,6 +45,7 @@ class FluxCsvParser: data_frame_index: list[str] | None = None, query_options: Incomplete | None = None, response_metadata_mode: FluxResponseMetadataMode = ..., + use_extension_dtypes: bool = False, ) -> None: ... def __enter__(self) -> Self: ... def __exit__( diff --git a/stubs/influxdb-client/influxdb_client/client/query_api.pyi b/stubs/influxdb-client/influxdb_client/client/query_api.pyi index 14bfed4c3..d155fdb8f 100644 --- a/stubs/influxdb-client/influxdb_client/client/query_api.pyi +++ b/stubs/influxdb-client/influxdb_client/client/query_api.pyi @@ -1,10 +1,11 @@ -from _typeshed import Incomplete +from _typeshed import Incomplete, SupportsItems from collections.abc import Callable, Generator from typing import Any from influxdb_client import Dialect from influxdb_client.client._base import _BaseQueryApi from influxdb_client.client.flux_table import CSVIterator, FluxRecord, TableList +from influxdb_client.domain.organization import Organization class QueryOptions: profilers: Incomplete @@ -20,29 +21,31 @@ class QueryApi(_BaseQueryApi): query: str, org: Incomplete | None = None, dialect: Dialect = ..., - params: dict[Incomplete, Incomplete] | None = None, + params: SupportsItems[str, Incomplete] | None = None, ) -> CSVIterator: ... def query_raw( - self, query: str, org: Incomplete | None = None, dialect=..., params: dict[Incomplete, Incomplete] | None = None + self, query: str, org: Incomplete | None = None, dialect=..., params: SupportsItems[str, Incomplete] | None = None ): ... def query( - self, query: str, org: Incomplete | None = None, params: dict[Incomplete, Incomplete] | None = None + self, query: str, org: Incomplete | None = None, params: SupportsItems[str, Incomplete] | None = None ) -> TableList: ... def query_stream( - self, query: str, org: Incomplete | None = None, params: dict[Incomplete, Incomplete] | None = None + self, query: str, org: Incomplete | None = None, params: SupportsItems[str, Incomplete] | None = None ) -> Generator[FluxRecord, Any, None]: ... def query_data_frame( self, query: str, - org: Incomplete | None = None, + org: Organization | str | None = None, data_frame_index: list[str] | None = None, - params: dict[Incomplete, Incomplete] | None = None, + params: SupportsItems[str, Incomplete] | None = None, + use_extension_dtypes: bool = False, ): ... def query_data_frame_stream( self, query: str, - org: Incomplete | None = None, + org: Organization | str | None = None, data_frame_index: list[str] | None = None, - params: dict[Incomplete, Incomplete] | None = None, + params: SupportsItems[str, Incomplete] | None = None, + use_extension_dtypes: bool = False, ): ... def __del__(self) -> None: ... diff --git a/stubs/influxdb-client/influxdb_client/client/query_api_async.pyi b/stubs/influxdb-client/influxdb_client/client/query_api_async.pyi index f65b647f0..4648d07d9 100644 --- a/stubs/influxdb-client/influxdb_client/client/query_api_async.pyi +++ b/stubs/influxdb-client/influxdb_client/client/query_api_async.pyi @@ -1,31 +1,39 @@ -from _typeshed import Incomplete +from _typeshed import Incomplete, SupportsItems from collections.abc import AsyncGenerator from influxdb_client.client._base import _BaseQueryApi from influxdb_client.client.flux_table import FluxRecord, TableList +from influxdb_client.domain.dialect import Dialect +from influxdb_client.domain.organization import Organization class QueryApiAsync(_BaseQueryApi): def __init__(self, influxdb_client, query_options=...) -> None: ... async def query( - self, query: str, org: Incomplete | None = None, params: dict[Incomplete, Incomplete] | None = None + self, query: str, org: Incomplete | None = None, params: SupportsItems[str, Incomplete] | None = None ) -> TableList: ... async def query_stream( - self, query: str, org: Incomplete | None = None, params: dict[Incomplete, Incomplete] | None = None + self, query: str, org: Incomplete | None = None, params: SupportsItems[str, Incomplete] | None = None ) -> AsyncGenerator[FluxRecord, None]: ... async def query_data_frame( self, query: str, - org: Incomplete | None = None, + org: str | Organization | None = None, data_frame_index: list[str] | None = None, - params: dict[Incomplete, Incomplete] | None = None, + params: SupportsItems[str, Incomplete] | None = None, + use_extension_dtypes: bool = False, ): ... async def query_data_frame_stream( self, query: str, - org: Incomplete | None = None, + org: str | Organization | None = None, data_frame_index: list[str] | None = None, - params: dict[Incomplete, Incomplete] | None = None, + params: SupportsItems[str, Incomplete] | None = None, + use_extension_dtypes: bool = False, ): ... async def query_raw( - self, query: str, org: Incomplete | None = None, dialect=..., params: dict[Incomplete, Incomplete] | None = None - ): ... + self, + query: str, + org: str | Organization | None = None, + dialect: Dialect = ..., + params: SupportsItems[str, Incomplete] | None = None, + ) -> str: ... diff --git a/stubs/influxdb-client/influxdb_client/client/util/helpers.pyi b/stubs/influxdb-client/influxdb_client/client/util/helpers.pyi index 385ebda06..15963b5ab 100644 --- a/stubs/influxdb-client/influxdb_client/client/util/helpers.pyi +++ b/stubs/influxdb-client/influxdb_client/client/util/helpers.pyi @@ -1 +1,4 @@ -def get_org_query_param(org, client, required_id: bool = False): ... +from influxdb_client.client.influxdb_client import InfluxDBClient +from influxdb_client.domain.organization import Organization + +def get_org_query_param(org: Organization | str | None, client: InfluxDBClient, required_id: bool = False) -> str: ... diff --git a/stubs/influxdb-client/influxdb_client/domain/organization.pyi b/stubs/influxdb-client/influxdb_client/domain/organization.pyi index 4988fa6c2..160835a90 100644 --- a/stubs/influxdb-client/influxdb_client/domain/organization.pyi +++ b/stubs/influxdb-client/influxdb_client/domain/organization.pyi @@ -7,7 +7,7 @@ class Organization: def __init__( self, links: Incomplete | None = None, - id: Incomplete | None = None, + id: str | None = None, name: Incomplete | None = None, default_storage_type: str | None = None, description: Incomplete | None = None, @@ -20,9 +20,9 @@ class Organization: @links.setter def links(self, links) -> None: ... @property - def id(self): ... + def id(self) -> str: ... @id.setter - def id(self, id) -> None: ... + def id(self, id: str) -> None: ... @property def name(self): ... @name.setter