quasardb 3.14.2.dev7__cp312-cp312-manylinux_2_26_aarch64.manylinux_2_28_aarch64.whl

quasardb/quasardb/_batch_inserter.pyi

@@ -0,0 +1,30 @@
+class TimeSeriesBatch:
+    def push(self) -> None:
+        """
+        Regular batch push
+        """
+
+    def push_async(self) -> None:
+        """
+        Asynchronous batch push that buffers data inside the QuasarDB daemon
+        """
+
+    def push_fast(self) -> None:
+        """
+        Fast, in-place batch push that is efficient when doing lots of small, incremental pushes.
+        """
+
+    def push_truncate(self, **kwargs) -> None:
+        """
+        Before inserting data, truncates any existing data. This is useful when you want your insertions to be idempotent, e.g. in case of a retry.
+        """
+
+    def set_blob(self, index: int, blob: bytes) -> None: ...
+    def set_double(self, index: int, double: float) -> None: ...
+    def set_int64(self, index: int, int64: int) -> None: ...
+    def set_string(self, index: int, string: str) -> None: ...
+    def set_timestamp(self, index: int, timestamp: object) -> None: ...
+    def start_row(self, ts: object) -> None:
+        """
+        Calling this function marks the beginning of processing a new row.
+        """

quasardb/quasardb/_blob.pyi

@@ -0,0 +1,16 @@
+import datetime
+
+from ._entry import ExpirableEntry
+
+class Blob(ExpirableEntry):
+    def compare_and_swap(self, new_content: str, comparand: str) -> bytes: ...
+    def get(self) -> bytes: ...
+    def get_and_remove(self) -> bytes: ...
+    def get_and_update(self, data: str) -> bytes: ...
+    def put(self, data: str) -> None: ...
+    def remove_if(self, comparand: str) -> None: ...
+    def update(
+        self,
+        data: str,
+        expiry: datetime.datetime = datetime.datetime.fromtimestamp(0),
+    ) -> None: ...

quasardb/quasardb/_cluster.pyi

@@ -0,0 +1,100 @@
+from __future__ import annotations
+
+import datetime
+
+import numpy as np
+
+from ._batch_column import BatchColumnInfo
+from ._batch_inserter import TimeSeriesBatch
+from ._blob import Blob
+from ._continuous import QueryContinuous
+from ._double import Double
+from ._integer import Integer
+from ._node import Node
+from ._options import Options
+from ._perf import Perf
+from ._properties import Properties
+from ._reader import Reader
+from ._string import String
+from ._table import Table
+from ._tag import Tag
+from ._timestamp import Timestamp
+from ._writer import Writer
+
+class Cluster:
+    """
+    Represents a connection to the QuasarDB cluster.
+    """
+
+    def __enter__(self) -> Cluster: ...
+    def __exit__(self, exc_type: object, exc_value: object, exc_tb: object) -> None: ...
+    def __init__(
+        self,
+        uri: str,
+        user_name: str = "",
+        user_private_key: str = "",
+        cluster_public_key: str = "",
+        user_security_file: str = "",
+        cluster_public_key_file: str = "",
+        timeout: datetime.timedelta = datetime.timedelta(minutes=1),
+        do_version_check: bool = False,
+        enable_encryption: bool = False,
+        compression_mode: Options.Compression = ...,  # balanced
+        client_max_parallelism: int = 0,
+    ) -> None: ...
+    def blob(self, alias: str) -> Blob: ...
+    def close(self) -> None: ...
+    def compact_abort(self) -> None: ...
+    def compact_full(self) -> None: ...
+    def compact_progress(self) -> int: ...
+    def double(self, alias: str) -> Double: ...
+    def endpoints(self) -> list[str]: ...
+    def find(self, query: str) -> list[str]: ...
+    def get_memory_info(self) -> str: ...
+    def inserter(self, column_info_list: list[BatchColumnInfo]) -> TimeSeriesBatch: ...
+    def integer(self, alias: str) -> Integer: ...
+    def is_open(self) -> bool: ...
+    def node(self, uri: str) -> Node: ...
+    def node_config(self, uri: str) -> dict[str, object]: ...
+    def node_status(self, uri: str) -> dict[str, object]: ...
+    def node_topology(self, uri: str) -> dict[str, object]: ...
+    def options(self) -> Options: ...
+    def perf(self) -> Perf: ...
+    def pinned_writer(self) -> Writer: ...
+    def prefix_count(self, prefix: str) -> int: ...
+    def prefix_get(self, prefix: str, max_count: int) -> list[str]: ...
+    def properties(self) -> Properties: ...
+    def purge_all(self, timeout: datetime.timedelta) -> None: ...
+    def purge_cache(self, timeout: datetime.timedelta) -> None: ...
+    def query(
+        self, query: str, blobs: bool | list[str] = False
+    ) -> list[dict[str, object]]: ...
+    def query_continuous_full(
+        self, query: str, pace: datetime.timedelta, blobs: bool | list[str] = False
+    ) -> QueryContinuous: ...
+    def query_continuous_new_values(
+        self, query: str, pace: datetime.timedelta, blobs: bool | list[str] = False
+    ) -> QueryContinuous: ...
+    def query_numpy(self, query: str) -> list[tuple[str, np.ma.MaskedArray]]: ...
+    def reader(
+        self,
+        table_names: list[str],
+        column_names: list[str] = [],
+        batch_size: int = 0,
+        ranges: list[tuple] = [],
+    ) -> Reader: ...
+    def string(self, alias: str) -> String: ...
+    def suffix_count(self, suffix: str) -> int: ...
+    def suffix_get(self, suffix: str, max_count: int) -> list[str]: ...
+    def table(self, alias: str) -> Table: ...
+    def tag(self, alias: str) -> Tag: ...
+    def tidy_memory(self) -> None: ...
+    def timestamp(self, alias: str) -> Timestamp: ...
+    def trim_all(
+        self, pause: datetime.timedelta, timeout: datetime.timedelta
+    ) -> None: ...
+    def ts(self, alias: str) -> Table: ...
+    def ts_batch(self, column_info_list: list[BatchColumnInfo]) -> TimeSeriesBatch: ...
+    def uri(self) -> str: ...
+    def wait_for_compaction(self) -> None: ...
+    def writer(self) -> Writer: ...

