dagster-postgres 0.26.9__py3-none-any.whl → 0.26.11rc0__py3-none-any.whl

dagster_postgres/event_log/event_log.py

@@ -228,7 +228,7 @@ class PostgresEventLogStorage(SqlEventLogStorage, ConfigurableClass):
         insert_event_statement = self.prepare_insert_event_batch(events)
         with self._connect() as conn:
             result = conn.execute(insert_event_statement.returning(SqlEventLogStorageTable.c.id))
-            event_ids = [cast(int, row[0]) for row in result.fetchall()]
+            event_ids = [cast("int", row[0]) for row in result.fetchall()]
 
         # We only update the asset table with the last event
         self.store_asset_event(events[-1], event_ids[-1])

dagster_postgres/test_fixtures/__init__.py

@@ -0,0 +1,75 @@
+import tempfile
+from contextlib import contextmanager
+from pathlib import Path
+
+import pytest
+from dagster._core.test_utils import instance_for_test
+from dagster._utils.merger import merge_dicts
+from dagster_test.fixtures import docker_compose_cm, network_name_from_yml
+
+from dagster_postgres.utils import get_conn_string, wait_for_connection
+
+compose_file = Path(__file__).parent / "docker-compose.yml"
+
+
+@pytest.fixture(scope="session")
+def postgres_network():
+    yield network_name_from_yml(compose_file)
+
+
+@pytest.fixture(scope="session")
+def postgres_hostname():
+    with docker_compose_cm(docker_compose_yml=compose_file) as hostnames:
+        yield hostnames["postgres"]
+
+
+@pytest.fixture(scope="session")
+def postgres_conn_str(postgres_hostname):
+    conn_str = get_conn_string(
+        username="test",
+        password="test",
+        hostname=postgres_hostname,
+        db_name="test",
+        params=dict(connect_timeout=5),
+    )
+    wait_for_connection(
+        conn_str,
+        retry_limit=10,
+        retry_wait=3,
+    )
+
+    yield conn_str
+
+
+@pytest.fixture
+def postgres_instance(postgres_conn_str):
+    @contextmanager
+    def _instance(overrides=None):
+        with tempfile.TemporaryDirectory() as temp_dir:
+            with instance_for_test(
+                temp_dir=temp_dir,
+                overrides=merge_dicts(
+                    {
+                        "run_storage": {
+                            "module": "dagster_postgres.run_storage.run_storage",
+                            "class": "PostgresRunStorage",
+                            "config": {"postgres_url": postgres_conn_str},
+                        },
+                        "event_log_storage": {
+                            "module": "dagster_postgres.event_log.event_log",
+                            "class": "PostgresEventLogStorage",
+                            "config": {"postgres_url": postgres_conn_str},
+                        },
+                        "schedule_storage": {
+                            "module": "dagster_postgres.schedule_storage.schedule_storage",
+                            "class": "PostgresScheduleStorage",
+                            "config": {"postgres_url": postgres_conn_str},
+                        },
+                    },
+                    overrides if overrides else {},
+                ),
+            ) as instance:
+                instance.wipe()
+                yield instance
+
+    return _instance

dagster_postgres/test_fixtures/docker-compose.yml

@@ -0,0 +1,10 @@
+services:
+  postgres:
+    image: postgres:16
+    container_name: postgres
+    ports:
+      - "5432:5432"
+    environment:
+      POSTGRES_PASSWORD: "test"
+      POSTGRES_USER: "test"
+      POSTGRES_DB: "test"

dagster_postgres/version.py

@@ -1 +1 @@
-__version__ = "0.26.9"
+__version__ = "0.26.11rc0"

