tigerbeetle 0.16.55__py3-none-any.whl → 0.16.57__py3-none-any.whl

tigerbeetle/__init__.py

@@ -1,3 +1,39 @@
 from .bindings import * # noqa
-from .client import ClientAsync, ClientSync, id, amount_max, configure_logging # noqa
+from .client import ClientAsync, ClientSync, id, amount_max, configure_logging, PacketError # noqa
 from .lib import IntegerOverflowError, NativeError
+
+# Explicitly declare public exports:
+__all__ = [
+    # from .client:
+    "ClientAsync",
+    "ClientSync",
+    "id",
+    "amount_max",
+    "configure_logging",
+    # from .lib:
+    "IntegerOverflowError",
+    "NativeError",
+    # from .bindings - everything treated as public:
+    "Operation",
+    "PacketStatus",
+    "InitStatus",
+    "ClientStatus",
+    "LogLevel",
+    "RegisterLogCallbackStatus",
+    "AccountFlags",
+    "TransferFlags",
+    "AccountFilterFlags",
+    "QueryFilterFlags",
+    "CreateAccountResult",
+    "CreateTransferResult",
+    "Account",
+    "Transfer",
+    "CreateAccountsResult",
+    "CreateTransfersResult",
+    "AccountFilter",
+    "AccountBalance",
+    "QueryFilter",
+    "InitParameters",
+    "AsyncStateMachineMixin",
+    "StateMachineMixin",
+]

tigerbeetle/bindings.py

@@ -6,10 +6,21 @@ from __future__ import annotations
 
 import ctypes
 import enum
+import sys
+from dataclasses import dataclass
 from collections.abc import Callable # noqa: TCH003
 from typing import Any
+if sys.version_info >= (3, 11):
+    from typing import Self
+else:
+    from typing_extensions import Self
 
-from .lib import c_uint128, dataclass, tbclient, validate_uint
+from .lib import c_uint128, tbclient, validate_uint
+
+# Use slots=True if the version of Python is new enough (3.10+) to support it.
+if sys.version_info >= (3, 10):
+    # mypy: ignore assignment (3.10+) and unused-ignore (pre 3.10)
+    dataclass = dataclass(slots=True) # type: ignore[assignment, unused-ignore]
 
 
 class Operation(enum.IntEnum):
@@ -236,13 +247,13 @@ class Transfer:
 @dataclass
 class CreateAccountsResult:
     index: int = 0
-    result: CreateAccountResult = 0
+    result: CreateAccountResult = CreateAccountResult.OK
 
 
 @dataclass
 class CreateTransfersResult:
     index: int = 0
-    result: CreateTransferResult = 0
+    result: CreateTransferResult = CreateTransferResult.OK
 
 
 @dataclass
@@ -282,7 +293,7 @@ class QueryFilter:
 
 class CPacket(ctypes.Structure):
     @classmethod
-    def from_param(cls, obj):
+    def from_param(cls, obj: Any) -> Self:
         validate_uint(bits=32, name="data_size", number=obj.data_size)
         validate_uint(bits=16, name="user_tag", number=obj.user_tag)
         validate_uint(bits=8, name="operation", number=obj.operation)
