quasardb 3.14.2.dev8__cp310-cp310-manylinux_2_26_x86_64.manylinux_2_28_x86_64.whl

quasardb/pool.py

@@ -0,0 +1,338 @@
+from __future__ import annotations
+
+import functools
+import logging
+import threading
+import weakref
+from types import TracebackType
+from typing import Any, Callable, Optional, Type, TypeVar, Union
+
+# import quasardb
+from quasardb import Cluster
+
+logger = logging.getLogger("quasardb.pool")
+
+T = TypeVar("T")
+
+
+def _create_conn(**kwargs: Any) -> Cluster:
+    return Cluster(**kwargs)
+
+
+class SessionWrapper:
+    def __init__(self, pool: Pool, conn: Cluster):
+        self._conn = conn
+        self._pool = pool
+
+    def __getattr__(self, attr: Any) -> Any:
+        # This hack copies all the quasardb.Cluster() properties, functions and
+        # whatnot, and pretends that this class is actually a quasardb.Cluster.
+        #
+        # The background is that when someone does this:
+        #
+        # with pool.connect() as conn:
+        #   ...
+        #
+        # we don't want the the connection to be closed near the end, but rather
+        # released back onto the pool.
+        #
+        # Now, my initial approach was to build a pool.Session class which inherited
+        # from quasardb.Cluster, and just overload the __exit__ function there. But,
+        # we want people to have the flexibility to pass in an external `get_conn` callback
+        # in the pool, which establishes a connection, because they may have to look up
+        # some dynamic security tokens. This function should then, in turn, return a vanilla
+        # quasardb.Cluster() object.
+        #
+        # And this is why we end up with the solution below.
+        if attr in self.__dict__:
+            return getattr(self, attr)
+
+        return getattr(self._conn, attr)
+
+    def __enter__(self: T) -> T:
+        return self
+
+    def __exit__(
+        self,
+        exc_type: Optional[Type[BaseException]],
+        exc_val: Optional[BaseException],
+        exc_tb: Optional[TracebackType],
+    ) -> None:
+        self._pool.release(self._conn)
+
+
+class Pool:
+    """
+    A connection pool. This class should not be initialized directly, but
+    rather the subclass `SingletonPool` should be initialized.
+
+    The constructor either accepts all regular `quasardb.Cluster()` connection parameters,
+    or a `get_conn` parameter which is invoked any time a new connection should be
+    created.
+
+    Example usage:
+    --------------
+
+    Initialize the pool by passthrough of `quasardb.Cluster()` arguments:
+    ```
+    import quasardb.pool as pool
+
+    pool.SingletonPool(uri='qdb://127.0.0.1:2836',
+                       cluster_public_key='...',
+                       user_private_key='...',
+                       user_name='...')
+    ```
+
+    Initialize pool by providing a callback function
+    ```
+    import quasardb
+    import quasardb.pool as pool
+
+    def my_qdb_connection_create():
+       # This function is invoked any time the pool needs to allocate
+       # a new connection.
+       return quasardb.Cluster(uri='qdb://127.0.0.1:2836',
+                               cluster_public_key='...',
+                               user_private_key='...',
+                               user_name='...')
+
+    pool.SingletonPool(get_conn=my_qdb_connection_create)
+    ```
+
+    """
+
+    def __init__(
+        self, get_conn: Optional[Callable[..., Cluster]] = None, **kwargs: Any
+    ):
+        self._all_connections: list[SessionWrapper] = []
+
+        if get_conn is None:
+            get_conn = functools.partial(_create_conn, **kwargs)
+
+        if not callable(get_conn):
+            raise TypeError("get_conn must be callable")
+
+        self._get_conn = get_conn
+
+    def __enter__(self: T) -> T:
+        return self
+
+    def __exit__(
+        self,
+        exc_type: Optional[Type[BaseException]],
+        exc_val: Optional[BaseException],
+        exc_tb: Optional[TracebackType],
+    ) -> None:
+        self.close()
+
+    def _create_conn(self) -> SessionWrapper:
+        return SessionWrapper(self, self._get_conn())
+
+    def close(self) -> None:
+        """
+        Close this connection pool, and all associated connections. This function
+        is automatically invoked when used in a with-block or when using the global
+        `instance()` singleton.
+        """
+        for conn in self._all_connections:
+            logger.debug("closing connection {}".format(conn))
+            conn.close()
+
+    def _do_connect(self) -> SessionWrapper:
+        raise NotImplementedError
+
+    def connect(self) -> SessionWrapper:
+        """
+        Acquire a new connection from the pool. Returned connection must either
+        be explicitly released using `pool.release()`, or should be wrapped in a
+        with-block.
+
+        Returns:
+        --------
+        `quasardb.Cluster`
+        """
+        logger.info("Acquiring connection from pool")
+        return self._do_connect()
+
+    def _do_release(self, conn: Cluster) -> None:
+        raise NotImplementedError
+
+    def release(self, conn: Cluster) -> None:
+        """
+        Put a connection back into the pool
+        """
+        logger.info("Putting connection back onto pool")
+        return self._do_release(conn)
+
+
+class SingletonPool(Pool):
+    """
+    Implementation of our connection pool that maintains just a single connection
+    per thread.
+
+    Example usage:
+    --------
+
+    Using pool.connect() in a with-block:
+    ```
+    import quasardb.pool as pool
+
+    with pool.Pool(uri='qdb://127.0.0.1:2836') as pool:
+      with pool.connect() as conn:
+        conn.query(...)
+    ```
+
+    Explicitly releasing the connection using `Pool.release()`:
+    ```
+    import quasardb.pool as pool
+
+    with pool.Pool(uri='qdb://127.0.0.1:2836') as pool:
+      conn = pool.connect()
+      try:
+        conn.query(...)
+      finally:
+        pool.release(conn)
+    ```
+    """
+
+    def __init__(self, **kwargs: Any):
+        Pool.__init__(self, **kwargs)
+        self._conn = threading.local()
+
+    def _do_connect(self) -> SessionWrapper:
+        try:
+            c = self._conn.current()
+            if c:
+                return c
+        except AttributeError:
+            pass
+
+        c = self._create_conn()
+        self._conn.current = weakref.ref(c)
+        self._all_connections.append(c)
+
+        return c
+
+    def _do_release(self, conn: Cluster) -> None:
+        # Thread-local connections do not have to be 'released'.
+        pass
+
+
+__instance = None
+
+
+def initialize(*args: Any, **kwargs: Any) -> None:
+    """
+    Initialize a new global SingletonPool. Forwards all arguments to the constructor of
+    `SingletonPool()`.
+
+    After initialization, the instance can be used by invoking `instance()`.
+
+    Example usage:
+    --------------
+
+    ```
+    import quasardb.pool as pool
+
+    pool.initialize(cluster='qdb://127.0.0.1:2836')
+
+    # ...
+
+    def myfunction()
+      with pool.instance().connect() as conn:
+        conn.query(...)
+    ```
+    """
+    global __instance
+    __instance = SingletonPool(*args, **kwargs)
+
+
+def instance() -> SingletonPool:
+    """
+    Singleton accessor. Instance must have been initialized using initialize()
+    prior to invoking this function.
+
+    Returns:
+    --------
+    `SingletonPool`
+
+    """
+    global __instance
+    assert (
+        __instance is not None
+    ), "Global connection pool uninitialized: please initialize by calling the initialize() function."
+    return __instance
+
+
+def _inject_conn_arg(
+    conn: SessionWrapper,
+    arg: Union[str, int],
+    args: tuple[Any, ...],
+    kwargs: dict[str, Any],
+) -> tuple[tuple[Any, ...], dict[str, Any]]:
+    """
+    Decorator utility function. Takes the argument provided to the decorator
+    that configures how we should inject the pool into the args to the callback
+    function, and injects it in the correct place.
+    """
+    if isinstance(arg, int):
+        # Invoked positional such as `@with_pool(arg=1)`, put the pool into the
+        # correct position.
+        #
+        # Because positional args are always a tuple, and tuples don't have an
+        # easy 'insert into position' function, we just cast to and from a list.
+        args_list = list(args)
+        args_list.insert(arg, conn)
+        args = tuple(args_list)
+    else:
+        assert isinstance(arg, str) == True
+        # If not a number, we assume it's a kwarg, which makes things easier
+        kwargs[arg] = conn
+
+    return (args, kwargs)
+
+
+def with_conn(
+    _fn: Optional[Callable[..., Any]] = None, *, arg: Union[str, int] = 0
+) -> Callable[..., Any]:
+    """
+    Decorator function that handles connection assignment, release and invocation for you.
+    Should be used in conjuction with the global singleton accessor, see also: `initialize()`.
+
+    By default, the decorator function injects the connection as the first argument to the
+    function:
+    ```
+    import quasardb.pool as pool
+    pool.initialize(...)
+
+    @pool.with_conn()
+    def myfunction(conn, arg1, arg2):
+       conn.query(...)
+    ```
+
+    You can optionally provide an `arg` parameter to denote which named keyword to provide
+    it as:
+    ```
+    import quasardb.pool as pool
+    pool.initialize(...)
+
+    @pool.with_conn(arg='conn')
+    def myfunction(arg1, arg2, conn=None):
+       conn.query(...)
+    ```
+    """
+
+    def inner(fn: Callable[..., Any]) -> Callable[..., Any]:
+        def wrapper(*args: Any, **kwargs: Any) -> Any:
+            pool = instance()
+
+            with pool.connect() as conn:
+                (args, kwargs) = _inject_conn_arg(conn, arg, args, kwargs)
+                return fn(*args, **kwargs)
+
+        return wrapper
+
+    if _fn is None:
+        return inner
+    else:
+        return inner(_fn)

