nginx-lens 0.3.4__py3-none-any.whl → 0.5.0__py3-none-any.whl

upstream_checker/dns_cache.py

@@ -0,0 +1,216 @@
+"""
+Модуль для кэширования результатов DNS резолвинга.
+"""
+import os
+import json
+import time
+import hashlib
+from typing import Optional, List, Dict
+from pathlib import Path
+
+
+class DNSCache:
+    """
+    Кэш для результатов DNS резолвинга.
+    """
+    
+    def __init__(self, cache_dir: Optional[str] = None, ttl: int = 300):
+        """
+        Инициализация кэша.
+        
+        Args:
+            cache_dir: Директория для хранения кэша (по умолчанию ~/.cache/nginx-lens/)
+            ttl: Время жизни кэша в секундах (по умолчанию 5 минут)
+        """
+        self.ttl = ttl
+        
+        if cache_dir:
+            self.cache_dir = Path(cache_dir)
+        else:
+            # Используем ~/.cache/nginx-lens/ или /tmp/nginx-lens-cache/
+            home_cache = Path.home() / ".cache" / "nginx-lens"
+            tmp_cache = Path("/tmp") / "nginx-lens-cache"
+            
+            # Пробуем использовать домашнюю директорию, если доступна
+            try:
+                home_cache.mkdir(parents=True, exist_ok=True)
+                self.cache_dir = home_cache
+            except (OSError, PermissionError):
+                # Fallback на /tmp
+                try:
+                    tmp_cache.mkdir(parents=True, exist_ok=True)
+                    self.cache_dir = tmp_cache
+                except (OSError, PermissionError):
+                    # Последний fallback - текущая директория
+                    self.cache_dir = Path.cwd() / ".nginx-lens-cache"
+                    self.cache_dir.mkdir(parents=True, exist_ok=True)
+        
+        self.cache_file = self.cache_dir / "dns_cache.json"
+        self._cache: Dict[str, Dict] = {}
+        self._load_cache()
+    
+    def _get_cache_key(self, host: str, port: str) -> str:
+        """
+        Генерирует ключ кэша для host:port.
+        
+        Args:
+            host: Имя хоста
+            port: Порт
+            
+        Returns:
+            Хеш ключ для кэша
+        """
+        key = f"{host}:{port}"
+        return hashlib.md5(key.encode()).hexdigest()
+    
+    def _load_cache(self):
+        """Загружает кэш из файла."""
+        if self.cache_file.exists():
+            try:
+                with open(self.cache_file, 'r') as f:
+                    self._cache = json.load(f)
+            except (json.JSONDecodeError, IOError):
+                self._cache = {}
+    
+    def _save_cache(self):
+        """Сохраняет кэш в файл."""
+        try:
+            with open(self.cache_file, 'w') as f:
+                json.dump(self._cache, f)
+        except IOError:
+            # Игнорируем ошибки записи
+            pass
+    
+    def get(self, host: str, port: str) -> Optional[List[str]]:
+        """
+        Получает результат из кэша.
+        
+        Args:
+            host: Имя хоста
+            port: Порт
+            
+        Returns:
+            Список резолвленных IP-адресов или None, если нет в кэше или истек TTL
+        """
+        key = self._get_cache_key(host, port)
+        
+        if key not in self._cache:
+            return None
+        
+        cached_data = self._cache[key]
+        cached_time = cached_data.get('timestamp', 0)
+        current_time = time.time()
+        
+        # Проверяем TTL
+        if current_time - cached_time > self.ttl:
+            # Удаляем устаревшую запись
+            del self._cache[key]
+            self._save_cache()
+            return None
+        
+        return cached_data.get('result')
+    
+    def set(self, host: str, port: str, result: List[str]):
+        """
+        Сохраняет результат в кэш.
+        
+        Args:
+            host: Имя хоста
+            port: Порт
+            result: Список резолвленных IP-адресов
+        """
+        key = self._get_cache_key(host, port)
+        
+        self._cache[key] = {
+            'timestamp': time.time(),
+            'result': result,
+            'host': host,
+            'port': port
+        }
+        
+        self._save_cache()
+    
+    def clear(self):
+        """Очищает весь кэш."""
+        self._cache = {}
+        if self.cache_file.exists():
+            try:
+                self.cache_file.unlink()
+            except IOError:
+                pass
+    
+    def get_cache_info(self) -> Dict[str, any]:
+        """
+        Возвращает информацию о кэше.
+        
+        Returns:
+            Словарь с информацией о кэше
+        """
+        current_time = time.time()
+        valid_entries = 0
+        expired_entries = 0
+        
+        for key, data in self._cache.items():
+            cached_time = data.get('timestamp', 0)
+            if current_time - cached_time <= self.ttl:
+                valid_entries += 1
+            else:
+                expired_entries += 1
+        
+        return {
+            'total_entries': len(self._cache),
+            'valid_entries': valid_entries,
+            'expired_entries': expired_entries,
+            'cache_dir': str(self.cache_dir),
+            'ttl': self.ttl
+        }
+
+
+# Глобальный экземпляр кэша (будет инициализирован при первом использовании)
+_cache_instance: Optional[DNSCache] = None
+_cache_enabled = True
+
+
+def get_cache(ttl: int = 300, cache_dir: Optional[str] = None) -> DNSCache:
+    """
+    Получает глобальный экземпляр кэша.
+    
+    Args:
+        ttl: Время жизни кэша в секундах
+        cache_dir: Директория для кэша
+        
+    Returns:
+        Экземпляр DNSCache
+    """
+    global _cache_instance
+    if _cache_instance is None:
+        _cache_instance = DNSCache(cache_dir=cache_dir, ttl=ttl)
+    elif ttl != _cache_instance.ttl:
+        # Обновляем TTL если изменился
+        _cache_instance.ttl = ttl
+    return _cache_instance
+
+
+def clear_cache():
+    """Очищает глобальный кэш."""
+    global _cache_instance
+    if _cache_instance:
+        _cache_instance.clear()
+
+
+def disable_cache():
+    """Отключает кэширование."""
+    global _cache_enabled
+    _cache_enabled = False
+
+
+def enable_cache():
+    """Включает кэширование."""
+    global _cache_enabled
+    _cache_enabled = True
+
+
+def is_cache_enabled() -> bool:
+    """Проверяет, включено ли кэширование."""
+    return _cache_enabled
+

