gitflow-analytics 1.0.1__py3-none-any.whl → 1.0.3__py3-none-any.whl

gitflow_analytics/integrations/orchestrator.py

@@ -1,7 +1,8 @@
 """Integration orchestrator for multiple platforms."""
+
 import json
 from datetime import datetime
-from typing import Any, Dict, List
+from typing import Any, Union
 
 from ..core.cache import GitAnalysisCache
 from .github_integration import GitHubIntegration
@@ -10,110 +11,110 @@ from .jira_integration import JIRAIntegration
 
 class IntegrationOrchestrator:
     """Orchestrate integrations with multiple platforms."""
-    
+
     def __init__(self, config: Any, cache: GitAnalysisCache):
         """Initialize integration orchestrator."""
         self.config = config
         self.cache = cache
-        self.integrations = {}
-        
+        self.integrations: dict[str, Union[GitHubIntegration, JIRAIntegration]] = {}
+
         # Initialize available integrations
         if config.github and config.github.token:
-            self.integrations['github'] = GitHubIntegration(
+            self.integrations["github"] = GitHubIntegration(
                 config.github.token,
                 cache,
                 config.github.max_retries,
                 config.github.backoff_factor,
-                allowed_ticket_platforms=getattr(config.analysis, 'ticket_platforms', None)
+                allowed_ticket_platforms=getattr(config.analysis, "ticket_platforms", None),
             )
-        
+
         # Initialize JIRA integration if configured
         if config.jira and config.jira.access_user and config.jira.access_token:
             # Get JIRA specific settings if available
-            jira_settings = getattr(config, 'jira_integration', {})
-            if hasattr(jira_settings, 'enabled') and jira_settings.enabled:
-                base_url = getattr(config.jira, 'base_url', None)
+            jira_settings = getattr(config, "jira_integration", {})
+            if hasattr(jira_settings, "enabled") and jira_settings.enabled:
+                base_url = getattr(config.jira, "base_url", None)
                 if base_url:
-                    self.integrations['jira'] = JIRAIntegration(
+                    self.integrations["jira"] = JIRAIntegration(
                         base_url,
                         config.jira.access_user,
                         config.jira.access_token,
                         cache,
-                        story_point_fields=getattr(jira_settings, 'story_point_fields', None)
+                        story_point_fields=getattr(jira_settings, "story_point_fields", None),
                     )
-    
-    def enrich_repository_data(self, repo_config: Any, commits: List[Dict[str, Any]], 
-                             since: datetime) -> Dict[str, Any]:
+
+    def enrich_repository_data(
+        self, repo_config: Any, commits: list[dict[str, Any]], since: datetime
+    ) -> dict[str, Any]:
         """Enrich repository data from all available integrations."""
-        enrichment = {
-            'prs': [],
-            'issues': [],
-            'pr_metrics': {}
-        }
-        
+        enrichment: dict[str, Any] = {"prs": [], "issues": [], "pr_metrics": {}}
+
         # GitHub enrichment
-        if 'github' in self.integrations and repo_config.github_repo:
-            github = self.integrations['github']
-            
-            try:
-                # Get PR data
-                prs = github.enrich_repository_with_prs(
-                    repo_config.github_repo, commits, since
-                )
-                enrichment['prs'] = prs
-                
-                # Calculate PR metrics
-                if prs:
-                    enrichment['pr_metrics'] = github.calculate_pr_metrics(prs)
-                    
-            except Exception as e:
-                print(f"   ⚠️  GitHub enrichment failed: {e}")
-        
+        if "github" in self.integrations and repo_config.github_repo:
+            github_integration = self.integrations["github"]
+            if isinstance(github_integration, GitHubIntegration):
+                try:
+                    # Get PR data
+                    prs = github_integration.enrich_repository_with_prs(
+                        repo_config.github_repo, commits, since
+                    )
+                    enrichment["prs"] = prs
+
+                    # Calculate PR metrics
+                    if prs:
+                        enrichment["pr_metrics"] = github_integration.calculate_pr_metrics(prs)
+
+                except Exception as e:
+                    print(f"   ⚠️  GitHub enrichment failed: {e}")
+
         # JIRA enrichment for story points
-        if 'jira' in self.integrations:
-            jira = self.integrations['jira']
-            try:
-                # Enrich commits with JIRA story points
-                jira.enrich_commits_with_jira_data(commits)
-                
-                # Enrich PRs with JIRA story points
-                if enrichment['prs']:
-                    jira.enrich_prs_with_jira_data(enrichment['prs'])
-                    
-            except Exception as e:
-                print(f"   ⚠️  JIRA enrichment failed: {e}")
-        
+        if "jira" in self.integrations:
+            jira_integration = self.integrations["jira"]
+            if isinstance(jira_integration, JIRAIntegration):
+                try:
+                    # Enrich commits with JIRA story points
+                    jira_integration.enrich_commits_with_jira_data(commits)
+
+                    # Enrich PRs with JIRA story points
+                    if enrichment["prs"]:
+                        jira_integration.enrich_prs_with_jira_data(enrichment["prs"])
+
+                except Exception as e:
+                    print(f"   ⚠️  JIRA enrichment failed: {e}")
+
         return enrichment
-    
-    def get_platform_issues(self, project_key: str, since: datetime) -> List[Dict[str, Any]]:
+
+    def get_platform_issues(self, project_key: str, since: datetime) -> list[dict[str, Any]]:
         """Get issues from all configured platforms."""
-        all_issues = []
-        
+        all_issues: list[dict[str, Any]] = []
+
         # Check cache first
         cached_issues = []
-        for platform in ['github', 'jira', 'clickup', 'linear']:
+        for platform in ["github", "jira", "clickup", "linear"]:
             cached = self.cache.get_cached_issues(platform, project_key)
             cached_issues.extend(cached)
-        
+
         if cached_issues:
             return cached_issues
-        
+
         # Future: Fetch from APIs if not cached
         # This is where we'd add actual API calls to each platform
-        
+
         return all_issues
-    
-    def export_to_json(self, 
-                      commits: List[Dict[str, Any]],
-                      prs: List[Dict[str, Any]],
-                      developer_stats: List[Dict[str, Any]],
-                      project_metrics: Dict[str, Any],
-                      dora_metrics: Dict[str, Any],
-                      output_path: str) -> str:
+
+    def export_to_json(
+        self,
+        commits: list[dict[str, Any]],
+        prs: list[dict[str, Any]],
+        developer_stats: list[dict[str, Any]],
+        project_metrics: dict[str, Any],
+        dora_metrics: dict[str, Any],
+        output_path: str,
+    ) -> str:
         """Export all data to JSON format for API consumption."""
-        
+
         # Prepare data for JSON serialization
-        def serialize_dates(obj):
+        def serialize_dates(obj: Any) -> Any:
             """Convert datetime objects to ISO format strings."""
             if isinstance(obj, datetime):
                 return obj.isoformat()
@@ -122,24 +123,24 @@ class IntegrationOrchestrator:
             elif isinstance(obj, list):
                 return [serialize_dates(item) for item in obj]
             return obj
-        
+
         export_data = {
-            'metadata': {
-                'generated_at': datetime.now().isoformat(),
-                'version': '1.0',
-                'total_commits': len(commits),
-                'total_prs': len(prs),
-                'total_developers': len(developer_stats)
+            "metadata": {
+                "generated_at": datetime.now().isoformat(),
+                "version": "1.0",
+                "total_commits": len(commits),
+                "total_prs": len(prs),
+                "total_developers": len(developer_stats),
             },
-            'commits': serialize_dates(commits),
-            'pull_requests': serialize_dates(prs),
-            'developers': serialize_dates(developer_stats),
-            'project_metrics': serialize_dates(project_metrics),
-            'dora_metrics': serialize_dates(dora_metrics)
+            "commits": serialize_dates(commits),
+            "pull_requests": serialize_dates(prs),
+            "developers": serialize_dates(developer_stats),
+            "project_metrics": serialize_dates(project_metrics),
+            "dora_metrics": serialize_dates(dora_metrics),
         }
-        
+
         # Write JSON file
-        with open(output_path, 'w') as f:
+        with open(output_path, "w") as f:
             json.dump(export_data, f, indent=2)
-        
-        return output_path
+
+        return output_path