quasardb/pybind11/CMakeFiles/CMakeDirectoryInformation.cmake

@@ -0,0 +1,16 @@
+# CMAKE generated file: DO NOT EDIT!
+# Generated by "Unix Makefiles" Generator, CMake Version 4.1
+
+# Relative path conversion top directories.
+set(CMAKE_RELATIVE_PATH_TOP_SOURCE "/home/teamcity/buildAgent/work/938b0bdf6727d1ad/thirdparty")
+set(CMAKE_RELATIVE_PATH_TOP_BINARY "/home/teamcity/buildAgent/work/938b0bdf6727d1ad/build/lib.linux-x86_64-cpython-310/quasardb")
+
+# Force unix paths in dependencies.
+set(CMAKE_FORCE_UNIX_PATHS 1)
+
+
+# The C and CXX include file regular expressions for this directory.
+set(CMAKE_C_INCLUDE_REGEX_SCAN "^.*$")
+set(CMAKE_C_INCLUDE_REGEX_COMPLAIN "^$")
+set(CMAKE_CXX_INCLUDE_REGEX_SCAN ${CMAKE_C_INCLUDE_REGEX_SCAN})
+set(CMAKE_CXX_INCLUDE_REGEX_COMPLAIN ${CMAKE_C_INCLUDE_REGEX_COMPLAIN})

