""" Document Sync Utilities Hilfsfunktionen für Document-Synchronisation mit xAI: - Distributed locking via Redis + syncStatus - Entscheidungslogik: Wann muss ein Document zu xAI? - Related Entities ermitteln (Many-to-Many Attachments) - xAI Collection Management """ from typing import Dict, Any, Optional, List, Tuple from datetime import datetime, timedelta import logging import redis import os logger = logging.getLogger(__name__) # Lock TTL in seconds (prevents deadlocks) LOCK_TTL_SECONDS = 900 # 15 minutes # Max retry before permanent failure MAX_SYNC_RETRIES = 5 # Retry backoff: Wartezeit zwischen Retries (in Minuten) RETRY_BACKOFF_MINUTES = [1, 5, 15, 60, 240] # 1min, 5min, 15min, 1h, 4h class DocumentSync: """Utility-Klasse für Document-Synchronisation mit xAI""" def __init__(self, espocrm_api, redis_client: redis.Redis = None, context=None): self.espocrm = espocrm_api self.context = context self.logger = context.logger if context else logger self.redis = redis_client or self._init_redis() def _init_redis(self) -> redis.Redis: """Initialize Redis client for distributed locking""" try: redis_host = os.getenv('REDIS_HOST', 'localhost') redis_port = int(os.getenv('REDIS_PORT', '6379')) redis_db = int(os.getenv('REDIS_DB_ADVOWARE_CACHE', '1')) client = redis.Redis( host=redis_host, port=redis_port, db=redis_db, decode_responses=True ) client.ping() return client except Exception as e: self._log(f"Redis connection failed: {e}", level='error') return None def _log(self, message: str, level: str = 'info'): """Logging mit Context-Support""" if self.context and hasattr(self.context, 'logger'): getattr(self.context.logger, level)(message) else: getattr(logger, level)(message) async def acquire_sync_lock(self, entity_id: str) -> bool: """ Atomic distributed lock via Redis + syncStatus update Args: entity_id: EspoCRM Document ID Returns: True wenn Lock erfolgreich, False wenn bereits im Sync """ try: # STEP 1: Atomic Redis lock (prevents race conditions) if self.redis: lock_key = f"sync_lock:document:{entity_id}" acquired = self.redis.set(lock_key, "locked", nx=True, ex=LOCK_TTL_SECONDS) if not acquired: self._log(f"Redis lock bereits aktiv für Document {entity_id}", level='warn') return False # STEP 2: Update syncStatus (für UI visibility) - falls Feld existiert # NOTE: Ggf. muss syncStatus bei Document Entity erst angelegt werden try: await self.espocrm.update_entity('Document', entity_id, { 'xaiSyncStatus': 'syncing' }) except Exception as e: self._log(f"Konnte xaiSyncStatus nicht setzen (Feld existiert evtl. nicht): {e}", level='debug') self._log(f"Sync-Lock für Document {entity_id} erworben") return True except Exception as e: self._log(f"Fehler beim Acquire Lock: {e}", level='error') # Clean up Redis lock on error if self.redis: try: lock_key = f"sync_lock:document:{entity_id}" self.redis.delete(lock_key) except: pass return False async def release_sync_lock( self, entity_id: str, success: bool = True, error_message: Optional[str] = None, extra_fields: Optional[Dict[str, Any]] = None ) -> None: """ Gibt Sync-Lock frei und setzt finalen Status Args: entity_id: EspoCRM Document ID success: Ob Sync erfolgreich war error_message: Optional: Fehlermeldung extra_fields: Optional: Zusätzliche Felder (z.B. xaiFileId, xaiCollections) """ try: update_data = {} # Status-Feld (falls vorhanden) try: update_data['xaiSyncStatus'] = 'synced' if success else 'failed' if error_message: update_data['xaiSyncError'] = error_message[:2000] else: update_data['xaiSyncError'] = None except: pass # Felder existieren evtl. nicht # Merge extra fields (z.B. xaiFileId, xaiCollections) if extra_fields: update_data.update(extra_fields) if update_data: await self.espocrm.update_entity('Document', entity_id, update_data) self._log(f"Sync-Lock released: Document {entity_id} → {'success' if success else 'failed'}") # Release Redis lock if self.redis: lock_key = f"sync_lock:document:{entity_id}" self.redis.delete(lock_key) except Exception as e: self._log(f"Fehler beim Release Lock: {e}", level='error') # Ensure Redis lock is released even on error if self.redis: try: lock_key = f"sync_lock:document:{entity_id}" self.redis.delete(lock_key) except: pass async def should_sync_to_xai(self, document: Dict[str, Any]) -> Tuple[bool, List[str], str]: """ Entscheidet ob ein Document zu xAI synchronisiert werden muss Args: document: Vollständiges Document Entity von EspoCRM Returns: Tuple[bool, List[str], str]: - bool: Ob Sync nötig ist - List[str]: Liste der Collection-IDs in die das Document soll - str: Grund/Beschreibung der Entscheidung """ doc_id = document.get('id') doc_name = document.get('name', 'Unbenannt') # xAI-relevante Felder xai_file_id = document.get('xaiFileId') xai_collections = document.get('xaiCollections') or [] self._log(f"📋 Document Analysis: {doc_name} (ID: {doc_id})") self._log(f" xaiFileId: {xai_file_id or 'N/A'}") self._log(f" xaiCollections: {xai_collections}") # ═══════════════════════════════════════════════════════════════ # FALL 1: Document ist bereits in xAI UND Collections sind gesetzt # ═══════════════════════════════════════════════════════════════ if xai_file_id and xai_collections: self._log(f"✅ Document bereits in xAI gesynct mit {len(xai_collections)} Collection(s)") # Prüfe ob Update nötig (z.B. wenn File selbst geändert wurde) # TODO: Implementiere File-Hash-Vergleich für Update-Erkennung return (False, xai_collections, "Bereits gesynct, keine Änderung erkannt") # ═══════════════════════════════════════════════════════════════ # FALL 2: Document hat xaiFileId aber Collections ist leer/None # ═══════════════════════════════════════════════════════════════ if xai_file_id and not xai_collections: self._log(f"⚠️ Document hat xaiFileId aber keine Collections - prüfe Related Entities") # Fallthrough zu FALL 3 - prüfe Related Entities # ═══════════════════════════════════════════════════════════════ # FALL 3: Prüfe Related Entities (Attachments) # ═══════════════════════════════════════════════════════════════ required_collections = await self._get_required_collections_from_relations(doc_id) if required_collections: self._log(f"✅ Document ist mit {len(required_collections)} Entity/ies verknüpft die Collections haben") return (True, required_collections, f"Verknüpft mit Entities die Collections benötigen") # ═══════════════════════════════════════════════════════════════ # FALL 4: Keine Collections gefunden → kein Sync nötig # ═══════════════════════════════════════════════════════════════ self._log(f"⏭️ Kein xAI-Sync nötig: Keine Related Entities mit Collections") return (False, [], "Keine verknüpften Entities mit xAI Collections") async def _get_required_collections_from_relations(self, document_id: str) -> List[str]: """ Ermittelt alle xAI Collection-IDs von Entities die mit diesem Document verknüpft sind EspoCRM Many-to-Many: Document kann mit beliebigen Entities verknüpft sein (CBeteiligte, Account, CVmhErstgespraech, etc.) Args: document_id: Document ID Returns: Liste von xAI Collection-IDs (dedupliziert) """ collections = set() # Liste von Entity-Types die xAI Collections haben können # NOTE: Erweiterbar für andere Entities entity_types_with_collections = [ 'CBeteiligte', 'Account', 'CVmhErstgespraech', # Weitere Entity-Types hier hinzufügen ] self._log(f"🔍 Prüfe Attachments von Document {document_id}...") for entity_type in entity_types_with_collections: try: # Finde alle Entities dieses Typs die dieses Document attached haben # EspoCRM API: Suche wo documentsIds das Document enthält result = await self.espocrm.list_entities( entity_type, where=[{ 'type': 'arrayAnyOf', 'attribute': 'documentsIds', 'value': [document_id] }], select='id,name,xaiCollectionId', max_size=100 ) entities = result.get('list', []) if entities: self._log(f" ✅ {len(entities)} {entity_type}(s) gefunden") for entity in entities: collection_id = entity.get('xaiCollectionId') if collection_id: collections.add(collection_id) self._log(f" → {entity.get('name')}: Collection {collection_id[:16]}...") except Exception as e: self._log(f" ⚠️ Fehler beim Prüfen von {entity_type}: {e}", level='warn') continue result = list(collections) self._log(f"📊 Gesamt: {len(result)} eindeutige Collection(s) gefunden") return result async def get_document_download_info(self, document_id: str) -> Optional[Dict[str, Any]]: """ Holt Download-Informationen für ein Document Returns: Dict mit: - download_url: URL zum Download - filename: Dateiname - mime_type: MIME-Type - size: Dateigröße in Bytes """ try: # Hole vollständiges Document doc = await self.espocrm.get_entity('Document', document_id) # EspoCRM Document hat Attachments (Attachment ID in attachmentsIds) attachment_ids = doc.get('attachmentsIds') or [] if not attachment_ids: self._log(f"⚠️ Document {document_id} hat keine Attachments", level='warn') return None # Nehme erstes Attachment (Documents haben normalerweise nur 1 File) attachment_id = attachment_ids[0] # Hole Attachment-Details attachment = await self.espocrm.get_entity('Attachment', attachment_id) return { 'attachment_id': attachment_id, 'download_url': f"/api/v1/Attachment/file/{attachment_id}", 'filename': attachment.get('name', 'unknown'), 'mime_type': attachment.get('type', 'application/octet-stream'), 'size': attachment.get('size', 0) } except Exception as e: self._log(f"❌ Fehler beim Laden von Download-Info: {e}", level='error') return None