quasardb/quasardb/_continuous.pyi

@@ -0,0 +1,16 @@
+from __future__ import annotations
+
+# import datetime
+
+class QueryContinuous:
+    def __iter__(self) -> QueryContinuous: ...
+    def __next__(self) -> list[dict[str, object]]: ...
+    def probe_results(self) -> list[dict[str, object]]: ...
+    def results(self) -> list[dict[str, object]]: ...
+    # def run(
+    #     self,
+    #     mode: qdb_query_continuous_mode_type_t,
+    #     pace: datetime.timedelta,
+    #     query: str,
+    # ) -> None: ...
+    def stop(self) -> None: ...

quasardb/quasardb/_double.pyi

@@ -0,0 +1,7 @@
+from ._entry import ExpirableEntry
+
+class Double(ExpirableEntry):
+    def add(self, addend: float) -> float: ...
+    def get(self) -> float: ...
+    def put(self, double: float) -> None: ...
+    def update(self, double: float) -> None: ...

quasardb/quasardb/_entry.pyi

@@ -0,0 +1,60 @@
+from __future__ import annotations
+
+import datetime
+
+class Entry:
+    class Metadata: ...
+
+    class Type:
+        Uninitialized: Entry.Type  # value = <Type.Uninitialized: -1>
+        Blob: Entry.Type  # value = <Type.Blob: 0>
+        Integer: Entry.Type  # value = <Type.Integer: 1>
+        HashSet: Entry.Type  # value = <Type.HashSet: 2>
+        Tag: Entry.Type  # value = <Type.Tag: 3>
+        Deque: Entry.Type  # value = <Type.Deque: 4>
+        Stream: Entry.Type  # value = <Type.Stream: 5>
+        Timeseries: Entry.Type  # value = <Type.Timeseries: 6>
+        __members__: dict[str, Entry.Type]
+        def __and__(self, other: object) -> object: ...
+        def __eq__(self, other: object) -> bool: ...
+        def __ge__(self, other: object) -> bool: ...
+        def __getstate__(self) -> int: ...
+        def __gt__(self, other: object) -> bool: ...
+        def __hash__(self) -> int: ...
+        def __index__(self) -> int: ...
+        def __init__(self, value: int) -> None: ...
+        def __int__(self) -> int: ...
+        def __invert__(self) -> object: ...
+        def __le__(self, other: object) -> bool: ...
+        def __lt__(self, other: object) -> bool: ...
+        def __ne__(self, other: object) -> bool: ...
+        def __or__(self, other: object) -> object: ...
+        def __rand__(self, other: object) -> object: ...
+        def __repr__(self) -> str: ...
+        def __ror__(self, other: object) -> object: ...
+        def __rxor__(self, other: object) -> object: ...
+        def __setstate__(self, state: int) -> None: ...
+        def __str__(self) -> str: ...
+        def __xor__(self, other: object) -> object: ...
+        @property
+        def name(self) -> str: ...
+        @property
+        def value(self) -> int: ...
+
+    def attach_tag(self, tag: str) -> bool: ...
+    def attach_tags(self, tags: list[str]) -> None: ...
+    def detach_tag(self, tag: str) -> bool: ...
+    def detach_tags(self, tags: list[str]) -> None: ...
+    def exists(self) -> bool: ...
+    def get_entry_type(self) -> Entry.Type: ...
+    def get_location(self) -> tuple[str, int]: ...
+    def get_metadata(self) -> Metadata: ...
+    def get_name(self) -> str: ...
+    def get_tags(self) -> list[str]: ...
+    def has_tag(self, tag: str) -> bool: ...
+    def remove(self) -> None: ...
+
+class ExpirableEntry(Entry):
+    def expires_at(self, expiry_time: datetime.datetime) -> None: ...
+    def expires_from_now(self, expiry_delta: datetime.timedelta) -> None: ...
+    def get_expiry_time(self) -> datetime.datetime: ...