utils/__init__.py

@@ -0,0 +1,4 @@
+"""
+Утилиты для nginx-lens.
+"""
+

utils/progress.py

@@ -0,0 +1,120 @@
+"""
+Утилиты для отображения прогресса операций.
+"""
+from typing import Optional, Callable, Any
+from rich.progress import Progress, SpinnerColumn, BarColumn, TextColumn, TimeElapsedColumn, TimeRemainingColumn, TaskID
+from rich.console import Console
+import signal
+import sys
+
+console = Console()
+
+
+class ProgressManager:
+    """
+    Менеджер для отображения прогресса операций с поддержкой отмены.
+    """
+    
+    def __init__(self, description: str = "Обработка", show_progress: bool = True):
+        """
+        Инициализация менеджера прогресса.
+        
+        Args:
+            description: Описание операции
+            show_progress: Показывать ли прогресс-бар
+        """
+        self.description = description
+        self.show_progress = show_progress
+        self.progress: Optional[Progress] = None
+        self.task_id: Optional[TaskID] = None
+        self.cancelled = False
+        
+        # Обработка сигнала прерывания
+        self._original_sigint = signal.signal(signal.SIGINT, self._handle_interrupt)
+    
+    def _handle_interrupt(self, signum, frame):
+        """Обработка прерывания (Ctrl+C)."""
+        self.cancelled = True
+        if self.progress:
+            self.progress.stop()
+        console.print("\n[yellow]Операция отменена пользователем[/yellow]")
+        sys.exit(130)  # Стандартный exit code для SIGINT
+    
+    def __enter__(self):
+        """Вход в контекстный менеджер."""
+        if self.show_progress:
+            self.progress = Progress(
+                SpinnerColumn(),
+                TextColumn("[progress.description]{task.description}"),
+                BarColumn(),
+                TextColumn("[progress.percentage]{task.percentage:>3.0f}%"),
+                TimeElapsedColumn(),
+                TimeRemainingColumn(),
+                console=console,
+                transient=False,
+            )
+            self.progress.start()
+            self.task_id = self.progress.add_task(self.description, total=None)
+        return self
+    
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        """Выход из контекстного менеджера."""
+        if self.progress:
+            self.progress.stop()
+        # Восстанавливаем оригинальный обработчик сигнала
+        signal.signal(signal.SIGINT, self._original_sigint)
+        return False
+    
+    def update(self, completed: int, total: Optional[int] = None, description: Optional[str] = None):
+        """
+        Обновляет прогресс.
+        
+        Args:
+            completed: Количество завершенных задач
+            total: Общее количество задач (если None, используется спиннер)
+            description: Описание текущей операции
+        """
+        if self.progress and self.task_id is not None:
+            if total is not None:
+                self.progress.update(self.task_id, total=total, completed=completed)
+            if description:
+                self.progress.update(self.task_id, description=description)
+    
+    def advance(self, advance: int = 1):
+        """
+        Увеличивает прогресс на указанное значение.
+        
+        Args:
+            advance: На сколько увеличить прогресс
+        """
+        if self.progress and self.task_id is not None:
+            self.progress.advance(self.task_id, advance)
+
+
+def with_progress(
+    description: str,
+    total: Optional[int] = None,
+    show_progress: bool = True
+) -> Callable:
+    """
+    Декоратор для добавления прогресс-бара к функции.
+    
+    Args:
+        description: Описание операции
+        total: Общее количество итераций (если известно)
+        show_progress: Показывать ли прогресс-бар
+        
+    Returns:
+        Декорированная функция
+    """
+    def decorator(func: Callable) -> Callable:
+        def wrapper(*args, **kwargs):
+            with ProgressManager(description=description, show_progress=show_progress) as pm:
+                # Передаем менеджер прогресса в функцию
+                kwargs['progress_manager'] = pm
+                if total is not None:
+                    pm.update(0, total=total)
+                return func(*args, **kwargs)
+        return wrapper
+    return decorator
+

