dagster-snowflake 0.26.21__py3-none-any.whl → 0.27.0__py3-none-any.whl

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/components/__init__.py

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

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/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/version.py

@@ -1 +1 @@
-__version__ = "0.26.21"
+__version__ = "0.27.0"

{dagster_snowflake-0.26.21.dist-info → dagster_snowflake-0.27.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dagster-snowflake
-Version: 0.26.21
+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.21
+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.27.0.dist-info/RECORD

@@ -0,0 +1,16 @@
+dagster_snowflake/__init__.py,sha256=yJ5cO_RsirZdZa7KPLrKJYlGLsOlZHOHMTzSmkU6qvA,762
+dagster_snowflake/constants.py,sha256=26PGL1eFncm0WfbgZjA7jaznfaRx75vVJnqbieVYeco,481
+dagster_snowflake/ops.py,sha256=L_MP28fLm7_hrJmzMoDocLwyvVnkpy1LVwUSLIArKWc,2225
+dagster_snowflake/py.typed,sha256=la67KBlbjXN-_-DfGNcdOcjYumVpKG_Tkw-8n5dnGB4,8
+dagster_snowflake/resources.py,sha256=QlEBJvBOF35FhxBLZcgYCXm68DNthD8TRUpCcnipCg8,32101
+dagster_snowflake/snowflake_io_manager.py,sha256=emRPwQz2d3nffOVMVnoAx_rqndU7OAaROtkW0WzdoUc,17674
+dagster_snowflake/version.py,sha256=Hjsc9XQ3AkqXm-5cdlX-H8iKjURzi1FQeWNi2xx34Uo,23
+dagster_snowflake/components/__init__.py,sha256=q_0zAPClh010KAAiQqXIiowRBwVPZ3wsxow0d2H8aaA,124
+dagster_snowflake/components/sql_component/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+dagster_snowflake/components/sql_component/component.py,sha256=qRMHhx4cm8qJhCCQjSsl5jNrr08DoBcCNmcZKmZ2BTY,905
+dagster_snowflake-0.27.0.dist-info/licenses/LICENSE,sha256=4lsMW-RCvfVD4_F57wrmpe3vX1xwUk_OAKKmV_XT7Z0,11348
+dagster_snowflake-0.27.0.dist-info/METADATA,sha256=bGUmkKo_zI5J6lMpey0quO9lQurnc2IWfW8QXwxBr1w,1299
+dagster_snowflake-0.27.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dagster_snowflake-0.27.0.dist-info/entry_points.txt,sha256=Ddljs88OCl4BNyhSyE8lIAe54bIqEms0cOkR9vUAqsM,72
+dagster_snowflake-0.27.0.dist-info/top_level.txt,sha256=uECYCiluOxLQ996SCUPBBwdK0CTyz45FjWqf7WDqMMc,18
+dagster_snowflake-0.27.0.dist-info/RECORD,,

dagster_snowflake-0.27.0.dist-info/entry_points.txt

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

dagster_snowflake-0.26.21.dist-info/RECORD

@@ -1,12 +0,0 @@
-dagster_snowflake/__init__.py,sha256=ljbByIlHgAiQ4pHa0dTkACbvBguBtJsSK7mem3D9Kwo,674
-dagster_snowflake/constants.py,sha256=26PGL1eFncm0WfbgZjA7jaznfaRx75vVJnqbieVYeco,481
-dagster_snowflake/ops.py,sha256=L_MP28fLm7_hrJmzMoDocLwyvVnkpy1LVwUSLIArKWc,2225
-dagster_snowflake/py.typed,sha256=la67KBlbjXN-_-DfGNcdOcjYumVpKG_Tkw-8n5dnGB4,8
-dagster_snowflake/resources.py,sha256=QlEBJvBOF35FhxBLZcgYCXm68DNthD8TRUpCcnipCg8,32101
-dagster_snowflake/snowflake_io_manager.py,sha256=p4gI1G4qylNx-74-DzugKhi_7ac61AJAiqXPjNftNUk,17688
-dagster_snowflake/version.py,sha256=gtsPKN_satzDwTv3o1otQl1HwWzwBXeUyb8XS8-QBNs,24
-dagster_snowflake-0.26.21.dist-info/licenses/LICENSE,sha256=4lsMW-RCvfVD4_F57wrmpe3vX1xwUk_OAKKmV_XT7Z0,11348
-dagster_snowflake-0.26.21.dist-info/METADATA,sha256=4r7sYiX_jSH-CPJSSCbXLT5qGD4m_FyqCLuXBYneC6k,1301
-dagster_snowflake-0.26.21.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-dagster_snowflake-0.26.21.dist-info/top_level.txt,sha256=uECYCiluOxLQ996SCUPBBwdK0CTyz45FjWqf7WDqMMc,18
-dagster_snowflake-0.26.21.dist-info/RECORD,,