quasardb/quasardb/_error.pyi

@@ -0,0 +1,15 @@
+class Error(RuntimeError): ...
+class AliasAlreadyExistsError(Error): ...
+class AliasNotFoundError(Error): ...
+class AsyncPipelineFullError(Error): ...
+class IncompatibleTypeError(Error): ...
+class InputBufferTooSmallError(Error): ...
+class InternalLocalError(Error): ...
+class InvalidArgumentError(Error): ...
+class InvalidDatetimeError(Error): ...
+class InvalidHandleError(Error): ...
+class InvalidQueryError(Error): ...
+class NotImplementedError(Error): ...
+class OutOfBoundsError(Error): ...
+class TryAgainError(Error): ...
+class UninitializedError(Error): ...

quasardb/quasardb/_integer.pyi

@@ -0,0 +1,7 @@
+from ._entry import ExpirableEntry
+
+class Integer(ExpirableEntry):
+    def add(self, addend: int) -> int: ...
+    def get(self) -> int: ...
+    def put(self, integer: int) -> None: ...
+    def update(self, integer: int) -> None: ...

quasardb/quasardb/_node.pyi

@@ -0,0 +1,26 @@
+class DirectBlob:
+    def get(self) -> bytes: ...
+    def put(self, data: str) -> None: ...
+    def remove(self) -> None: ...
+    def update(self, data: str) -> None: ...
+
+class DirectInteger:
+    def get(self) -> int: ...
+    def put(self, integer: int) -> None: ...
+    def remove(self) -> None: ...
+    def update(self, integer: int) -> None: ...
+
+class Node:
+    def __init__(
+        self,
+        uri: str,
+        user_name: str = "",
+        user_private_key: str = "",
+        cluster_public_key: str = "",
+        user_security_file: str = "",
+        cluster_public_key_file: str = "",
+        enable_encryption: bool = False,
+    ) -> None: ...
+    def blob(self, alias: str) -> DirectBlob: ...
+    def integer(self, alias: str) -> DirectInteger: ...
+    def prefix_get(self, prefix: str, max_count: int) -> list[str]: ...

quasardb/quasardb/_options.pyi