quasardb/pybind11/CMakeFiles/progress.marks

@@ -0,0 +1 @@
+0

quasardb/pybind11/Makefile

@@ -0,0 +1,189 @@
+# CMAKE generated file: DO NOT EDIT!
+# Generated by "Unix Makefiles" Generator, CMake Version 4.1
+
+# Default target executed when no arguments are given to make.
+default_target: all
+.PHONY : default_target
+
+# Allow only one "make -f Makefile2" at a time, but pass parallelism.
+.NOTPARALLEL:
+
+#=============================================================================
+# Special targets provided by cmake.
+
+# Disable implicit rules so canonical targets will work.
+.SUFFIXES:
+
+# Disable VCS-based implicit rules.
+% : %,v
+
+# Disable VCS-based implicit rules.
+% : RCS/%
+
+# Disable VCS-based implicit rules.
+% : RCS/%,v
+
+# Disable VCS-based implicit rules.
+% : SCCS/s.%
+
+# Disable VCS-based implicit rules.
+% : s.%
+
+.SUFFIXES: .hpux_make_needs_suffix_list
+
+# Command-line flag to silence nested $(MAKE).
+$(VERBOSE)MAKESILENT = -s
+
+#Suppress display of executed commands.
+$(VERBOSE).SILENT:
+
+# A target that is always out of date.
+cmake_force:
+.PHONY : cmake_force
+
+#=============================================================================
+# Set environment variables for the build.
+
+# The shell in which to execute make rules.
+SHELL = /bin/sh
+
+# The CMake executable.
+CMAKE_COMMAND = /home/teamcity/buildAgent/temp/buildTmp/build-env-rhi2owi3/lib/python3.10/site-packages/cmake/data/bin/cmake
+
+# The command to remove a file.
+RM = /home/teamcity/buildAgent/temp/buildTmp/build-env-rhi2owi3/lib/python3.10/site-packages/cmake/data/bin/cmake -E rm -f
+
+# Escaping for special characters.
+EQUALS = =
+
+# The top-level source directory on which CMake was run.
+CMAKE_SOURCE_DIR = /home/teamcity/buildAgent/work/938b0bdf6727d1ad/quasardb
+
+# The top-level build directory on which CMake was run.
+CMAKE_BINARY_DIR = /home/teamcity/buildAgent/work/938b0bdf6727d1ad/build/temp.linux-x86_64-cpython-310
+
+#=============================================================================
+# Targets provided globally by CMake.
+
+# Special rule for the target edit_cache
+edit_cache:
+	@$(CMAKE_COMMAND) -E cmake_echo_color "--switch=$(COLOR)" --cyan "No interactive CMake dialog available..."
+	/home/teamcity/buildAgent/temp/buildTmp/build-env-rhi2owi3/lib/python3.10/site-packages/cmake/data/bin/cmake -E echo No\ interactive\ CMake\ dialog\ available.
+.PHONY : edit_cache
+
+# Special rule for the target edit_cache
+edit_cache/fast: edit_cache
+.PHONY : edit_cache/fast
+
+# Special rule for the target rebuild_cache
+rebuild_cache:
+	@$(CMAKE_COMMAND) -E cmake_echo_color "--switch=$(COLOR)" --cyan "Running CMake to regenerate build system..."
+	/home/teamcity/buildAgent/temp/buildTmp/build-env-rhi2owi3/lib/python3.10/site-packages/cmake/data/bin/cmake --regenerate-during-build -S$(CMAKE_SOURCE_DIR) -B$(CMAKE_BINARY_DIR)
+.PHONY : rebuild_cache
+
+# Special rule for the target rebuild_cache
+rebuild_cache/fast: rebuild_cache
+.PHONY : rebuild_cache/fast
+
+# Special rule for the target list_install_components
+list_install_components:
+	@$(CMAKE_COMMAND) -E cmake_echo_color "--switch=$(COLOR)" --cyan "Available install components are: \"Unspecified\""
+.PHONY : list_install_components
+
+# Special rule for the target list_install_components
+list_install_components/fast: list_install_components
+.PHONY : list_install_components/fast
+
+# Special rule for the target install
+install: preinstall
+	@$(CMAKE_COMMAND) -E cmake_echo_color "--switch=$(COLOR)" --cyan "Install the project..."
+	/home/teamcity/buildAgent/temp/buildTmp/build-env-rhi2owi3/lib/python3.10/site-packages/cmake/data/bin/cmake -P cmake_install.cmake
+.PHONY : install
+
+# Special rule for the target install
+install/fast: preinstall/fast
+	@$(CMAKE_COMMAND) -E cmake_echo_color "--switch=$(COLOR)" --cyan "Install the project..."
+	/home/teamcity/buildAgent/temp/buildTmp/build-env-rhi2owi3/lib/python3.10/site-packages/cmake/data/bin/cmake -P cmake_install.cmake
+.PHONY : install/fast
+
+# Special rule for the target install/local
+install/local: preinstall
+	@$(CMAKE_COMMAND) -E cmake_echo_color "--switch=$(COLOR)" --cyan "Installing only the local directory..."
+	/home/teamcity/buildAgent/temp/buildTmp/build-env-rhi2owi3/lib/python3.10/site-packages/cmake/data/bin/cmake -DCMAKE_INSTALL_LOCAL_ONLY=1 -P cmake_install.cmake
+.PHONY : install/local
+
+# Special rule for the target install/local
+install/local/fast: preinstall/fast
+	@$(CMAKE_COMMAND) -E cmake_echo_color "--switch=$(COLOR)" --cyan "Installing only the local directory..."
+	/home/teamcity/buildAgent/temp/buildTmp/build-env-rhi2owi3/lib/python3.10/site-packages/cmake/data/bin/cmake -DCMAKE_INSTALL_LOCAL_ONLY=1 -P cmake_install.cmake
+.PHONY : install/local/fast
+
+# Special rule for the target install/strip
+install/strip: preinstall
+	@$(CMAKE_COMMAND) -E cmake_echo_color "--switch=$(COLOR)" --cyan "Installing the project stripped..."
+	/home/teamcity/buildAgent/temp/buildTmp/build-env-rhi2owi3/lib/python3.10/site-packages/cmake/data/bin/cmake -DCMAKE_INSTALL_DO_STRIP=1 -P cmake_install.cmake
+.PHONY : install/strip
+
+# Special rule for the target install/strip
+install/strip/fast: preinstall/fast
+	@$(CMAKE_COMMAND) -E cmake_echo_color "--switch=$(COLOR)" --cyan "Installing the project stripped..."
+	/home/teamcity/buildAgent/temp/buildTmp/build-env-rhi2owi3/lib/python3.10/site-packages/cmake/data/bin/cmake -DCMAKE_INSTALL_DO_STRIP=1 -P cmake_install.cmake
+.PHONY : install/strip/fast
+
+# The main all target
+all: cmake_check_build_system
+	cd /home/teamcity/buildAgent/work/938b0bdf6727d1ad/build/temp.linux-x86_64-cpython-310 && $(CMAKE_COMMAND) -E cmake_progress_start /home/teamcity/buildAgent/work/938b0bdf6727d1ad/build/temp.linux-x86_64-cpython-310/CMakeFiles /home/teamcity/buildAgent/work/938b0bdf6727d1ad/build/lib.linux-x86_64-cpython-310/quasardb/pybind11//CMakeFiles/progress.marks
+	cd /home/teamcity/buildAgent/work/938b0bdf6727d1ad/build/temp.linux-x86_64-cpython-310 && $(MAKE) $(MAKESILENT) -f CMakeFiles/Makefile2 /home/teamcity/buildAgent/work/938b0bdf6727d1ad/build/lib.linux-x86_64-cpython-310/quasardb/pybind11/all
+	$(CMAKE_COMMAND) -E cmake_progress_start /home/teamcity/buildAgent/work/938b0bdf6727d1ad/build/temp.linux-x86_64-cpython-310/CMakeFiles 0
+.PHONY : all
+
+# The main clean target
+clean:
+	cd /home/teamcity/buildAgent/work/938b0bdf6727d1ad/build/temp.linux-x86_64-cpython-310 && $(MAKE) $(MAKESILENT) -f CMakeFiles/Makefile2 /home/teamcity/buildAgent/work/938b0bdf6727d1ad/build/lib.linux-x86_64-cpython-310/quasardb/pybind11/clean
+.PHONY : clean
+
+# The main clean target
+clean/fast: clean
+.PHONY : clean/fast
+
+# Prepare targets for installation.
+preinstall: all
+	cd /home/teamcity/buildAgent/work/938b0bdf6727d1ad/build/temp.linux-x86_64-cpython-310 && $(MAKE) $(MAKESILENT) -f CMakeFiles/Makefile2 /home/teamcity/buildAgent/work/938b0bdf6727d1ad/build/lib.linux-x86_64-cpython-310/quasardb/pybind11/preinstall
+.PHONY : preinstall
+
+# Prepare targets for installation.
+preinstall/fast:
+	cd /home/teamcity/buildAgent/work/938b0bdf6727d1ad/build/temp.linux-x86_64-cpython-310 && $(MAKE) $(MAKESILENT) -f CMakeFiles/Makefile2 /home/teamcity/buildAgent/work/938b0bdf6727d1ad/build/lib.linux-x86_64-cpython-310/quasardb/pybind11/preinstall
+.PHONY : preinstall/fast
+
+# clear depends
+depend:
+	cd /home/teamcity/buildAgent/work/938b0bdf6727d1ad/build/temp.linux-x86_64-cpython-310 && $(CMAKE_COMMAND) -S$(CMAKE_SOURCE_DIR) -B$(CMAKE_BINARY_DIR) --check-build-system CMakeFiles/Makefile.cmake 1
+.PHONY : depend
+
+# Help Target
+help:
+	@echo "The following are some of the valid targets for this Makefile:"
+	@echo "... all (the default if no target is provided)"
+	@echo "... clean"
+	@echo "... depend"
+	@echo "... edit_cache"
+	@echo "... install"
+	@echo "... install/local"
+	@echo "... install/strip"
+	@echo "... list_install_components"
+	@echo "... rebuild_cache"
+.PHONY : help
+
+
+
+#=============================================================================
+# Special targets to cleanup operation of make.
+
+# Special rule to run CMake to check the build system integrity.
+# No rule that depends on this can have commands that come from listfiles
+# because they might be regenerated.
+cmake_check_build_system:
+	cd /home/teamcity/buildAgent/work/938b0bdf6727d1ad/build/temp.linux-x86_64-cpython-310 && $(CMAKE_COMMAND) -S$(CMAKE_SOURCE_DIR) -B$(CMAKE_BINARY_DIR) --check-build-system CMakeFiles/Makefile.cmake 0
+.PHONY : cmake_check_build_system
+

