Implement workflow & activity registry (cadence-workflow#14)

Author: timl3136

cadence/__init__.py

@@ -0,0 +1,14 @@
+"""
+Cadence Python Client
+
+A Python framework for authoring workflows and activities for Cadence.
+"""
+
+# Import main client functionality
+from .client import Client
+
+__version__ = "0.1.0"
+
+__all__ = [
+    "Client",
+]

cadence/worker/__init__.py

@@ -5,7 +5,16 @@
     WorkerOptions
 )
 
+from ._registry import (
+    Registry,
+    RegisterWorkflowOptions,
+    RegisterActivityOptions,
+)
+
 __all__ = [
     "Worker",
-    "WorkerOptions"
-]
+    "WorkerOptions",
+    'Registry',
+    'RegisterWorkflowOptions',
+    'RegisterActivityOptions',
+]

cadence/worker/_registry.py

@@ -0,0 +1,174 @@
+#!/usr/bin/env python3
+"""
+Workflow and Activity Registry for Cadence Python Client.
+
+This module provides a registry system for managing workflows and activities,
+similar to the Go client's registry.go implementation.
+"""
+
+import logging
+from typing import Callable, Dict, Optional, Unpack, TypedDict
+
+
+logger = logging.getLogger(__name__)
+
+
+class RegisterWorkflowOptions(TypedDict, total=False):
+    """Options for registering a workflow."""
+    name: Optional[str]
+    alias: Optional[str]
+
+
+class RegisterActivityOptions(TypedDict, total=False):
+    """Options for registering an activity."""
+    name: Optional[str]
+    alias: Optional[str]
+
+
+class Registry:
+    """
+    Registry for managing workflows and activities.
+    
+    This class provides functionality to register, retrieve, and manage
+    workflows and activities in a Cadence application.
+    """
+    
+    def __init__(self):
+        """Initialize the registry."""
+        self._workflows: Dict[str, Callable] = {}
+        self._activities: Dict[str, Callable] = {}
+        self._workflow_aliases: Dict[str, str] = {}  # alias -> name mapping
+        self._activity_aliases: Dict[str, str] = {}  # alias -> name mapping
+        
+    def workflow(
+        self,
+        func: Optional[Callable] = None,
+        **kwargs: Unpack[RegisterWorkflowOptions]
+    ) -> Callable:
+        """
+        Register a workflow function.
+        
+        This method can be used as a decorator or called directly.
+        
+        Args:
+            func: The workflow function to register
+            **kwargs: Options for registration (name, alias)
+            
+        Returns:
+            The decorated function or the function itself
+            
+        Raises:
+            KeyError: If workflow name already exists
+        """
+        options = RegisterWorkflowOptions(**kwargs)
+        
+        def decorator(f: Callable) -> Callable:
+            workflow_name = options.get('name') or f.__name__
+            
+            if workflow_name in self._workflows:
+                raise KeyError(f"Workflow '{workflow_name}' is already registered")
+            
+            self._workflows[workflow_name] = f
+            
+            # Register alias if provided
+            alias = options.get('alias')
+            if alias:
+                if alias in self._workflow_aliases:
+                    raise KeyError(f"Workflow alias '{alias}' is already registered")
+                self._workflow_aliases[alias] = workflow_name
+            
+            logger.info(f"Registered workflow '{workflow_name}'")
+            return f
+        
+        if func is None:
+            return decorator
+        return decorator(func)
+    
+    def activity(
+        self,
+        func: Optional[Callable] = None,
+        **kwargs: Unpack[RegisterActivityOptions]
+    ) -> Callable:
+        """
+        Register an activity function.
+        
+        This method can be used as a decorator or called directly.
+        
+        Args:
+            func: The activity function to register
+            **kwargs: Options for registration (name, alias)
+            
+        Returns:
+            The decorated function or the function itself
+            
+        Raises:
+            KeyError: If activity name already exists
+        """
+        options = RegisterActivityOptions(**kwargs)
+        
+        def decorator(f: Callable) -> Callable:
+            activity_name = options.get('name') or f.__name__
+            
+            if activity_name in self._activities:
+                raise KeyError(f"Activity '{activity_name}' is already registered")
+            
+            self._activities[activity_name] = f
+            
+            # Register alias if provided
+            alias = options.get('alias')
+            if alias:
+                if alias in self._activity_aliases:
+                    raise KeyError(f"Activity alias '{alias}' is already registered")
+                self._activity_aliases[alias] = activity_name
+            
+            logger.info(f"Registered activity '{activity_name}'")
+            return f
+        
+        if func is None:
+            return decorator
+        return decorator(func)
+    
+    def get_workflow(self, name: str) -> Callable:
+        """
+        Get a registered workflow by name.
+        
+        Args:
+            name: Name or alias of the workflow
+            
+        Returns:
+            The workflow function
+            
+        Raises:
+            KeyError: If workflow is not found
+        """
+        # Check if it's an alias
+        actual_name = self._workflow_aliases.get(name, name)
+        
+        if actual_name not in self._workflows:
+            raise KeyError(f"Workflow '{name}' not found in registry")
+        
+        return self._workflows[actual_name]
+    
+    def get_activity(self, name: str) -> Callable:
+        """
+        Get a registered activity by name.
+        
+        Args:
+            name: Name or alias of the activity
+            
+        Returns:
+            The activity function
+            
+        Raises:
+            KeyError: If activity is not found
+        """
+        # Check if it's an alias
+        actual_name = self._activity_aliases.get(name, name)
+        
+        if actual_name not in self._activities:
+            raise KeyError(f"Activity '{name}' not found in registry")
+        
+        return self._activities[actual_name]
+    
+
+    

pyproject.toml

@@ -35,7 +35,7 @@ dependencies = [
 
 [project.optional-dependencies]
 dev = [
-    "pytest>=7.0.0",
+    "pytest>=8.4.1",
     "pytest-cov>=4.0.0",
     "pytest-asyncio>=0.21.0",
     "black>=23.0.0",