dagster-postgres 0.21.13__tar.gz → 0.21.14__tar.gz

{dagster-postgres-0.21.13/dagster_postgres.egg-info → dagster-postgres-0.21.14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dagster-postgres
-Version: 0.21.13
+Version: 0.21.14
 Summary: A Dagster integration for postgres
 Home-page: https://github.com/dagster-io/dagster/tree/master/python_modules/libraries/dagster-postgres
 Author: Dagster Labs

{dagster-postgres-0.21.13 → dagster-postgres-0.21.14}/dagster_postgres/event_log/event_log.py

@@ -1,4 +1,5 @@
-from typing import Any, ContextManager, Mapping, Optional, Sequence
+from contextlib import contextmanager
+from typing import Any, ContextManager, Iterator, Mapping, Optional, Sequence
 
 import dagster._check as check
 import sqlalchemy as db
@@ -287,6 +288,17 @@ class PostgresEventLogStorage(SqlEventLogStorage, ConfigurableClass):
     def index_connection(self) -> ContextManager[Connection]:
         return self._connect()
 
+    @contextmanager
+    def index_transaction(self) -> Iterator[Connection]:
+        """Context manager yielding a connection to the index shard that has begun a transaction."""
+        with self.index_connection() as conn:
+            if conn.in_transaction():
+                yield conn
+            else:
+                conn = conn.execution_options(isolation_level="READ COMMITTED")  # noqa: PLW2901
+                with conn.begin():
+                    yield conn
+
     def has_table(self, table_name: str) -> bool:
         return bool(self._engine.dialect.has_table(self._engine.connect(), table_name))
 

dagster-postgres-0.21.14/dagster_postgres/version.py

@@ -0,0 +1 @@
+__version__ = "0.21.14"

{dagster-postgres-0.21.13 → dagster-postgres-0.21.14/dagster_postgres.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dagster-postgres
-Version: 0.21.13
+Version: 0.21.14
 Summary: A Dagster integration for postgres
 Home-page: https://github.com/dagster-io/dagster/tree/master/python_modules/libraries/dagster-postgres
 Author: Dagster Labs

{dagster-postgres-0.21.13 → dagster-postgres-0.21.14}/dagster_postgres.egg-info/requires.txt

@@ -1,2 +1,2 @@
-dagster==1.5.13
+dagster==1.5.14
 psycopg2-binary

{dagster-postgres-0.21.13 → dagster-postgres-0.21.14}/setup.py

@@ -38,6 +38,6 @@ setup(
         ]
     },
     include_package_data=True,
-    install_requires=["dagster==1.5.13", "psycopg2-binary"],
+    install_requires=["dagster==1.5.14", "psycopg2-binary"],
     zip_safe=False,
 )

dagster-postgres-0.21.13/dagster_postgres/version.py

@@ -1 +0,0 @@
-__version__ = "0.21.13"