weave-python 0.33.0__tar.gz → 0.34.0__tar.gz

{weave_python-0.33.0 → weave_python-0.34.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: weave-python
-Version: 0.33.0
+Version: 0.34.0
 Author: Weave Labs
 License: FCL-1.0-ALv2
 License-File: LICENSE

{weave_python-0.33.0 → weave_python-0.34.0}/weave/weavesql/atcdb/event.py

@@ -117,6 +117,20 @@ class ListEventsByTypeRow:
     occurred_at: datetime.datetime
 
 
+LIST_RUN_I_DS = """-- name: list_run_i_ds \\:many
+SELECT DISTINCT run_id
+FROM atc.events
+WHERE tenant_id = :p1
+  AND transaction_id < pg_snapshot_xmin(pg_current_snapshot())
+ORDER BY run_id
+"""
+
+
+@dataclasses.dataclass()
+class ListRunIDsParams:
+    tenant_id: str
+
+
 LOAD_ALL_EVENTS = """-- name: load_all_events \\:many
 SELECT
     id,
@@ -410,6 +424,13 @@ class Querier:
                 occurred_at=row[8],
             )
 
+    def list_run_i_ds(self, arg: ListRunIDsParams) -> Iterator[str]:
+        result = self._conn.execute(
+            sqlalchemy.text(LIST_RUN_I_DS), {"p1": arg.tenant_id}
+        )
+        for row in result:
+            yield row[0]
+
     def load_all_events(self, arg: LoadAllEventsParams) -> Iterator[LoadAllEventsRow]:
         result = self._conn.execute(
             sqlalchemy.text(LOAD_ALL_EVENTS), {"p1": arg.tenant_id, "p2": arg.run_id}
@@ -606,6 +627,13 @@ class AsyncQuerier:
                 occurred_at=row[8],
             )
 
+    async def list_run_i_ds(self, arg: ListRunIDsParams) -> AsyncIterator[str]:
+        result = await self._conn.stream(
+            sqlalchemy.text(LIST_RUN_I_DS), {"p1": arg.tenant_id}
+        )
+        async for row in result:
+            yield row[0]
+
     async def load_all_events(
         self, arg: LoadAllEventsParams
     ) -> AsyncIterator[LoadAllEventsRow]: