contextbase-plugin-microsoft-calendar 0.5.0__py3-none-any.whl

contextbase_plugin_microsoft_calendar-0.5.0.dist-info/METADATA

@@ -0,0 +1,14 @@
+Metadata-Version: 2.3
+Name: contextbase-plugin-microsoft-calendar
+Version: 0.5.0
+Summary: Microsoft Calendar plugin for ContextBase
+Author: Alizain Feerasta
+Author-email: Alizain Feerasta <alizain.feerasta@gmail.com>
+Requires-Dist: azure-identity>=1.25.1
+Requires-Dist: dagster==1.12.14
+Requires-Dist: dagster-dlt==0.28.14
+Requires-Dist: dlt[duckdb]>=1.26.0
+Requires-Dist: msgraph-beta-sdk>=1.59.0
+Requires-Dist: pydantic>=2.12.0
+Requires-Dist: contextbase-shared-plugins==0.5.0
+Requires-Python: >=3.12, <3.13

contextbase_plugin_microsoft_calendar-0.5.0.dist-info/RECORD

@@ -0,0 +1,16 @@
+plugin_microsoft_calendar/__init__.py,sha256=lHKUHWlw_OzYGpygFYyeENy5Z8Ex2wKXcHDokONpTzI,41
+plugin_microsoft_calendar/binding_config.py,sha256=GHhAXcmuSkSbKCSKKAFPeA8YIZCbDAP86qdz_gITM2g,235
+plugin_microsoft_calendar/component.py,sha256=9F3xuTPYccdUoADQ-ilBJ2dfwDDMFe8NiyjbWTH_JKw,4084
+plugin_microsoft_calendar/models/__init__.py,sha256=lnbXYLwz1CncE7srQad9rxuV8upNrA9yzfWqgTG6sXg,53
+plugin_microsoft_calendar/models/ctx.py,sha256=CyChavg1FXWbWyxINkpgPnw8AYH3g49f2KsUpnZnLKg,21522
+plugin_microsoft_calendar/models/ingress.py,sha256=R_9mOMaV69pIBouxaMstL3KRf_daRO1HZrpQ_m0ftVM,11437
+plugin_microsoft_calendar/models/translators.py,sha256=POAwJYz0Ha17lgyZhxStMcH-bj7aAfJlz95dlJvqDQc,8639
+plugin_microsoft_calendar/plugin.json,sha256=SmGTdp6eSzKt8uk-BrWNT2K0DHYazmeYXaGVQMFI9O0,105
+plugin_microsoft_calendar/sources/__init__.py,sha256=a6Vtc3DWBf1OglF6UdCTwNtrUJ-eiybC_6qAKZnWv18,53
+plugin_microsoft_calendar/sources/sync.py,sha256=pfF5wEeMweF8HdU1abM_Cs4qb7Nv-PI5x6Rype6xKqs,12721
+plugin_microsoft_calendar/utils/__init__.py,sha256=CjzjKMbRq4S7OUZuQ_M9U_b8W0uGtW3tkgVewb8tGWE,57
+plugin_microsoft_calendar/utils/client.py,sha256=Uz7h5AYyCmyBopjRliW23vriYqGL2aWhv4NM5_DUqxM,4612
+contextbase_plugin_microsoft_calendar-0.5.0.dist-info/WHEEL,sha256=i9aSRDivn5iP9LaR1BLQX2GNAuriQWPsFwbbWygTX2k,81
+contextbase_plugin_microsoft_calendar-0.5.0.dist-info/entry_points.txt,sha256=FVqaBtkD9T61fUfiLj9m7Y2SsZY47FKQ7hDp7Lzfsmk,108
+contextbase_plugin_microsoft_calendar-0.5.0.dist-info/METADATA,sha256=G_qfOQEniqCd3Z_V2JQnk0G64JBI-QmmA4AcqqQ34sU,510
+contextbase_plugin_microsoft_calendar-0.5.0.dist-info/RECORD,,

contextbase_plugin_microsoft_calendar-0.5.0.dist-info/WHEEL

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

contextbase_plugin_microsoft_calendar-0.5.0.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[contextlayer.plugins]
+microsoft_calendar = plugin_microsoft_calendar.component:MicrosoftCalendarComponent
+

plugin_microsoft_calendar/__init__.py

@@ -0,0 +1 @@
+"""Microsoft Calendar plugin package."""

plugin_microsoft_calendar/binding_config.py

@@ -0,0 +1,8 @@
+from __future__ import annotations
+
+from shared_plugins.bindings import BaseBindingConfigModel, NonEmptyText
+
+
+class MicrosoftCalendarBindingConfig(BaseBindingConfigModel):
+    tenant_id: NonEmptyText
+    mailbox_user_id: NonEmptyText

