contextbase-plugin-asana 0.2.0__py3-none-any.whl

contextbase_plugin_asana-0.2.0.dist-info/METADATA

@@ -0,0 +1,9 @@
+Metadata-Version: 2.3
+Name: contextbase-plugin-asana
+Version: 0.2.0
+Summary: Asana plugin for ContextBase (PyAirbyte-backed)
+Author: Alizain Feerasta
+Author-email: Alizain Feerasta <alizain.feerasta@gmail.com>
+Requires-Dist: contextbase-shared-plugins==0.2.0
+Requires-Dist: dagster==1.12.14
+Requires-Python: >=3.14, <3.15

contextbase_plugin_asana-0.2.0.dist-info/RECORD

@@ -0,0 +1,9 @@
+plugin_asana/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+plugin_asana/binding_config.py,sha256=dOQIGFcLDliHucOKh58LaXc_U0e8PrukpD4yjAmJmzU,303
+plugin_asana/component.py,sha256=0Apc22xPukpTtyMDWQOaokyeqa09QfkHVE_wref2OwQ,4199
+plugin_asana/defs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+plugin_asana/defs/defs.yaml,sha256=3ZBC-MkD3er3uhI-c-LIovSeUgjTG9pElxcXsrEzfA0,48
+plugin_asana/plugin.json,sha256=4xaJc0VRV9uAFnEK4JU73IJEN8u9-alJlFfovYgl4BY,81
+contextbase_plugin_asana-0.2.0.dist-info/WHEEL,sha256=fWriCkzqm-pffF5af4gJC9iI5FMFaJTuN9UxxxzOmdY,81
+contextbase_plugin_asana-0.2.0.dist-info/METADATA,sha256=MNGhp0ADcomriT94bDZoU_eyC9rxDoT1tC3Hlft1hO8,322
+contextbase_plugin_asana-0.2.0.dist-info/RECORD,,

contextbase_plugin_asana-0.2.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: uv 0.11.14
+Root-Is-Purelib: true
+Tag: py3-none-any

plugin_asana/binding_config.py

@@ -0,0 +1,12 @@
+from __future__ import annotations
+
+from pydantic import Field
+
+from shared_plugins.bindings import BaseBindingConfigModel, NonEmptyText
+
+
+class AsanaBindingConfig(BaseBindingConfigModel):
+    organization_export_ids: list[NonEmptyText] | None = Field(
+        default=None,
+        min_length=1,
+    )

plugin_asana/component.py

@@ -0,0 +1,124 @@
+import dagster as dg
+from dagster import AssetExecutionContext
+from shared_plugins.automation import non_overlapping_automation_condition
+from shared_plugins.bindings import (
+    parse_binding_config,
+    require_api_key,
+)
+from shared_plugins.control_plane import ControlPlaneClient
+from shared_plugins.dlt import resolve_partition_binding
+from shared_plugins.exceptions import PluginConfigurationError
+from shared_plugins.naming import (
+    dagster_airbyte_sync_asset_key,
+    dagster_asset_group_name,
+    dagster_asset_tags,
+    dagster_partition_def_name,
+    dagster_pool_name,
+    plugin_id_from_module,
+)
+from shared_plugins.pyairbyte import (
+    build_pyairbyte_source,
+    require_pyairbyte_selected_stream_names,
+    run_pyairbyte_sync,
+)
+from shared_types.api_key_auth import ApiKeyAuth
+from shared_types.dagster_binding_plan import DagsterAllPlanBinding
+
+from .binding_config import AsanaBindingConfig
+
+PLUGIN_ID = plugin_id_from_module(__file__)
+DOCKER_IMAGE = "airbyte/source-asana:1.5.2"
+SOURCE_UPDATED_AT_FIELDS = {
+    "tasks": "modified_at",
+    "projects": "modified_at",
+}
+
+
+def _build_connector_config(
+    cfg: AsanaBindingConfig,
+    auth: ApiKeyAuth,
+) -> dict[str, object]:
+    return {
+        **cfg.model_dump(mode="json", exclude_none=True),
+        "credentials": {
+            "option_title": "PAT Credentials",
+            "personal_access_token": auth.api_key,
+        },
+    }
+
+
+def _resolve_stream_selection(
+    binding: DagsterAllPlanBinding,
+    cfg: AsanaBindingConfig,
+) -> tuple[str, ...]:
+    selected_stream_names = require_pyairbyte_selected_stream_names(binding)
+
+    if (
+        "organization_exports" in selected_stream_names
+        and cfg.organization_export_ids is None
+    ):
+        raise PluginConfigurationError(
+            "asana binding.config.organization_export_ids is required when "
+            "models.active includes organization_exports."
+        )
+
+    return selected_stream_names
+
+
+class AsanaSyncComponent(dg.Component):
+    def build_defs(self, context: dg.ComponentLoadContext) -> dg.Definitions:
+        partitions_def = dg.DynamicPartitionsDefinition(
+            name=dagster_partition_def_name(PLUGIN_ID)
+        )
+
+        @dg.asset(
+            key=dagster_airbyte_sync_asset_key(PLUGIN_ID),
+            group_name=dagster_asset_group_name(PLUGIN_ID),
+            tags=dagster_asset_tags(PLUGIN_ID),
+            automation_condition=non_overlapping_automation_condition(
+                dg.AutomationCondition.on_missing()
+                | dg.AutomationCondition.on_cron("*/15 * * * *")
+            ),
+            partitions_def=partitions_def,
+            pool=dagster_pool_name(PLUGIN_ID),
+        )
+        def asana_sync_asset(
+            context: AssetExecutionContext,
+            control_plane: dg.ResourceParam[ControlPlaneClient],
+        ):
+            binding = resolve_partition_binding(
+                context=context,
+                control_plane=control_plane,
+                plugin_id=PLUGIN_ID,
+            )
+            binding_id = str(binding.binding_id)
+            cfg = parse_binding_config(binding, AsanaBindingConfig)
+            auth = require_api_key(binding)
+            selected_stream_names = _resolve_stream_selection(binding, cfg)
+
+            connector_config = _build_connector_config(cfg, auth)
+            source = build_pyairbyte_source(
+                docker_image=DOCKER_IMAGE,
+                connector_config=connector_config,
+            )
+
+            return run_pyairbyte_sync(
+                context=context,
+                plugin_id=PLUGIN_ID,
+                binding_id=binding_id,
+                selected_stream_names=selected_stream_names,
+                source_updated_at_fields=SOURCE_UPDATED_AT_FIELDS,
+                source=source,
+            )
+
+        automation_sensor = dg.AutomationConditionSensorDefinition(
+            name="asana_automation_sensor",
+            target=dg.AssetSelection.assets(asana_sync_asset),
+            default_status=dg.DefaultSensorStatus.RUNNING,
+            minimum_interval_seconds=30,
+        )
+
+        return dg.Definitions(
+            assets=[asana_sync_asset],
+            sensors=[automation_sensor],
+        )

plugin_asana/defs/defs.yaml

@@ -0,0 +1 @@
+type: plugin_asana.component.AsanaSyncComponent

plugin_asana/plugin.json

@@ -0,0 +1,7 @@
+{
+	"auth": {
+		"type": "api_key"
+	},
+	"mode": "dagster",
+	"plugin_id": "asana"
+}