dagster-snowflake 0.26.20__tar.gz → 0.27.0__tar.gz

{dagster_snowflake-0.26.20/dagster_snowflake.egg-info → dagster_snowflake-0.27.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dagster-snowflake
-Version: 0.26.20
+Version: 0.27.0
 Summary: Package for Snowflake Dagster framework components.
 Home-page: https://github.com/dagster-io/dagster/tree/master/python_modules/libraries/dagster-snowflake
 Author: Dagster Labs
@@ -14,7 +14,7 @@ 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.20
+Requires-Dist: dagster==1.11.0
 Requires-Dist: snowflake-connector-python>=3.4.0
 Requires-Dist: pyOpenSSL>=22.1.0
 Provides-Extra: snowflake-sqlalchemy

{dagster_snowflake-0.26.20 → dagster_snowflake-0.27.0}/dagster_snowflake/__init__.py

@@ -1,5 +1,6 @@
 from dagster_shared.libraries import DagsterLibraryRegistry
 
+from dagster_snowflake.components import SnowflakeSqlComponent as SnowflakeSqlComponent
 from dagster_snowflake.ops import snowflake_op_for_query as snowflake_op_for_query
 from dagster_snowflake.resources import (
     SnowflakeConnection as SnowflakeConnection,

dagster_snowflake-0.27.0/dagster_snowflake/components/__init__.py

@@ -0,0 +1,3 @@
+from dagster_snowflake.components.sql_component.component import SnowflakeSqlComponent
+
+__all__ = ["SnowflakeSqlComponent"]

dagster_snowflake-0.27.0/dagster_snowflake/components/sql_component/component.py

@@ -0,0 +1,24 @@
+from typing import Annotated
+
+from dagster._core.execution.context.asset_execution_context import AssetExecutionContext
+from dagster.components.base.sql_component import TemplatedSqlComponent
+from pydantic import Field
+
+from dagster_snowflake.resources import SnowflakeResource
+
+
+class SnowflakeSqlComponent(TemplatedSqlComponent[SnowflakeResource]):
+    """A component that executes SQL from a file in Snowflake."""
+
+    resource_key: Annotated[
+        str, Field(description="The resource key to use for the Snowflake resource.")
+    ] = "snowflake"
+
+    def execute(self, context: AssetExecutionContext, resource: SnowflakeResource) -> None:
+        """Execute the SQL content using the Snowflake resource."""
+        with resource.get_connection() as conn:
+            conn.cursor().execute(self.sql_content)
+
+    @property
+    def resource_keys(self) -> set[str]:
+        return {self.resource_key}

{dagster_snowflake-0.26.20 → dagster_snowflake-0.27.0}/dagster_snowflake/snowflake_io_manager.py

@@ -30,7 +30,7 @@ def build_snowflake_io_manager(
     Args:
         type_handlers (Sequence[DbTypeHandler]): Each handler defines how to translate between
             slices of Snowflake tables and an in-memory type - e.g. a Pandas DataFrame. If only
-            one DbTypeHandler is provided, it will be used as teh default_load_type.
+            one DbTypeHandler is provided, it will be used as the default_load_type.
         default_load_type (Type): When an input has no type annotation, load it as this type.
 
     Returns:
@@ -59,7 +59,7 @@ def build_snowflake_io_manager(
 
             snowflake_io_manager = build_snowflake_io_manager([SnowflakePandasTypeHandler(), SnowflakePySparkTypeHandler()])
 
-            defs = Definitions(
+            Definitions(
                 assets=[my_table, my_second_table],
                 resources={
                     "io_manager": snowflake_io_manager.configured({
@@ -75,7 +75,7 @@ def build_snowflake_io_manager(
 
         .. code-block:: python
 
-            defs = Definitions(
+            Definitions(
                 assets=[my_table]
                 resources={"io_manager" snowflake_io_manager.configured(
                     {"database": "my_database", "schema": "my_schema", ...} # will be used as the schema

dagster_snowflake-0.27.0/dagster_snowflake/version.py

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

{dagster_snowflake-0.26.20 → dagster_snowflake-0.27.0/dagster_snowflake.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dagster-snowflake
-Version: 0.26.20
+Version: 0.27.0
 Summary: Package for Snowflake Dagster framework components.
 Home-page: https://github.com/dagster-io/dagster/tree/master/python_modules/libraries/dagster-snowflake
 Author: Dagster Labs
@@ -14,7 +14,7 @@ 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.20
+Requires-Dist: dagster==1.11.0
 Requires-Dist: snowflake-connector-python>=3.4.0
 Requires-Dist: pyOpenSSL>=22.1.0
 Provides-Extra: snowflake-sqlalchemy

{dagster_snowflake-0.26.20 → dagster_snowflake-0.27.0}/dagster_snowflake.egg-info/SOURCES.txt

@@ -13,6 +13,10 @@ dagster_snowflake/version.py
 dagster_snowflake.egg-info/PKG-INFO
 dagster_snowflake.egg-info/SOURCES.txt
 dagster_snowflake.egg-info/dependency_links.txt
+dagster_snowflake.egg-info/entry_points.txt
 dagster_snowflake.egg-info/not-zip-safe
 dagster_snowflake.egg-info/requires.txt
-dagster_snowflake.egg-info/top_level.txt
+dagster_snowflake.egg-info/top_level.txt
+dagster_snowflake/components/__init__.py
+dagster_snowflake/components/sql_component/__init__.py
+dagster_snowflake/components/sql_component/component.py

dagster_snowflake-0.27.0/dagster_snowflake.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[dagster_dg_cli.registry_modules]
+dagster_snowflake = dagster_snowflake

{dagster_snowflake-0.26.20 → dagster_snowflake-0.27.0}/dagster_snowflake.egg-info/requires.txt

@@ -1,4 +1,4 @@
-dagster==1.10.20
+dagster==1.11.0
 snowflake-connector-python>=3.4.0
 pyOpenSSL>=22.1.0
 

{dagster_snowflake-0.26.20 → dagster_snowflake-0.27.0}/setup.py

@@ -34,7 +34,7 @@ setup(
     include_package_data=True,
     python_requires=">=3.9,<3.13",
     install_requires=[
-        "dagster==1.10.20",
+        "dagster==1.11.0",
         "snowflake-connector-python>=3.4.0",
         # Workaround for incorrect pin in the snowflake-connector-python package
         # See https://github.com/snowflakedb/snowflake-connector-python/issues/2109
@@ -51,4 +51,9 @@ setup(
         ],
     },
     zip_safe=False,
+    entry_points={
+        "dagster_dg_cli.registry_modules": [
+            "dagster_snowflake = dagster_snowflake",
+        ],
+    },
 )

dagster_snowflake-0.26.20/dagster_snowflake/version.py

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