first commit

app/services/replication_state.py

@@ -0,0 +1,89 @@
+import json
+from datetime import datetime
+from pathlib import Path
+from typing import Dict, Any, Optional
+from app.repository.replication_metadata_repo import replication_metadata_repo
+from app.core.logging import migration_logger
+from app.core.config import settings
+
+
+class ReplicationState:
+    """Хранение состояния репликации по ID"""
+    
+    def __init__(self):
+        self.replication_repo = replication_metadata_repo
+        self.replication_repo.init_metadata_table()
+    
+    def get_last_replication_time(self) -> Optional[datetime]:
+        """Получает время последней репликации (максимальное по всем таблицам)"""
+        stats = self.replication_repo.get_all_stats()
+        if stats['tables']:
+            times = [t['last_sync'] for t in stats['tables'] if t['last_sync']]
+            return max(times) if times else None
+        return None
+    
+    def get_last_replication_info(self) -> dict:
+        """Получает информацию о последней репликации"""
+        stats = self.replication_repo.get_all_stats()
+        return {
+            'last_replication': self.get_last_replication_time(),
+            'total_rows': stats['total_rows'],
+            'tables_count': stats['tables_count']
+        }
+    
+    def get_table_last_sync(self, table_name: str) -> Optional[datetime]:
+        """Получает время последней синхронизации таблицы"""
+        return self.replication_repo.get_last_sync_time(table_name)
+    
+    def get_last_id(self, table_name: str) -> Optional[int]:
+        """Получает последний обработанный ID для таблицы"""
+        return self.replication_repo.get_last_id(table_name)
+    
+    def update_last_id(self, table_name: str, last_id: int):
+        """Обновляет последний обработанный ID"""
+        self.replication_repo.update_last_id(table_name, last_id)
+    
+    def update_table_sync_time(self, table_name: str):
+        """Обновляет время синхронизации таблицы"""
+        self.replication_repo.update_sync_time(table_name)
+    
+    def update_table_stats(self, table_name: str, added_rows: int):
+        """Обновляет статистику таблицы"""
+        self.replication_repo.update_table_stats(table_name, added_rows)
+    
+    def get_all_stats(self) -> dict:
+        """Получает статистику по всем таблицам"""
+        return self.replication_repo.get_all_stats()
+    
+    def log_operation(self, table_name: str, operation: str, records_count: int, 
+                     status: str = "SUCCESS", error_message: Optional[str] = None):
+        """Логирует операцию"""
+        self.replication_repo.log_operation(table_name, operation, records_count, status, error_message)
+
+    def _set_table_total_rows(self, table_name: str, total_rows: int):
+        """
+        Устанавливает точное количество строк для таблицы
+        (в отличие от update_table_stats, который добавляет)
+        """
+        from app.models.replication import ReplicationMetadata
+        
+        session = self.replication_repo.get_session()
+        try:
+            metadata = session.query(ReplicationMetadata).filter_by(
+                table_name=table_name
+            ).first()
+            
+            if metadata:
+                metadata.total_rows = total_rows
+                metadata.updated_at = datetime.now()
+                session.commit()
+                migration_logger.debug(f"  Установлено total_rows={total_rows} для {table_name}")
+            else:
+                migration_logger.warning(f"Метаданные для {table_name} не найдены")
+        except Exception as e:
+            session.rollback()
+            migration_logger.error(f"Ошибка установки total_rows для {table_name}: {e}")
+        finally:
+            session.close()
+
+replication_state = ReplicationState()