unify anayasa tools

2025-07-17 20:09:50 +03:00
parent 856ecdf13d
commit f96c1a2e44
6 changed files with 271 additions and 152 deletions
@@ -0,0 +1,122 @@
+# anayasa_mcp_module/unified_client.py
+# Unified client for both Norm Denetimi and Bireysel Başvuru
+
+import logging
+from typing import Optional
+from urllib.parse import urlparse
+
+from .models import (
+    AnayasaUnifiedSearchRequest,
+    AnayasaUnifiedSearchResult, 
+    AnayasaUnifiedDocumentMarkdown,
+    AnayasaDecisionTypeEnum,
+    AnayasaNormDenetimiSearchRequest,
+    AnayasaBireyselReportSearchRequest
+)
+from .client import AnayasaMahkemesiApiClient
+from .bireysel_client import AnayasaBireyselBasvuruApiClient
+
+logger = logging.getLogger(__name__)
+
+class AnayasaUnifiedClient:
+    """Unified client that handles both Norm Denetimi and Bireysel Başvuru searches."""
+    
+    def __init__(self, request_timeout: float = 60.0):
+        self.norm_client = AnayasaMahkemesiApiClient(request_timeout)
+        self.bireysel_client = AnayasaBireyselBasvuruApiClient(request_timeout)
+    
+    async def search_unified(self, params: AnayasaUnifiedSearchRequest) -> AnayasaUnifiedSearchResult:
+        """Unified search that routes to appropriate client based on decision_type."""
+        
+        if params.decision_type == AnayasaDecisionTypeEnum.NORM_DENETIMI:
+            # Convert to norm denetimi request
+            norm_params = AnayasaNormDenetimiSearchRequest(
+                keywords_all=params.keywords_all or params.keywords,
+                keywords_any=params.keywords_any,
+                application_type=params.decision_type_norm,
+                page_to_fetch=params.page_to_fetch,
+                results_per_page=params.results_per_page
+            )
+            
+            result = await self.norm_client.search_norm_denetimi_decisions(norm_params)
+            
+            # Convert to unified format
+            decisions_list = [decision.model_dump() for decision in result.decisions]
+            
+            return AnayasaUnifiedSearchResult(
+                decision_type=AnayasaDecisionTypeEnum.NORM_DENETIMI,
+                decisions=decisions_list,
+                total_records_found=result.total_records_found,
+                retrieved_page_number=result.retrieved_page_number
+            )
+            
+        elif params.decision_type == AnayasaDecisionTypeEnum.BIREYSEL_BASVURU:
+            # Convert to bireysel başvuru request
+            bireysel_params = AnayasaBireyselReportSearchRequest(
+                keywords=params.keywords,
+                decision_start_date=params.decision_start_date,
+                decision_end_date=params.decision_end_date,
+                norm_type=params.norm_type,
+                subject_category=params.subject_category,
+                page_to_fetch=params.page_to_fetch,
+                results_per_page=params.results_per_page
+            )
+            
+            result = await self.bireysel_client.search_bireysel_basvuru_report(bireysel_params)
+            
+            # Convert to unified format
+            decisions_list = [decision.model_dump() for decision in result.decisions]
+            
+            return AnayasaUnifiedSearchResult(
+                decision_type=AnayasaDecisionTypeEnum.BIREYSEL_BASVURU,
+                decisions=decisions_list,
+                total_records_found=result.total_records_found,
+                retrieved_page_number=result.retrieved_page_number
+            )
+        
+        else:
+            raise ValueError(f"Unsupported decision type: {params.decision_type}")
+    
+    async def get_document_unified(self, document_url: str, page_number: int = 1) -> AnayasaUnifiedDocumentMarkdown:
+        """Unified document retrieval that auto-detects the appropriate client."""
+        
+        # Auto-detect decision type based on URL
+        parsed_url = urlparse(document_url)
+        
+        if "normkararlarbilgibankasi" in parsed_url.netloc or "/ND/" in document_url:
+            # Norm Denetimi document
+            result = await self.norm_client.get_decision_document_as_markdown(document_url, page_number)
+            
+            return AnayasaUnifiedDocumentMarkdown(
+                decision_type=AnayasaDecisionTypeEnum.NORM_DENETIMI,
+                source_url=result.source_url,
+                document_data=result.model_dump(),
+                markdown_chunk=result.markdown_chunk,
+                current_page=result.current_page,
+                total_pages=result.total_pages,
+                is_paginated=result.is_paginated
+            )
+            
+        elif "kararlarbilgibankasi" in parsed_url.netloc or "/BB/" in document_url:
+            # Bireysel Başvuru document
+            result = await self.bireysel_client.get_decision_document_as_markdown(document_url, page_number)
+            
+            return AnayasaUnifiedDocumentMarkdown(
+                decision_type=AnayasaDecisionTypeEnum.BIREYSEL_BASVURU,
+                source_url=result.source_url,
+                document_data=result.model_dump(),
+                markdown_chunk=result.markdown_chunk,
+                current_page=result.current_page,
+                total_pages=result.total_pages,
+                is_paginated=result.is_paginated
+            )
+        
+        else:
+            raise ValueError(f"Cannot determine document type from URL: {document_url}")
+    
+    async def close_client_session(self):
+        """Close both client sessions."""
+        if hasattr(self.norm_client, 'close_client_session'):
+            await self.norm_client.close_client_session()
+        if hasattr(self.bireysel_client, 'close_client_session'):
+            await self.bireysel_client.close_client_session()