@@ -0,0 +1,105 @@
+from __future__ import annotations
+
+import datetime
+
+class Options:
+    class Compression:
+        Disabled: Options.Compression  # value = <Compression.Disabled: 0>
+        Best: Options.Compression  # value = <Compression.Best: 2>
+        Balanced: Options.Compression  # value = <Compression.Balanced: 3>
+        __members__: dict[str, Options.Compression]
+        def __and__(self, other: object) -> object: ...
+        def __eq__(self, other: object) -> bool: ...
+        def __ge__(self, other: object) -> bool: ...
+        def __getstate__(self) -> int: ...
+        def __gt__(self, other: object) -> bool: ...
+        def __hash__(self) -> int: ...
+        def __index__(self) -> int: ...
+        def __init__(self, value: int) -> None: ...
+        def __int__(self) -> int: ...
+        def __invert__(self) -> object: ...
+        def __le__(self, other: object) -> bool: ...
+        def __lt__(self, other: object) -> bool: ...
+        def __ne__(self, other: object) -> bool: ...
+        def __or__(self, other: object) -> object: ...
+        def __rand__(self, other: object) -> object: ...
+        def __repr__(self) -> str: ...
+        def __ror__(self, other: object) -> object: ...
+        def __rxor__(self, other: object) -> object: ...
+        def __setstate__(self, state: int) -> None: ...
+        def __str__(self) -> str: ...
+        def __xor__(self, other: object) -> object: ...
+        @property
+        def name(self) -> str: ...
+        @property
+        def value(self) -> int: ...
+
+    class Encryption:
+        Disabled: Options.Encryption  # value = <Encryption.Disabled: 0>
+        AES256GCM: Options.Encryption  # value = <Encryption.AES256GCM: 1>
+        __members__: dict[str, Options.Encryption]
+        def __and__(self, other: object) -> object: ...
+        def __eq__(self, other: object) -> bool: ...
+        def __ge__(self, other: object) -> bool: ...
+        def __getstate__(self) -> int: ...
+        def __gt__(self, other: object) -> bool: ...
+        def __hash__(self) -> int: ...
+        def __index__(self) -> int: ...
+        def __init__(self, value: int) -> None: ...
+        def __int__(self) -> int: ...
+        def __invert__(self) -> object: ...
+        def __le__(self, other: object) -> bool: ...
+        def __lt__(self, other: object) -> bool: ...
+        def __ne__(self, other: object) -> bool: ...
+        def __or__(self, other: object) -> object: ...
+        def __rand__(self, other: object) -> object: ...
+        def __repr__(self) -> str: ...
+        def __ror__(self, other: object) -> object: ...
+        def __rxor__(self, other: object) -> object: ...
+        def __setstate__(self, state: int) -> None: ...
+        def __str__(self) -> str: ...
+        def __xor__(self, other: object) -> object: ...
+        @property
+        def name(self) -> str: ...
+        @property
+        def value(self) -> int: ...
+
+    def disable_user_properties(self) -> None: ...
+    def enable_user_properties(self) -> None: ...
+    def get_client_max_batch_load(self) -> int:
+        """
+        Get the number of shards per thread used for the batch writer.
+        """
+
+    def get_client_max_in_buf_size(self) -> int: ...
+    def get_client_max_parallelism(self) -> int: ...
+    def get_cluster_max_in_buf_size(self) -> int: ...
+    def get_connection_per_address_soft_limit(self) -> int:
+        """
+        Get the maximum number of connections per qdbd node
+        """
+
+    def get_query_max_length(self) -> int: ...
+    def get_stabilization_max_wait(self) -> datetime.timedelta: ...
+    def get_timeout(self) -> datetime.timedelta: ...
+    def get_timezone(self) -> str: ...
+    def set_client_max_batch_load(self, shard_count: int) -> None:
+        """
+        Adjust the number of shards per thread used for the batch writer.
+        """
+
+    def set_client_max_in_buf_size(self, max_size: int) -> None: ...
+    def set_client_soft_memory_limit(self, limit: int) -> None: ...
+    def set_cluster_public_key(self, key: str) -> None: ...
+    def set_connection_per_address_soft_limit(self, max_count: int) -> None:
+        """
+        Adjust the maximum number of connections per qdbd node
+        """
+
+    def set_encryption(self, algorithm: Options.Encryption) -> None: ...
+    def set_max_cardinality(self, cardinality: int) -> None: ...
+    def set_query_max_length(self, query_max_length: int) -> None: ...
+    def set_stabilization_max_wait(self, max_wait: datetime.timedelta) -> None: ...
+    def set_timeout(self, timeout: datetime.timedelta) -> None: ...
+    def set_timezone(self, tz: str) -> None: ...
+    def set_user_credentials(self, user: str, private_key: str) -> None: ...