quasardb/pybind11/cmake_install.cmake

@@ -0,0 +1,50 @@
+# Install script for directory: /home/teamcity/buildAgent/work/938b0bdf6727d1ad/thirdparty/pybind11
+
+# Set the install prefix
+if(NOT DEFINED CMAKE_INSTALL_PREFIX)
+  set(CMAKE_INSTALL_PREFIX "/usr/local")
+endif()
+string(REGEX REPLACE "/$" "" CMAKE_INSTALL_PREFIX "${CMAKE_INSTALL_PREFIX}")
+
+# Set the install configuration name.
+if(NOT DEFINED CMAKE_INSTALL_CONFIG_NAME)
+  if(BUILD_TYPE)
+    string(REGEX REPLACE "^[^A-Za-z0-9_]+" ""
+           CMAKE_INSTALL_CONFIG_NAME "${BUILD_TYPE}")
+  else()
+    set(CMAKE_INSTALL_CONFIG_NAME "Release")
+  endif()
+  message(STATUS "Install configuration: \"${CMAKE_INSTALL_CONFIG_NAME}\"")
+endif()
+
+# Set the component getting installed.
+if(NOT CMAKE_INSTALL_COMPONENT)
+  if(COMPONENT)
+    message(STATUS "Install component: \"${COMPONENT}\"")
+    set(CMAKE_INSTALL_COMPONENT "${COMPONENT}")
+  else()
+    set(CMAKE_INSTALL_COMPONENT)
+  endif()
+endif()
+
+# Install shared libraries without execute permission?
+if(NOT DEFINED CMAKE_INSTALL_SO_NO_EXE)
+  set(CMAKE_INSTALL_SO_NO_EXE "0")
+endif()
+
+# Is this installation the result of a crosscompile?
+if(NOT DEFINED CMAKE_CROSSCOMPILING)
+  set(CMAKE_CROSSCOMPILING "FALSE")
+endif()
+
+# Set path to fallback-tool for dependency-resolution.
+if(NOT DEFINED CMAKE_OBJDUMP)
+  set(CMAKE_OBJDUMP "/opt/rh/gcc-toolset-14/root/usr/bin/objdump")
+endif()
+
+string(REPLACE ";" "\n" CMAKE_INSTALL_MANIFEST_CONTENT
+       "${CMAKE_INSTALL_MANIFEST_FILES}")
+if(CMAKE_INSTALL_LOCAL_ONLY)
+  file(WRITE "/home/teamcity/buildAgent/work/938b0bdf6727d1ad/build/lib.linux-x86_64-cpython-310/quasardb/pybind11/install_local_manifest.txt"
+     "${CMAKE_INSTALL_MANIFEST_CONTENT}")
+endif()