nginx_lens-0.3.4.dist-info/RECORD

@@ -1,38 +0,0 @@
-analyzer/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-analyzer/base.py,sha256=oGKg78BfMVmuzYafc08oq9p31-jEgYolGjLkUcIdkN8,607
-analyzer/conflicts.py,sha256=NSNZc8e2x51K41dflSUvuwlDq-rzBXU5ITi6WfxFbfU,2796
-analyzer/dead_locations.py,sha256=uvMu5qBGTVi0Nn960x3WpRvTljGbQuVFivU4nfe36oY,1435
-analyzer/diff.py,sha256=idvXnoLzBVUYgKi_s3uDu0v2GNMV3B8aDqTROXcdQdo,1749
-analyzer/duplicates.py,sha256=jpy_6k-BzWxaXFt2Wb3rlulIXUEzbFe9xYRm7rWR50U,1215
-analyzer/empty_blocks.py,sha256=7Zu4-5I5PS3bjhH0Ppq1CvM7rMTeRIc4fHx5n5vkMIw,517
-analyzer/include.py,sha256=FhKR4VsogLknykjLD2N8jX9OtwxZcWik5oPpvp-_luE,2465
-analyzer/rewrite.py,sha256=-jSLLG1jqmGU-dXWvU6NHCW6muB8Lfro6fXX1tDCHCQ,1834
-analyzer/route.py,sha256=71dkmQaTrHqDTf4Up5gAsrgmgktNpXqWmxr7-0RAVtg,2370
-analyzer/unused.py,sha256=Ixzv0bPsw9IafblVwLiAOgugdg2dGu1MJDtuoqzPZiY,1066
-analyzer/warnings.py,sha256=zC36QMvegA2eQPvZ-P1eysrX_kXHx5A1MUKHKKNvG5c,5784
-commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-commands/analyze.py,sha256=W6begSgXNjgKJGoGeguR3WKgHPLkClWXxxpDcqvsJdc,8343
-commands/cli.py,sha256=brzp6xDDWIrm7ibaoT4x94hgAdBB2DVWniXoK8dRylE,782
-commands/diff.py,sha256=C7gRIWh6DNWHzjiQBPVTn-rZ40m2KCY75Zd6Q4URJIE,2076
-commands/graph.py,sha256=xB6KjXBkLmm5gII3e-5BMRGO7WeTwc7EFxRGzYgnme4,5947
-commands/health.py,sha256=lxrvuD-ClMLpTJ7gs7kn50fXwA4uBlHMPWA1ZxdEtAE,4167
-commands/include.py,sha256=5PTYG5C00-AlWfIgpQXLq9E7C9yTFSv7HrZkM5ogDps,2224
-commands/logs.py,sha256=RkPUdIpbO9dOVL56lemreYRuAjMjcqqMxRCKOFv2gC4,3691
-commands/resolve.py,sha256=MRruIH46tIelUyyrdrF70ai-tluuEJ13Jcj2nyRCSPA,2820
-commands/route.py,sha256=-x_71u6ENl3iO-oxK3bdE8v5eZKf4xRCydeUyXMFVrY,3163
-commands/syntax.py,sha256=ZWFdaL8LVv9S694wlk2aV3HJKb0OSKjw3wNgTlNvFR8,3418
-commands/tree.py,sha256=mDfx0Aeg1EDQSYQoJ2nJIkSd_uP7ZR7pEqy7Cw3clQ0,2161
-exporter/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-exporter/graph.py,sha256=WYUrqUgCaK6KihgxAcRHaQn4oMo6b7ybC8yb_36ZIsA,3995
-exporter/html.py,sha256=uquEM-WvBt2aV9GshgaI3UVhYd8sD0QQ-OmuNtvYUdU,798
-exporter/markdown.py,sha256=_0mXQIhurGEZ0dO-eq9DbsuKNrgEDIblgtL3DAgYNo8,724
-nginx_lens-0.3.4.dist-info/licenses/LICENSE,sha256=g8QXKdvZZC56rU8E12vIeYF6R4jeTWOsblOnYAda3K4,1073
-parser/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-parser/nginx_parser.py,sha256=Sa9FtGAkvTqNzoehBvgLUWPJHLLIZYWH9ugSHW50X8s,3699
-upstream_checker/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-upstream_checker/checker.py,sha256=i3L6XqUHUH5hcyLq5PXx6wOyjzEL_Z7xYCA3FffFOrU,11257
-nginx_lens-0.3.4.dist-info/METADATA,sha256=E7ce-3zNXvNVp7-y_TvlnmtCTIjMhid2sdqzbrpgy8U,552
-nginx_lens-0.3.4.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-nginx_lens-0.3.4.dist-info/entry_points.txt,sha256=qEcecjSyLqcJjbIVlNlTpqAhPqDyaujUV5ZcBTAr3po,48
-nginx_lens-0.3.4.dist-info/top_level.txt,sha256=mxLJO4rZg0rbixVGhplF3fUNFs8vxDIL25ronZNvRy4,51
-nginx_lens-0.3.4.dist-info/RECORD,,