quasardb/quasardb/_perf.pyi

@@ -0,0 +1,5 @@
+class Perf:
+    def clear(self) -> None: ...
+    def disable(self) -> None: ...
+    def enable(self) -> None: ...
+    def get(self, flame: bool = False, outfile: str = "") -> object: ...

quasardb/quasardb/_properties.pyi

@@ -0,0 +1,5 @@
+class Properties:
+    def clear(self) -> None: ...
+    def get(self, key: str) -> str | None: ...
+    def put(self, key: str, value: str) -> None: ...
+    def remove(self, key: str) -> None: ...

quasardb/quasardb/_query.pyi

@@ -0,0 +1,2 @@
+class FindQuery:
+    def run(self) -> list[str]: ...

quasardb/quasardb/_reader.pyi

@@ -0,0 +1,9 @@
+from __future__ import annotations
+
+import typing
+
+class Reader:
+    def __enter__(self) -> Reader: ...
+    def __exit__(self, exc_type: object, exc_value: object, exc_tb: object) -> None: ...
+    def __iter__(self) -> typing.Iterator[dict[str, object]]: ...
+    def get_batch_size(self) -> int: ...

quasardb/quasardb/_retry.pyi

@@ -0,0 +1,16 @@
+import datetime
+
+class RetryOptions:
+    delay: datetime.timedelta
+    exponent: int
+    jitter: float
+    retries_left: int
+    def __init__(
+        self,
+        retries: int = 3,
+        delay: datetime.timedelta = ...,
+        exponent: int = 2,
+        jitter: float = 0.1,
+    ) -> None: ...
+    def has_next(self) -> bool: ...
+    def next(self) -> RetryOptions: ...

quasardb/quasardb/_string.pyi

@@ -0,0 +1,12 @@
+import datetime
+
+from ._entry import ExpirableEntry
+
+class String(ExpirableEntry):
+    def compare_and_swap(self, new_content: str, comparand: str) -> str: ...
+    def get(self) -> str: ...
+    def get_and_remove(self) -> str: ...
+    def get_and_update(self, data: str) -> str: ...
+    def put(self, data: str) -> None: ...
+    def remove_if(self, comparand: str) -> None: ...
+    def update(self, data: str, expiry: datetime.datetime = ...) -> None: ...

quasardb/quasardb/_table.pyi

