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

tigerbeetle/__init__.py

@@ -1,5 +1,5 @@
 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:

tigerbeetle/client.py

@@ -155,12 +155,6 @@ class Client:
             c_event_type=c_event_type,
             c_result_type=c_result_type)
 
-    def close(self) -> None:
-        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  # type: ignore[misc]
     def _c_on_completion(completion_ctx: int, packet: Any, timestamp: int, bytes_ptr: Any, len_: int) -> None:
@@ -200,12 +194,6 @@ class Client:
 
         inflight_packet.on_completion(inflight_packet)
 
-    def __enter__(self) -> Self:
-        return self
-
-    def __exit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> None:
-        self.close()
-
 
 class ClientSync(Client, bindings.StateMachineMixin):
     def _on_completion(self, inflight_packet: InflightPacket) -> None:
@@ -235,6 +223,19 @@ 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: InflightPacket) -> None:
@@ -280,6 +281,32 @@ 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]
+
+    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:

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tigerbeetle
-Version: 0.16.56
+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.56.dist-info/RECORD

@@ -1,15 +0,0 @@
-tigerbeetle/__init__.py,sha256=L9lyq47kzyZO0uXR455LXZTXWrt7yxfs-pFs8Mfv-r4,944
-tigerbeetle/bindings.py,sha256=NSND7UuFpnd4TRBDA9aDJLRS115QNi3m0dVksekVlZg,28155
-tigerbeetle/client.py,sha256=QaHWXeDvLYGia1IOUArR5rNShka9y3GwveFELCcoXwY,10967
-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=qPbLSTvbDH49ePtGNuxciy9QYrd_LGSsRkMeEJNNa98,692104
-tigerbeetle/lib/aarch64-linux-musl/libtb_client.so,sha256=BZ6a2nbclS4g0HU4495b9DipBefrLEMvZgUD0Jpi4Os,690384
-tigerbeetle/lib/aarch64-macos/libtb_client.dylib,sha256=zEAJLzDJjIBhSG2tgWww39zq7Rl-Qsf5Q4lmD6UX2M8,700224
-tigerbeetle/lib/x86_64-linux-gnu.2.27/libtb_client.so,sha256=EgoOoAELkFc-nHgHY-AlTViVO2Sx35EqF4RbFr5lMZY,815368
-tigerbeetle/lib/x86_64-linux-musl/libtb_client.so,sha256=zoxZYw_eH0l2GriVIGr1c8_hdNk33TyVCKhbttYnKGM,813816
-tigerbeetle/lib/x86_64-macos/libtb_client.dylib,sha256=KQaKRbtqzsIrzBcJxejIU0b8E9EaKzXgxz_MF7BpP98,792558
-tigerbeetle/lib/x86_64-windows/tb_client.dll,sha256=dtB6y1LLLrI9swXI98MS8kKxjuMf0rHUXBauqXhkjMc,663552
-tigerbeetle-0.16.56.dist-info/METADATA,sha256=70k1J6tzY0I04K4xqCKjDCAB8wYI3uict2-j79Ym3xg,23268
-tigerbeetle-0.16.56.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-tigerbeetle-0.16.56.dist-info/RECORD,,