plugin_microsoft_calendar/component.py

@@ -0,0 +1,113 @@
+import dagster as dg
+from dagster import AssetExecutionContext
+from dagster_dlt import DagsterDltResource
+from shared_plugins.automation import non_overlapping_automation_condition
+from shared_plugins.base_platform import BasePlatformClient
+from shared_plugins.bindings import parse_binding_config, require_client_credentials
+from shared_plugins.dlt import resolve_partition_binding, run_dlt_pipeline
+from shared_plugins.naming import (
+    dagster_asset_group_name,
+    dagster_asset_tags,
+    dagster_dlt_asset_key,
+    dagster_partition_def_name,
+    dagster_pool_name,
+    dlt_source_name,
+    plugin_id_from_module,
+)
+from shared_plugins.resources import DLT_RESOURCE
+
+from .binding_config import MicrosoftCalendarBindingConfig
+from .sources.sync import microsoft_calendar_source
+from .utils.client import SyncGraphCalendarClient
+
+PLUGIN_ID = plugin_id_from_module(__file__)
+SYNC_JOB = "sync"
+
+SYNC_SOURCE_NAME = dlt_source_name(PLUGIN_ID, SYNC_JOB)
+
+CALENDARS_ASSET_KEY = dagster_dlt_asset_key(SYNC_SOURCE_NAME, "calendars")
+EVENTS_ASSET_KEY = dagster_dlt_asset_key(SYNC_SOURCE_NAME, "events")
+
+
+def _build_sync_specs(
+    partitions_def: dg.PartitionsDefinition,
+    automation_condition: dg.AutomationCondition,
+) -> list[dg.AssetSpec]:
+    shared = dict(
+        group_name=dagster_asset_group_name(PLUGIN_ID),
+        tags=dagster_asset_tags(PLUGIN_ID, extra_tags={"ctx_primary": "true"}),
+        automation_condition=automation_condition,
+        partitions_def=partitions_def,
+    )
+    return [
+        dg.AssetSpec(key=CALENDARS_ASSET_KEY, **shared),
+        dg.AssetSpec(key=EVENTS_ASSET_KEY, deps=[CALENDARS_ASSET_KEY], **shared),
+    ]
+
+
+class MicrosoftCalendarComponent(dg.Component):
+    def build_defs(self, context: dg.ComponentLoadContext) -> dg.Definitions:
+        partitions_def = dg.DynamicPartitionsDefinition(
+            name=dagster_partition_def_name(PLUGIN_ID)
+        )
+
+        sync_specs = _build_sync_specs(
+            partitions_def,
+            non_overlapping_automation_condition(
+                dg.AutomationCondition.on_missing()
+                | dg.AutomationCondition.on_cron("*/15 * * * *")
+            ),
+        )
+
+        @dg.multi_asset(
+            specs=sync_specs,
+            can_subset=False,
+            name="microsoft_calendar_sync",
+            pool=dagster_pool_name(PLUGIN_ID),
+        )
+        def microsoft_calendar_sync_assets(
+            context: AssetExecutionContext,
+            dlt_resource: DagsterDltResource,
+            base_platform: dg.ResourceParam[BasePlatformClient],
+        ):
+            binding = resolve_partition_binding(
+                context=context,
+                base_platform=base_platform,
+                plugin_id=PLUGIN_ID,
+            )
+            binding_id = binding.binding_id
+            cfg = parse_binding_config(binding, MicrosoftCalendarBindingConfig)
+            auth = require_client_credentials(binding)
+
+            with SyncGraphCalendarClient(
+                auth=auth,
+                tenant_id=cfg.tenant_id,
+                mailbox_user_id=cfg.mailbox_user_id,
+            ) as client:
+                source = microsoft_calendar_source(
+                    binding_id,
+                    client=client,
+                )
+                yield from run_dlt_pipeline(
+                    context=context,
+                    dlt_resource=dlt_resource,
+                    source=source,
+                    plugin_id=PLUGIN_ID,
+                    binding_id=binding_id,
+                    job_name=SYNC_JOB,
+                )
+
+        automation_sensor = dg.AutomationConditionSensorDefinition(
+            name="microsoft_calendar_automation_sensor",
+            target=dg.AssetSelection.assets(microsoft_calendar_sync_assets),
+            default_status=dg.DefaultSensorStatus.RUNNING,
+            minimum_interval_seconds=30,
+        )
+
+        return dg.Definitions(
+            assets=[microsoft_calendar_sync_assets],
+            sensors=[automation_sensor],
+            resources={
+                "dlt_resource": DLT_RESOURCE,
+            },
+        )

plugin_microsoft_calendar/models/__init__.py

@@ -0,0 +1 @@
+"""Data models for the Microsoft Calendar plugin."""