@@ -0,0 +1,125 @@
+from __future__ import annotations
+
+import datetime
+import typing
+
+import numpy
+
+from quasardb.quasardb._reader import Reader
+
+from ._entry import Entry
+
+class ColumnType:
+    Uninitialized: ColumnType  # value = <ColumnType.Uninitialized: -1>
+    Double: ColumnType  # value = <ColumnType.Double: 0>
+    Blob: ColumnType  # value = <ColumnType.Blob: 1>
+    Int64: ColumnType  # value = <ColumnType.Int64: 2>
+    Timestamp: ColumnType  # value = <ColumnType.Timestamp: 3>
+    String: ColumnType  # value = <ColumnType.String: 4>
+    Symbol: ColumnType  # value = <ColumnType.Symbol: 5>
+    __members__: dict[str, ColumnType]
+    def __and__(self, other: object) -> object: ...
+    def __eq__(self, other: object) -> bool: ...
+    def __ge__(self, other: object) -> bool: ...
+    def __getstate__(self) -> int: ...
+    def __gt__(self, other: object) -> bool: ...
+    def __hash__(self) -> int: ...
+    def __index__(self) -> int: ...
+    def __init__(self, value: int) -> None: ...
+    def __int__(self) -> int: ...
+    def __invert__(self) -> object: ...
+    def __le__(self, other: object) -> bool: ...
+    def __lt__(self, other: object) -> bool: ...
+    def __ne__(self, other: object) -> bool: ...
+    def __or__(self, other: object) -> object: ...
+    def __rand__(self, other: object) -> object: ...
+    def __repr__(self) -> str: ...
+    def __ror__(self, other: object) -> object: ...
+    def __rxor__(self, other: object) -> object: ...
+    def __setstate__(self, state: int) -> None: ...
+    def __str__(self) -> str: ...
+    def __xor__(self, other: object) -> object: ...
+    @property
+    def name(self) -> str: ...
+    @property
+    def value(self) -> int: ...
+
+class ColumnInfo:
+    name: str
+    symtable: str
+    type: ColumnType
+    @typing.overload
+    def __init__(self, type: ColumnType, name: str) -> None: ...
+    @typing.overload
+    def __init__(self, type: ColumnType, name: str, symtable: str) -> None: ...
+    def __repr__(self) -> str: ...
+
+class IndexedColumnInfo:
+    @typing.overload
+    def __init__(self, type: ColumnType, index: int) -> None: ...
+    @typing.overload
+    def __init__(self, type: ColumnType, index: int, symtable: str) -> None: ...
+    @property
+    def index(self) -> int: ...
+    @property
+    def symtable(self) -> str: ...
+    @property
+    def type(self) -> ColumnType: ...
+
+class Table(Entry):
+    def __repr__(self) -> str: ...
+    def blob_get_ranges(
+        self, column: str, ranges: object = None
+    ) -> tuple[numpy.ndarray, numpy.ma.MaskedArray]: ...
+    def blob_insert(
+        self, column: str, timestamps: numpy.ndarray, values: numpy.ma.MaskedArray
+    ) -> None: ...
+    def column_id_by_index(self, index: int) -> str: ...
+    def column_index_by_id(self, alias: str) -> int: ...
+    def column_info_by_index(self, index: int) -> ColumnInfo: ...
+    def column_type_by_id(self, alias: str) -> ColumnType: ...
+    def column_type_by_index(self, index: int) -> ColumnType: ...
+    def create(
+        self,
+        columns: list[ColumnInfo],
+        shard_size: datetime.timedelta = ...,
+        ttl: datetime.timedelta = ...,
+    ) -> None: ...
+    def double_get_ranges(
+        self, column: str, ranges: object = None
+    ) -> tuple[numpy.ndarray, numpy.ma.MaskedArray]: ...
+    def double_insert(
+        self, column: str, timestamps: numpy.ndarray, values: numpy.ma.MaskedArray
+    ) -> None: ...
+    def erase_ranges(self, column: str, ranges: object) -> int: ...
+    def get_shard_size(self) -> datetime.timedelta: ...
+    def get_ttl(self) -> datetime.timedelta: ...
+    def has_ttl(self) -> bool: ...
+    def insert_columns(self, columns: list[ColumnInfo]) -> None: ...
+    def int64_get_ranges(
+        self, column: str, ranges: object = None
+    ) -> tuple[numpy.ndarray, numpy.ma.MaskedArray]: ...
+    def int64_insert(
+        self, column: str, timestamps: numpy.ndarray, values: numpy.ma.MaskedArray
+    ) -> None: ...
+    def list_columns(self) -> list[ColumnInfo]: ...
+    def reader(
+        self,
+        column_names: list[str] = [],
+        batch_size: int = 0,
+        ranges: list[tuple] = [],
+    ) -> Reader: ...
+    def retrieve_metadata(self) -> None: ...
+    def string_get_ranges(
+        self, column: str, ranges: object = None
+    ) -> tuple[numpy.ndarray, numpy.ma.MaskedArray]: ...
+    def string_insert(
+        self, column: str, timestamps: numpy.ndarray, values: numpy.ma.MaskedArray
+    ) -> None: ...
+    def subscribe(self, conn: object) -> object: ...
+    def timestamp_get_ranges(
+        self, column: str, ranges: object = None
+    ) -> tuple[numpy.ndarray, numpy.ma.MaskedArray]: ...
+    def timestamp_insert(
+        self, column: str, timestamps: numpy.ndarray, values: numpy.ma.MaskedArray
+    ) -> None: ...

quasardb/quasardb/_tag.pyi

@@ -0,0 +1,5 @@
+from ._entry import Entry
+
+class Tag(Entry):
+    def count(self) -> int: ...
+    def get_entries(self) -> list[str]: ...

quasardb/quasardb/_timestamp.pyi

@@ -0,0 +1,9 @@
+import datetime
+
+from ._entry import ExpirableEntry
+
+class Timestamp(ExpirableEntry):
+    def add(self, addend: datetime.datetime) -> datetime.datetime: ...
+    def get(self) -> datetime.datetime: ...
+    def put(self, timestamp: datetime.datetime) -> None: ...
+    def update(self, timestamp: datetime.datetime) -> None: ...