quasardb/quasardb/__init__.pyi

@@ -0,0 +1,97 @@
+import datetime
+
+from . import metrics
+from ._batch_column import BatchColumnInfo
+from ._batch_inserter import TimeSeriesBatch
+from ._blob import Blob
+from ._cluster import Cluster
+from ._continuous import QueryContinuous
+from ._double import Double
+from ._entry import Entry, ExpirableEntry
+from ._error import (
+    AliasAlreadyExistsError,
+    AliasNotFoundError,
+    AsyncPipelineFullError,
+    Error,
+    IncompatibleTypeError,
+    InputBufferTooSmallError,
+    InternalLocalError,
+    InvalidArgumentError,
+    InvalidDatetimeError,
+    InvalidHandleError,
+    InvalidQueryError,
+    NotImplementedError,
+    OutOfBoundsError,
+    TryAgainError,
+    UninitializedError,
+)
+from ._integer import Integer
+from ._node import DirectBlob, DirectInteger, Node
+from ._options import Options
+from ._perf import Perf
+from ._query import FindQuery
+from ._reader import Reader
+from ._retry import RetryOptions
+from ._string import String
+from ._table import ColumnInfo, ColumnType, IndexedColumnInfo, Table
+from ._tag import Tag
+from ._timestamp import Timestamp
+from ._writer import Writer, WriterData, WriterPushMode
+
+__all__ = [
+    "BatchColumnInfo",
+    "TimeSeriesBatch",
+    "Blob",
+    "Cluster",
+    "QueryContinuous",
+    "Double",
+    "Entry",
+    "ExpirableEntry",
+    "Error",
+    "AliasAlreadyExistsError",
+    "AliasNotFoundError",
+    "AsyncPipelineFullError",
+    "IncompatibleTypeError",
+    "InputBufferTooSmallError",
+    "InternalLocalError",
+    "InvalidArgumentError",
+    "InvalidDatetimeError",
+    "InvalidHandleError",
+    "InvalidQueryError",
+    "NotImplementedError",
+    "OutOfBoundsError",
+    "TryAgainError",
+    "UninitializedError",
+    "Integer",
+    "DirectBlob",
+    "DirectInteger",
+    "Node",
+    "Options",
+    "Perf",
+    "FindQuery",
+    "Reader",
+    "RetryOptions",
+    "String",
+    "ColumnInfo",
+    "ColumnType",
+    "IndexedColumnInfo",
+    "Table",
+    "Tag",
+    "Timestamp",
+    "Writer",
+    "WriterData",
+    "WriterPushMode",
+    "metrics",
+]
+
+never_expires: datetime.datetime = ...
+
+def build() -> str:
+    """
+    Return build number
+    """
+
+def version() -> str:
+    """
+    Return version number
+    """

quasardb/quasardb/_batch_column.pyi

@@ -0,0 +1,5 @@
+class BatchColumnInfo:
+    column: str
+    elements_count_hint: int
+    timeseries: str
+    def __init__(self, ts_name: str, col_name: str, size_hint: int = 0) -> None: ...