dagster-snowflake 0.27.3__tar.gz → 0.27.4__tar.gz

{dagster_snowflake-0.27.3/dagster_snowflake.egg-info → dagster_snowflake-0.27.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dagster-snowflake
-Version: 0.27.3
+Version: 0.27.4
 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
@@ -15,7 +15,7 @@ Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.9,<3.14
 License-File: LICENSE
-Requires-Dist: dagster==1.11.3
+Requires-Dist: dagster==1.11.4
 Requires-Dist: snowflake-connector-python>=3.4.0
 Requires-Dist: pyOpenSSL>=22.1.0
 Provides-Extra: snowflake-sqlalchemy

{dagster_snowflake-0.27.3 → dagster_snowflake-0.27.4}/dagster_snowflake/__init__.py

@@ -1,6 +1,8 @@
 from dagster_shared.libraries import DagsterLibraryRegistry
 
-from dagster_snowflake.components import SnowflakeSqlComponent as SnowflakeSqlComponent
+from dagster_snowflake.components import (
+    SnowflakeConnectionComponent as SnowflakeConnectionComponent,
+)
 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.4/dagster_snowflake/components/__init__.py

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

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

@@ -0,0 +1,59 @@
+from functools import cached_property
+from typing import Any, cast
+
+import dagster as dg
+from dagster._annotations import preview, public
+from dagster._core.definitions.definitions_class import Definitions
+from dagster.components.core.context import ComponentLoadContext
+from dagster.components.lib.sql_component.sql_client import SQLClient
+from pydantic import BaseModel, create_model
+
+from dagster_snowflake.resources import SnowflakeResource
+
+
+@public
+@preview
+class SnowflakeConnectionComponentBase(dg.Component, dg.Resolvable, dg.Model, SQLClient):
+    """A component that represents a Snowflake connection."""
+
+    @cached_property
+    def _snowflake_resource(self) -> SnowflakeResource:
+        return SnowflakeResource(
+            **{
+                (field.alias or field_name): getattr(self, field_name)
+                for field_name, field in self.__class__.model_fields.items()
+            }
+        )
+
+    def connect_and_execute(self, sql: str) -> None:
+        """Connect to the SQL database and execute the SQL query."""
+        return self._snowflake_resource.connect_and_execute(sql)
+
+    def build_defs(self, context: ComponentLoadContext) -> Definitions:
+        return Definitions()
+
+
+def _copy_fields_to_model(
+    copy_from: type[BaseModel], copy_to: type[BaseModel], new_model_cls_name: str
+) -> None:
+    """Given two models, creates a copy of the second model with the fields of the first model."""
+    field_definitions: dict[str, tuple[type, Any]] = {
+        field_name: (cast("type", field.annotation), field)
+        for field_name, field in copy_from.model_fields.items()
+    }
+
+    return create_model(
+        new_model_cls_name,
+        __base__=copy_to,
+        __doc__=copy_to.__doc__,
+        **field_definitions,  # type: ignore
+    )
+
+
+SnowflakeConnectionComponent = public(preview)(
+    _copy_fields_to_model(
+        copy_from=SnowflakeResource,
+        copy_to=SnowflakeConnectionComponentBase,
+        new_model_cls_name="SnowflakeConnectionComponent",
+    )
+)

{dagster_snowflake-0.27.3 → dagster_snowflake-0.27.4}/dagster_snowflake/resources.py

@@ -19,6 +19,7 @@ from dagster._annotations import public
 from dagster._core.definitions.resource_definition import dagster_maintained_resource
 from dagster._core.storage.event_log.sql_event_log import SqlDbConnection
 from dagster._utils.cached_method import cached_method
+from dagster.components.lib.sql_component.sql_client import SQLClient
 from pydantic import Field, model_validator, validator
 
 from dagster_snowflake.constants import (
@@ -40,7 +41,7 @@ except ImportError:
     raise
 
 
-class SnowflakeResource(ConfigurableResource, IAttachDifferentObjectToOpContext):
+class SnowflakeResource(ConfigurableResource, IAttachDifferentObjectToOpContext, SQLClient):
     """A resource for connecting to the Snowflake data warehouse.
 
     If connector configuration is not set, SnowflakeResource.get_connection() will return a
@@ -577,6 +578,10 @@ class SnowflakeResource(ConfigurableResource, IAttachDifferentObjectToOpContext)
             snowflake_connection_resource=self,
         )
 
+    def connect_and_execute(self, sql: str) -> None:
+        with self.get_connection() as conn:
+            conn.cursor().execute(sql)
+
 
 class SnowflakeConnection:
     """A connection to Snowflake that can execute queries. In general this class should not be

dagster_snowflake-0.27.4/dagster_snowflake/version.py

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

{dagster_snowflake-0.27.3 → dagster_snowflake-0.27.4/dagster_snowflake.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dagster-snowflake
-Version: 0.27.3
+Version: 0.27.4
 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
@@ -15,7 +15,7 @@ Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.9,<3.14
 License-File: LICENSE
-Requires-Dist: dagster==1.11.3
+Requires-Dist: dagster==1.11.4
 Requires-Dist: snowflake-connector-python>=3.4.0
 Requires-Dist: pyOpenSSL>=22.1.0
 Provides-Extra: snowflake-sqlalchemy

{dagster_snowflake-0.27.3 → dagster_snowflake-0.27.4}/dagster_snowflake.egg-info/requires.txt

@@ -1,4 +1,4 @@
-dagster==1.11.3
+dagster==1.11.4
 snowflake-connector-python>=3.4.0
 pyOpenSSL>=22.1.0
 

{dagster_snowflake-0.27.3 → dagster_snowflake-0.27.4}/setup.py

@@ -35,7 +35,7 @@ setup(
     include_package_data=True,
     python_requires=">=3.9,<3.14",
     install_requires=[
-        "dagster==1.11.3",
+        "dagster==1.11.4",
         "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

dagster_snowflake-0.27.3/dagster_snowflake/components/__init__.py

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

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

@@ -1,24 +0,0 @@
-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.27.3/dagster_snowflake/version.py

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