@@ -309,7 +320,7 @@ CPacket._fields_ = [ # noqa: SLF001
 
 class CClient(ctypes.Structure):
     @classmethod
-    def from_param(cls, obj):
+    def from_param(cls, obj: Any) -> Self:
         return cls(
             opaque=obj.opaque,
         )
@@ -321,7 +332,7 @@ CClient._fields_ = [ # noqa: SLF001
 
 class CAccount(ctypes.Structure):
     @classmethod
-    def from_param(cls, obj):
+    def from_param(cls, obj: Any) -> Self:
         validate_uint(bits=128, name="id", number=obj.id)
         validate_uint(bits=128, name="debits_pending", number=obj.debits_pending)
         validate_uint(bits=128, name="debits_posted", number=obj.debits_posted)
@@ -349,7 +360,7 @@ class CAccount(ctypes.Structure):
         )
 
 
-    def to_python(self):
+    def to_python(self) -> Account:
         return Account(
             id=self.id.to_python(),
             debits_pending=self.debits_pending.to_python(),
@@ -384,7 +395,7 @@ CAccount._fields_ = [ # noqa: SLF001
 
 class CTransfer(ctypes.Structure):
     @classmethod
-    def from_param(cls, obj):
+    def from_param(cls, obj: Any) -> Self:
         validate_uint(bits=128, name="id", number=obj.id)
         validate_uint(bits=128, name="debit_account_id", number=obj.debit_account_id)
         validate_uint(bits=128, name="credit_account_id", number=obj.credit_account_id)
@@ -414,7 +425,7 @@ class CTransfer(ctypes.Structure):
         )
 
 
-    def to_python(self):
+    def to_python(self) -> Transfer:
         return Transfer(
             id=self.id.to_python(),
             debit_account_id=self.debit_account_id.to_python(),
@@ -450,7 +461,7 @@ CTransfer._fields_ = [ # noqa: SLF001
 
 class CCreateAccountsResult(ctypes.Structure):
     @classmethod
-    def from_param(cls, obj):
+    def from_param(cls, obj: Any) -> Self:
         validate_uint(bits=32, name="index", number=obj.index)
         return cls(
             index=obj.index,
@@ -458,7 +469,7 @@ class CCreateAccountsResult(ctypes.Structure):
         )
 
 
-    def to_python(self):
+    def to_python(self) -> CreateAccountsResult:
         return CreateAccountsResult(
             index=self.index,
             result=CreateAccountResult(self.result),
@@ -472,7 +483,7 @@ CCreateAccountsResult._fields_ = [ # noqa: SLF001
 
 class CCreateTransfersResult(ctypes.Structure):
     @classmethod
-    def from_param(cls, obj):
+    def from_param(cls, obj: Any) -> Self:
         validate_uint(bits=32, name="index", number=obj.index)
         return cls(
             index=obj.index,
@@ -480,7 +491,7 @@ class CCreateTransfersResult(ctypes.Structure):
         )
 
 
-    def to_python(self):
+    def to_python(self) -> CreateTransfersResult:
         return CreateTransfersResult(
             index=self.index,
             result=CreateTransferResult(self.result),
@@ -494,7 +505,7 @@ CCreateTransfersResult._fields_ = [ # noqa: SLF001
 
 class CAccountFilter(ctypes.Structure):
     @classmethod
-    def from_param(cls, obj):
+    def from_param(cls, obj: Any) -> Self:
         validate_uint(bits=128, name="account_id", number=obj.account_id)
         validate_uint(bits=128, name="user_data_128", number=obj.user_data_128)
         validate_uint(bits=64, name="user_data_64", number=obj.user_data_64)
@@ -516,7 +527,7 @@ class CAccountFilter(ctypes.Structure):
         )
 
 
-    def to_python(self):
+    def to_python(self) -> AccountFilter:
         return AccountFilter(
             account_id=self.account_id.to_python(),
             user_data_128=self.user_data_128.to_python(),
@@ -545,7 +556,7 @@ CAccountFilter._fields_ = [ # noqa: SLF001
 
 class CAccountBalance(ctypes.Structure):
     @classmethod
-    def from_param(cls, obj):
+    def from_param(cls, obj: Any) -> Self:
         validate_uint(bits=128, name="debits_pending", number=obj.debits_pending)
         validate_uint(bits=128, name="debits_posted", number=obj.debits_posted)
         validate_uint(bits=128, name="credits_pending", number=obj.credits_pending)
@@ -560,7 +571,7 @@ class CAccountBalance(ctypes.Structure):
         )
 
 
-    def to_python(self):
+    def to_python(self) -> AccountBalance:
         return AccountBalance(
             debits_pending=self.debits_pending.to_python(),
             debits_posted=self.debits_posted.to_python(),
@@ -581,7 +592,7 @@ CAccountBalance._fields_ = [ # noqa: SLF001
 
 class CQueryFilter(ctypes.Structure):
     @classmethod
-    def from_param(cls, obj):
+    def from_param(cls, obj: Any) -> Self:
         validate_uint(bits=128, name="user_data_128", number=obj.user_data_128)
         validate_uint(bits=64, name="user_data_64", number=obj.user_data_64)
         validate_uint(bits=32, name="user_data_32", number=obj.user_data_32)
@@ -603,7 +614,7 @@ class CQueryFilter(ctypes.Structure):
         )
 
 
-    def to_python(self):
+    def to_python(self) -> QueryFilter:
         return QueryFilter(
             user_data_128=self.user_data_128.to_python(),
             user_data_64=self.user_data_64,
@@ -678,13 +689,13 @@ tb_client_submit.argtypes = [ctypes.POINTER(CClient), ctypes.POINTER(CPacket)]
 tb_client_register_log_callback = tbclient.tb_client_register_log_callback
 tb_client_register_log_callback.restype = RegisterLogCallbackStatus
 # Need to pass in None to clear - ctypes will error if argtypes is set.
-#tb_client_register_log_callback.argtypes = [LogHandler, ctypes.c_bool]
+# tb_client_register_log_callback.argtypes = [LogHandler, ctypes.c_bool]
 
 
 class AsyncStateMachineMixin:
     _submit: Callable[[Operation, Any, Any, Any], Any]
     async def create_accounts(self, accounts: list[Account]) -> list[CreateAccountsResult]:
-        return await self._submit(
+        return await self._submit(  # type: ignore[no-any-return]
             Operation.CREATE_ACCOUNTS,
             accounts,
             CAccount,
@@ -692,7 +703,7 @@ class AsyncStateMachineMixin:
         )
 
     async def create_transfers(self, transfers: list[Transfer]) -> list[CreateTransfersResult]:
-        return await self._submit(
+        return await self._submit(  # type: ignore[no-any-return]
             Operation.CREATE_TRANSFERS,
             transfers,
             CTransfer,
@@ -700,7 +711,7 @@ class AsyncStateMachineMixin:
         )
 
     async def lookup_accounts(self, accounts: list[int]) -> list[Account]:
-        return await self._submit(
+        return await self._submit(  # type: ignore[no-any-return]
             Operation.LOOKUP_ACCOUNTS,
             accounts,
             c_uint128,
@@ -708,7 +719,7 @@ class AsyncStateMachineMixin:
         )
 
     async def lookup_transfers(self, transfers: list[int]) -> list[Transfer]:
-        return await self._submit(
+        return await self._submit(  # type: ignore[no-any-return]
             Operation.LOOKUP_TRANSFERS,
             transfers,
             c_uint128,
@@ -716,7 +727,7 @@ class AsyncStateMachineMixin:
         )
 
     async def get_account_transfers(self, filter: AccountFilter) -> list[Transfer]:
-        return await self._submit(
+        return await self._submit(  # type: ignore[no-any-return]
             Operation.GET_ACCOUNT_TRANSFERS,
             [filter],
             CAccountFilter,
@@ -724,7 +735,7 @@ class AsyncStateMachineMixin:
         )
 
     async def get_account_balances(self, filter: AccountFilter) -> list[AccountBalance]:
-        return await self._submit(
+        return await self._submit(  # type: ignore[no-any-return]
             Operation.GET_ACCOUNT_BALANCES,
             [filter],
             CAccountFilter,
@@ -732,7 +743,7 @@ class AsyncStateMachineMixin:
         )
 
     async def query_accounts(self, query_filter: QueryFilter) -> list[Account]:
-        return await self._submit(
+        return await self._submit(  # type: ignore[no-any-return]
             Operation.QUERY_ACCOUNTS,
             [query_filter],
             CQueryFilter,
@@ -740,7 +751,7 @@ class AsyncStateMachineMixin:
         )
 
     async def query_transfers(self, query_filter: QueryFilter) -> list[Transfer]:
-        return await self._submit(
+        return await self._submit(  # type: ignore[no-any-return]
             Operation.QUERY_TRANSFERS,
             [query_filter],
             CQueryFilter,
@@ -752,7 +763,7 @@ class AsyncStateMachineMixin:
 class StateMachineMixin:
     _submit: Callable[[Operation, Any, Any, Any], Any]
     def create_accounts(self, accounts: list[Account]) -> list[CreateAccountsResult]:
-        return self._submit(
+        return self._submit(  # type: ignore[no-any-return]
             Operation.CREATE_ACCOUNTS,
             accounts,
             CAccount,
@@ -760,7 +771,7 @@ class StateMachineMixin:
         )
 
     def create_transfers(self, transfers: list[Transfer]) -> list[CreateTransfersResult]:
-        return self._submit(
+        return self._submit(  # type: ignore[no-any-return]
             Operation.CREATE_TRANSFERS,
             transfers,
             CTransfer,
@@ -768,7 +779,7 @@ class StateMachineMixin:
         )
 
     def lookup_accounts(self, accounts: list[int]) -> list[Account]:
-        return self._submit(
+        return self._submit(  # type: ignore[no-any-return]
             Operation.LOOKUP_ACCOUNTS,
             accounts,
             c_uint128,
@@ -776,7 +787,7 @@ class StateMachineMixin:
         )
 
     def lookup_transfers(self, transfers: list[int]) -> list[Transfer]:
-        return self._submit(
+        return self._submit(  # type: ignore[no-any-return]
             Operation.LOOKUP_TRANSFERS,
             transfers,
             c_uint128,
@@ -784,7 +795,7 @@ class StateMachineMixin:
         )
 
     def get_account_transfers(self, filter: AccountFilter) -> list[Transfer]:
-        return self._submit(
+        return self._submit(  # type: ignore[no-any-return]
             Operation.GET_ACCOUNT_TRANSFERS,
             [filter],
             CAccountFilter,
@@ -792,7 +803,7 @@ class StateMachineMixin:
         )
 
     def get_account_balances(self, filter: AccountFilter) -> list[AccountBalance]:
-        return self._submit(
+        return self._submit(  # type: ignore[no-any-return]
             Operation.GET_ACCOUNT_BALANCES,
             [filter],
             CAccountFilter,
@@ -800,7 +811,7 @@ class StateMachineMixin:
         )
 
     def query_accounts(self, query_filter: QueryFilter) -> list[Account]:
-        return self._submit(
+        return self._submit(  # type: ignore[no-any-return]
             Operation.QUERY_ACCOUNTS,
             [query_filter],
             CQueryFilter,
@@ -808,7 +819,7 @@ class StateMachineMixin:
         )
 
     def query_transfers(self, query_filter: QueryFilter) -> list[Transfer]:
-        return self._submit(
+        return self._submit(  # type: ignore[no-any-return]
             Operation.QUERY_TRANSFERS,
             [query_filter],
             CQueryFilter,

tigerbeetle/client.py

@@ -4,11 +4,16 @@ import asyncio
 import ctypes
 import logging
 import os
+import sys
 import threading
 import time
 from collections.abc import Callable  # noqa: TCH003
 from dataclasses import dataclass
 from typing import Any
+if sys.version_info >= (3, 11):
+    from typing import Self
+else:
+    from typing_extensions import Self
 
 from . import bindings
 from .lib import tb_assert, c_uint128
@@ -17,11 +22,11 @@ logger = logging.getLogger("tigerbeetle")
 
 
 class AtomicInteger:
-    def __init__(self, value=0):
+    def __init__(self, value: int = 0) -> None:
         self._value = value
         self._lock = threading.Lock()
 
-    def increment(self):
+    def increment(self) -> int:
         with self._lock:
             self._value += 1
             return self._value
@@ -45,30 +50,43 @@ class InflightPacket:
     operation: bindings.Operation
     c_event_type: Any
     c_result_type: Any
-    on_completion: Callable | None
+    on_completion: Callable[[Self], None] | None
     on_completion_context: CompletionContextSync | CompletionContextAsync | None
 
+class _IDGenerator:
+    """
+    Generator for Universally Unique and Sortable Identifiers as a 128-bit integers, based on ULIDs.
+
+    Keeps a monotonically increasing millisecond timestamp between calls to `.generate()`.
+    """
+    def __init__(self) -> None:
+        self._time_ms_last = time.time_ns() // (1000 * 1000)
+
+    def generate(self) -> int:
+        time_ms = time.time_ns() // (1000 * 1000)
+
+        # Ensure time_ms monotonically increases.
+        if time_ms <= self._time_ms_last:
+            time_ms = self._time_ms_last
+        else:
+            self._time_ms_last = time_ms
+
+        randomness = os.urandom(10)
+
+        return int.from_bytes(
+            time_ms.to_bytes(6, "big") + randomness,
+            "big",
+        )
+
+# Module-level singleton instance.
+_id_generator = _IDGenerator()
 
 
 def id() -> int:
     """
     Generates a Universally Unique and Sortable Identifier as a 128-bit integer. Based on ULIDs.
     """
-    time_ms = time.time_ns() // (1000 * 1000)
-
-    # Ensure time_ms monotonically increases.
-    time_ms_last = getattr(id, "_time_ms_last", 0)
-    if time_ms <= time_ms_last:
-        time_ms = time_ms_last
-    else:
-        id._time_ms_last = time_ms
-
-    randomness = os.urandom(10)
-
-    return int.from_bytes(
-        time_ms.to_bytes(6, "big") + randomness,
-        "big",
-    )
+    return _id_generator.generate()
 
 
 amount_max = (2 ** 128) - 1
@@ -137,15 +155,9 @@ class Client:
             c_event_type=c_event_type,
             c_result_type=c_result_type)
 
-    def close(self):
-        bindings.tb_client_deinit(ctypes.byref(self._client))
-        tb_assert(self._client is not None)
-        tb_assert(len(self._inflight_packets) == 0)
-        del Client._clients[self._client_key]
-
     @staticmethod
-    @bindings.OnCompletion
-    def _c_on_completion(completion_ctx, packet, timestamp, bytes_ptr, len_):
+    @bindings.OnCompletion  # type: ignore[misc]
+    def _c_on_completion(completion_ctx: int, packet: Any, timestamp: int, bytes_ptr: Any, len_: int) -> None:
         """
         Invoked in a separate thread
         """
@@ -156,7 +168,9 @@ class Client:
 
         if packet[0].status != bindings.PacketStatus.OK.value:
             inflight_packet.response = PacketError(repr(bindings.PacketStatus(packet[0].status)))
-            tb_assert(inflight_packet.on_completion is not None)
+            if inflight_packet.on_completion is None:
+                # Can't use tb_assert here, as mypy complains later that it might be None.
+                raise TypeError("inflight_packet.on_completion not set")
             inflight_packet.on_completion(inflight_packet)
             return
 
@@ -174,22 +188,21 @@ class Client:
 
         inflight_packet.response = results
 
-        tb_assert(inflight_packet.on_completion is not None)
-        inflight_packet.on_completion(inflight_packet)
-
-    def __enter__(self):
-        return self
+        if inflight_packet.on_completion is None:
+            # Can't use tb_assert here, as mypy complains later that it might be None.
+            raise TypeError("inflight_packet.on_completion not set")
 
-    def __exit__(self, exc_type, exc_val, exc_tb):
-        self.close()
+        inflight_packet.on_completion(inflight_packet)
 
 
 class ClientSync(Client, bindings.StateMachineMixin):
-    def _on_completion(self, inflight_packet):
+    def _on_completion(self, inflight_packet: InflightPacket) -> None:
+        if not isinstance(inflight_packet.on_completion_context, CompletionContextSync):
+            raise TypeError(repr(inflight_packet.on_completion_context))
         inflight_packet.on_completion_context.event.set()
 
     def _submit(self, operation: bindings.Operation, operations: list[Any],
-                c_event_type: Any, c_result_type: Any):
+                c_event_type: Any, c_result_type: Any) -> Any:
         inflight_packet = self._acquire_packet(operation, operations, c_event_type, c_result_type)
         self._inflight_packets[inflight_packet.packet.user_data] = inflight_packet
 
@@ -210,24 +223,41 @@ class ClientSync(Client, bindings.StateMachineMixin):
 
         return inflight_packet.response
 
+    def close(self) -> None:
+        tb_assert(self._client is not None)
+        bindings.tb_client_deinit(ctypes.byref(self._client))
+
+        tb_assert(len(self._inflight_packets) == 0)
+        del Client._clients[self._client_key]
+
+    def __enter__(self) -> Self:
+        return self
+
+    def __exit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> None:
+        self.close()
+
 
 class ClientAsync(Client, bindings.AsyncStateMachineMixin):
-    def _on_completion(self, inflight_packet):
+    def _on_completion(self, inflight_packet: InflightPacket) -> None:
         """
         Called by Client._c_on_completion, which itself is called from a different thread. Use
         `call_soon_threadsafe` to return to the thread of the event loop the request was invoked
         from, so _trigger_event() can trigger the async event and allow the client to progress.
         """
+        if not isinstance(inflight_packet.on_completion_context, CompletionContextAsync):
+            raise TypeError(repr(inflight_packet.on_completion_context))
         inflight_packet.on_completion_context.loop.call_soon_threadsafe(
             self._trigger_event,
             inflight_packet
         )
 
-    def _trigger_event(self, inflight_packet):
+    def _trigger_event(self, inflight_packet:InflightPacket) -> None:
+        if not isinstance(inflight_packet.on_completion_context, CompletionContextAsync):
+            raise TypeError(repr(inflight_packet.on_completion_context))
         inflight_packet.on_completion_context.event.set()
 
     async def _submit(self, operation: bindings.Operation, operations: Any,
-                      c_event_type: Any, c_result_type: Any):
+                      c_event_type: Any, c_result_type: Any) -> Any:
         inflight_packet = self._acquire_packet(operation, operations, c_event_type, c_result_type)
         self._inflight_packets[inflight_packet.packet.user_data] = inflight_packet
 
@@ -251,9 +281,35 @@ class ClientAsync(Client, bindings.AsyncStateMachineMixin):
 
         return inflight_packet.response
 
+    async def close(self) -> None:
+        tb_assert(self._client is not None)
+        bindings.tb_client_deinit(ctypes.byref(self._client))
+
+        # tb_client_deinit internally clears any inflight requests, and calls their callbacks, so
+        # the client needs to stick around until that's done.
+        #
+        # This isn't a problem for ClientSync, but ClientAsync invokes the callbacks using
+        # call_soon_threadsafe(), so wait for the event to fire on all pending packets.
+        all_events = []
+        for packet in self._inflight_packets.values():
+            if not isinstance(packet.on_completion_context, CompletionContextAsync):
+                raise AssertionError()
+            all_events.append(packet.on_completion_context.event.wait())
+
+        await asyncio.gather(*all_events)
+
+        tb_assert(len(self._inflight_packets) == 0)
+        del Client._clients[self._client_key]
 
-@bindings.LogHandler
-def log_handler(level_zig, message_ptr, message_len):
+    async def __aenter__(self) -> Self:
+        return self
+
+    async def __aexit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> None:
+        await self.close()
+
+
+@bindings.LogHandler  # type: ignore[misc]
+def log_handler(level_zig: bindings.LogLevel, message_ptr: Any, message_len: int) -> None:
     level_python = {
         bindings.LogLevel.ERR: logging.ERROR,
         bindings.LogLevel.WARN: logging.WARNING,
@@ -265,10 +321,15 @@ def log_handler(level_zig, message_ptr, message_len):
 tb_assert(bindings.tb_client_register_log_callback(log_handler, True) ==
     bindings.RegisterLogCallbackStatus.SUCCESS)
 
-def configure_logging(*, debug, log_handler=log_handler):
+
+def configure_logging(
+    *,
+    debug: bool,
+    handler: Callable[[bindings.LogLevel, Any, int], None] = log_handler,
+) -> None:
     # First disable the existing log handler, before enabling the new one.
     tb_assert(bindings.tb_client_register_log_callback(None, debug) ==
         bindings.RegisterLogCallbackStatus.SUCCESS)
 
-    tb_assert(bindings.tb_client_register_log_callback(log_handler, debug) ==
+    tb_assert(bindings.tb_client_register_log_callback(handler, debug) ==
         bindings.RegisterLogCallbackStatus.SUCCESS)

tigerbeetle/lib.py

@@ -1,17 +1,23 @@
 import ctypes
-import dataclasses
 import platform
+import sys
 from pathlib import Path
+from typing import Any
+if sys.version_info >= (3, 11):
+    from typing import Self
+else:
+    from typing_extensions import Self
 
 
 class NativeError(Exception):
     pass
 
+
 class IntegerOverflowError(ValueError):
     pass
 
 
-def _load_tbclient():
+def _load_tbclient() -> ctypes.CDLL:
     prefix = ""
     arch = ""
     system = ""
@@ -50,39 +56,35 @@ def _load_tbclient():
 
     source_path = Path(__file__)
     source_dir = source_path.parent
-    library_path = source_dir / "lib" / f"{arch}-{system}{linux_libc}" / f"{prefix}tb_client{suffix}"
+    library_path = (
+        source_dir / "lib" / f"{arch}-{system}{linux_libc}" / f"{prefix}tb_client{suffix}"
+    )
     return ctypes.CDLL(str(library_path))
 
 
-def validate_uint(*, bits: int, name: str, number: int):
+def validate_uint(*, bits: int, name: str, number: int) -> None:
     if number > 2**bits - 1:
         raise IntegerOverflowError(f"{name}=={number} is too large to fit in {bits} bits")
     if number < 0:
         raise IntegerOverflowError(f"{name}=={number} cannot be negative")
 
 
-class c_uint128(ctypes.Structure): # noqa: N801
-    _fields_ = [("_low", ctypes.c_uint64), ("_high", ctypes.c_uint64)] # noqa: RUF012
+class c_uint128(ctypes.Structure):  # noqa: N801
+    _fields_ = [("_low", ctypes.c_uint64), ("_high", ctypes.c_uint64)]  # noqa: RUF012
 
     @classmethod
-    def from_param(cls, obj):
-        return cls(_high=obj >> 64, _low=obj & 0xffffffffffffffff)
-
-    def to_python(self):
-        return self._high << 64 | self._low
+    def from_param(cls, obj: int) -> Self:
+        return cls(_high=obj >> 64, _low=obj & 0xFFFFFFFFFFFFFFFF)
 
+    def to_python(self) -> int:
+        return int(self._high << 64 | self._low)
 
-# Use slots=True if the version of Python is new enough (3.10+) to support it.
-try:
-    dataclass = dataclasses.dataclass(slots=True)
-except TypeError:
-    dataclass = dataclasses.dataclass()
-
-def tb_assert(value):
+def tb_assert(value: Any) -> None:
     """
     Python's built-in assert can be silently disabled if Python is run with -O.
     """
     if not value:
         raise AssertionError()
 
+
 tbclient = _load_tbclient()

{tigerbeetle-0.16.55.dist-info → tigerbeetle-0.16.57.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tigerbeetle
-Version: 0.16.55
+Version: 0.16.57
 Summary: The TigerBeetle client for Python.
 Project-URL: Homepage, https://github.com/tigerbeetle/tigerbeetle
 Project-URL: Issues, https://github.com/tigerbeetle/tigerbeetle/issues
@@ -91,6 +91,11 @@ environment variable and defaults to port `3000`.
 with tb.ClientSync(cluster_id=0, replica_addresses=os.getenv("TB_ADDRESS", "3000")) as client:
     # Use the client.
     pass
+
+# Alternatively:
+async with tb.ClientAsync(cluster_id=0, replica_addresses=os.getenv("TB_ADDRESS", "3000")) as client:
+    # Use the client, async!
+    pass
 ```
 
 The following are valid addresses:

tigerbeetle-0.16.57.dist-info/RECORD

@@ -0,0 +1,15 @@
+tigerbeetle/__init__.py,sha256=vrZ9rqFxHpXtACa9qDhwKwSkhv3CGhAvg7vF3Ges8c0,957
+tigerbeetle/bindings.py,sha256=NSND7UuFpnd4TRBDA9aDJLRS115QNi3m0dVksekVlZg,28155
+tigerbeetle/client.py,sha256=iLvRvUk-_YNip2ZIzPV3C5BbqjvlYa4vHDXbQpAoDnU,12048
+tigerbeetle/lib.py,sha256=1Qln-Z4ZoCHJJ43sXlJJEs7-ZY5aZJjNNu7eISooAK8,2432
+tigerbeetle/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+tigerbeetle/lib/aarch64-linux-gnu.2.27/libtb_client.so,sha256=9Ni-kr77uva3dIxwq6g89rxdbC0LuiDPbXYu7d8ZDrQ,693336
+tigerbeetle/lib/aarch64-linux-musl/libtb_client.so,sha256=o3VknLRJjkZpBWlPO33KBclshNtrX7teHgvixKUiD1A,691616
+tigerbeetle/lib/aarch64-macos/libtb_client.dylib,sha256=zqXLlUX1V2yJdG_8YbdVux7LsIFykZuuf2hjdgrsMBE,700224
+tigerbeetle/lib/x86_64-linux-gnu.2.27/libtb_client.so,sha256=9WKWUV7QqmntjceqIu8-iD90zONv6Y2HW8riHYwyFyE,816808
+tigerbeetle/lib/x86_64-linux-musl/libtb_client.so,sha256=6d7__mj4Vfc_WrHG6HuCBeCrE8g__awoQnlQvenG3b4,815256
+tigerbeetle/lib/x86_64-macos/libtb_client.dylib,sha256=3MX0n9AdFfq3pjXgahavPiob6DEfEGA8-xz_pOoOYB4,792590
+tigerbeetle/lib/x86_64-windows/tb_client.dll,sha256=R53xX0nn3hTgVoeR8jQLL8w7M5d00jbZf5IyaVoD_ek,664064
+tigerbeetle-0.16.57.dist-info/METADATA,sha256=1EUIAa2ekf9x41mGf9dLHGaRIsu5ZW0b3ILqnK4Ac-E,23426
+tigerbeetle-0.16.57.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+tigerbeetle-0.16.57.dist-info/RECORD,,

tigerbeetle-0.16.55.dist-info/RECORD

@@ -1,14 +0,0 @@
-tigerbeetle/__init__.py,sha256=3cmov0HPkByJw1znLKoTqIuZUEeEQMEg1fflWY3ty9A,168
-tigerbeetle/bindings.py,sha256=syIRjAQQfoE15iGK6kxmJLToRdU2KWDY0nnu6u-t6xk,26978
-tigerbeetle/client.py,sha256=oGg1vAFqI8UZ5awI3sshZfxUdBsroUKwgp2rUa75lFc,9255
-tigerbeetle/lib.py,sha256=JKreDxdw_YejtH7F135_m6I6mNBSwjokaI6QicKm7cc,2425
-tigerbeetle/lib/aarch64-linux-gnu.2.27/libtb_client.so,sha256=bEgYJ674Nu-naxUASpdvpTVKAetinNZ9j4jdIVIAs4U,692104
-tigerbeetle/lib/aarch64-linux-musl/libtb_client.so,sha256=g64c-V-NJ1WQA_gfDpBlaSvJ0IUOsAemvp-SVhZ_cCk,690384
-tigerbeetle/lib/aarch64-macos/libtb_client.dylib,sha256=AMGOyDarLDXbIfKq5hQu6tCrY-tWBLBSQ3f21lc2P74,700224
-tigerbeetle/lib/x86_64-linux-gnu.2.27/libtb_client.so,sha256=ZecvJ8l-cn6-ud9nAQa9r4ATOfFQ7slKYPOw7MwpvnE,815368
-tigerbeetle/lib/x86_64-linux-musl/libtb_client.so,sha256=K1XGJ5n6uw2aW8RIcdhxQcNGcic1zTbkbAwPJiUs6qw,813816
-tigerbeetle/lib/x86_64-macos/libtb_client.dylib,sha256=mFQ3CxmCYfQuLcnHkHmg6DxlewPM5G_TGpRU4fo5Ldg,792558
-tigerbeetle/lib/x86_64-windows/tb_client.dll,sha256=KFrZ8XK2gTlzoCFOMfcRoGk8i3LA3MBiyQLuz2PSBYc,663552
-tigerbeetle-0.16.55.dist-info/METADATA,sha256=kuhGPnre8bPfIbC9EQbWz6OLSU_3pOYwwIarE3ccMbY,23268
-tigerbeetle-0.16.55.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-tigerbeetle-0.16.55.dist-info/RECORD,,