VER: Release 0.52.0

See release notes.

Author: nmacholl

.github/workflows/test.yml

@@ -32,11 +32,11 @@ jobs:
       - name: Test (release)
         timeout-minutes: 3
         if: ${{ github.ref == 'refs/heads/main' }}
-        run: scripts/test.sh --release
+        run: scripts/test.sh -vvv --release
         shell: bash
 
       - name: Test
         timeout-minutes: 3
         if: ${{ github.ref != 'refs/heads/main' }}
-        run: scripts/test.sh
+        run: scripts/test.sh -vvv
         shell: bash

CHANGELOG.md

@@ -1335,7 +1335,7 @@ def __init__(
         count: int | None = None,
     ) -> None:
         self._reader = reader
-        self._dtype: np.typing.DTypeLike = np.dtype(dtype)
+        self._dtype: np.dtype[Any] = np.dtype(dtype)
         self._offset = offset
         self._count = count
         self._close_on_next = False

README.md

@@ -132,6 +132,7 @@ class SubscriptionRequest(GatewayControl):
     symbols: str
     start: int | None = None
     snapshot: int = 0
+    id: int | None = None
 
 
 @dataclasses.dataclass

databento/common/dbnstore.py

@@ -284,6 +284,7 @@ def subscribe(
         stype_in: SType | str = SType.RAW_SYMBOL,
         start: str | int | None = None,
         snapshot: bool = False,
+        subscription_id: int | None = None,
     ) -> list[SubscriptionRequest]:
         """
         Send a SubscriptionRequest to the gateway. Returns a list of all
@@ -302,6 +303,8 @@ def subscribe(
             within 24 hours.
         snapshot: bool, default to 'False'
             Request subscription with snapshot. The `start` parameter must be `None`.
+        subscription_id : int, optional
+            A numerical identifier to associate with this subscription.
 
         Returns
         -------
@@ -329,6 +332,7 @@ def subscribe(
                 symbols=batch_str,
                 start=optional_datetime_to_unix_nanoseconds(start),
                 snapshot=int(snapshot),
+                id=subscription_id,
             )
             subscriptions.append(message)
 

databento/live/gateway.py

@@ -332,6 +332,7 @@ def __init__(
         self._transport: asyncio.Transport | None = None
         self._session_id: str | None = None
 
+        self._subscription_counter = 0
         self._subscriptions: list[SubscriptionRequest] = []
         self._reconnect_policy = ReconnectPolicy(reconnect_policy)
         self._reconnect_task: asyncio.Task[None] | None = None
@@ -499,13 +500,15 @@ def subscribe(
             if self._protocol is None:
                 self._connect(dataset=dataset)
 
+            self._subscription_counter += 1
             self._subscriptions.extend(
                 self._protocol.subscribe(
                     schema=schema,
                     symbols=symbols,
                     stype_in=stype_in,
                     start=start,
                     snapshot=snapshot,
+                    subscription_id=self._subscription_counter,
                 ),
             )
 
@@ -672,6 +675,7 @@ async def _reconnect(self) -> None:
                             stype_in=sub.stype_in,
                             snapshot=bool(sub.snapshot),
                             start=None,
+                            subscription_id=sub.id,
                         )
 
                     if should_restart:

databento/live/protocol.py

@@ -1 +1 @@
-__version__ = "0.51.0"
+__version__ = "0.52.0"

databento/live/session.py

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "databento"
-version = "0.51.0"
+version = "0.52.0"
 description = "Official Python client library for Databento"
 authors = [
     "Databento <support@databento.com>",
@@ -32,7 +32,7 @@ aiohttp = [
     {version = "^3.8.3", python = "<3.12"},
     {version = "^3.9.0", python = "^3.12"}
 ]
-databento-dbn = "0.31.0"
+databento-dbn = "0.32.0"
 numpy = [
     {version = ">=1.23.5", python = "<3.12"},
     {version = ">=1.26.0", python = "^3.12"}