{dagster_postgres-0.26.9.dist-info → dagster_postgres-0.26.11rc0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dagster-postgres
-Version: 0.26.9
+Version: 0.26.11rc0
 Summary: A Dagster integration for postgres
 Home-page: https://github.com/dagster-io/dagster/tree/master/python_modules/libraries/dagster-postgres
 Author: Dagster Labs
@@ -14,6 +14,6 @@ Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.9,<3.13
 License-File: LICENSE
-Requires-Dist: dagster ==1.10.9
+Requires-Dist: dagster ==1.10.11rc0
 Requires-Dist: psycopg2-binary
 

{dagster_postgres-0.26.9.dist-info → dagster_postgres-0.26.11rc0.dist-info}/RECORD

@@ -2,16 +2,18 @@ dagster_postgres/__init__.py,sha256=rZfjhBDGC_SIGnKuiSCzi7D0xD36mAnchDLRL-n1Tmk,
 dagster_postgres/py.typed,sha256=la67KBlbjXN-_-DfGNcdOcjYumVpKG_Tkw-8n5dnGB4,8
 dagster_postgres/storage.py,sha256=CiPA773j0BQvPtCEz5vVqKjap9h0ODAxR_R_CvbI-bI,4310
 dagster_postgres/utils.py,sha256=AjuUYZ3y4Q3oaa-6I2zd6SNwl66FE8cJ_e2rIKrI9IQ,5930
-dagster_postgres/version.py,sha256=ZG17eZIeBB7Zg7MM33Pu3Eg0xdJHjWdE5Wl10Wuq_wo,23
+dagster_postgres/version.py,sha256=eS-xt-o3NPyxbPUwgn1WSzCSvbkXM3calY6j5hmh228,27
 dagster_postgres/alembic/alembic.ini,sha256=GovyDEhu_6HvkWV6txqjdDBOe4BseSM0YDWGxXM5_cA,986
 dagster_postgres/event_log/__init__.py,sha256=wRcUR-StRNrPCqpEzi0MRY8b-r_TEWV17OsEynFqlLs,100
-dagster_postgres/event_log/event_log.py,sha256=jCVQcaYKCIjaE5Vb0HcErS9MFLzIAm6CKnyiptRVyeQ,15707
+dagster_postgres/event_log/event_log.py,sha256=fPmO1w7ZweU9P_Kw366APG_aKSSRp74-L21sYspCAYc,15709
 dagster_postgres/run_storage/__init__.py,sha256=oW_546mJ5K-e-RF0Ou7r-4fHWxFthHgPPhWxklsVK1g,94
 dagster_postgres/run_storage/run_storage.py,sha256=VMXT5sb5n04ViWOYj2eOoZqOk78hqL38J6CuikM79No,9757
 dagster_postgres/schedule_storage/__init__.py,sha256=-jW-1S4Xf5Ew-cz-DjKjU5sVs9EEly_2ELMLOXTewv0,123
 dagster_postgres/schedule_storage/schedule_storage.py,sha256=iPKcvnj1eEeG5LH73GoFvoS3hmE8MFddiFyf7QoKZ3g,8826
-dagster_postgres-0.26.9.dist-info/LICENSE,sha256=TMatHW4_G9ldRdodEAp-l2Xa2WvsdeOh60E3v1R2jis,11349
-dagster_postgres-0.26.9.dist-info/METADATA,sha256=z3FjwbyyjTRv2QGpl6_Y71ewbVj6wzbDH2QWETgXUGY,713
-dagster_postgres-0.26.9.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
-dagster_postgres-0.26.9.dist-info/top_level.txt,sha256=lScMtAEKDX1yIv2tGa1nzntBa0HEStfWPfCwD8FWlHk,17
-dagster_postgres-0.26.9.dist-info/RECORD,,
+dagster_postgres/test_fixtures/__init__.py,sha256=ZIcbGYa_CMmVF0YkuiX4uI3wRjBke_2VE8QXKcji0Y4,2447
+dagster_postgres/test_fixtures/docker-compose.yml,sha256=hp2VTnENYK6CL2Yae3IsktssahPFQiHyhpNcG36CivM,208
+dagster_postgres-0.26.11rc0.dist-info/LICENSE,sha256=TMatHW4_G9ldRdodEAp-l2Xa2WvsdeOh60E3v1R2jis,11349
+dagster_postgres-0.26.11rc0.dist-info/METADATA,sha256=0xmjPHGH6dr9MKVjeomebugSfH4e1PMrUt4sFGrCQu8,721
+dagster_postgres-0.26.11rc0.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
+dagster_postgres-0.26.11rc0.dist-info/top_level.txt,sha256=lScMtAEKDX1yIv2tGa1nzntBa0HEStfWPfCwD8FWlHk,17
+dagster_postgres-0.26.11rc0.dist-info/RECORD,,