aws-inventory-manager 0.2.0__py3-none-any.whl

src/utils/hash.py

@@ -0,0 +1,60 @@
+"""Configuration hashing utility for change detection."""
+
+import hashlib
+import json
+from typing import Any, Dict, Set
+
+# Attributes to exclude from hashing (volatile data)
+EXCLUDE_ATTRIBUTES: Set[str] = {
+    "ResponseMetadata",
+    "LastModifiedDate",
+    "CreatedDate",
+    "CreateDate",
+    "State",
+    "Status",
+    "RequestId",
+    "VersionId",
+    "LastUpdateTime",
+    "LastUpdatedTime",
+    "ModifiedTime",
+}
+
+
+def compute_config_hash(resource_data: Dict[str, Any]) -> str:
+    """Compute stable SHA256 hash of resource configuration.
+
+    This hash is used for change detection. Volatile attributes
+    (timestamps, states, etc.) are excluded to prevent false positives.
+
+    Args:
+        resource_data: Resource configuration dictionary
+
+    Returns:
+        64-character SHA256 hex string
+    """
+    # Deep copy and remove excluded attributes
+    clean_data = _remove_volatile_attributes(resource_data, EXCLUDE_ATTRIBUTES)
+
+    # Normalize: sort keys for deterministic JSON
+    normalized = json.dumps(clean_data, sort_keys=True, default=str)
+
+    # Hash
+    return hashlib.sha256(normalized.encode()).hexdigest()
+
+
+def _remove_volatile_attributes(data: Any, exclude_set: Set[str]) -> Any:
+    """Recursively remove excluded attributes from nested dict/list.
+
+    Args:
+        data: Data structure to clean (dict, list, or primitive)
+        exclude_set: Set of attribute names to exclude
+
+    Returns:
+        Cleaned data structure
+    """
+    if isinstance(data, dict):
+        return {k: _remove_volatile_attributes(v, exclude_set) for k, v in data.items() if k not in exclude_set}
+    elif isinstance(data, list):
+        return [_remove_volatile_attributes(item, exclude_set) for item in data]
+    else:
+        return data

src/utils/logging.py

@@ -0,0 +1,63 @@
+"""Logging configuration for AWS Baseline Snapshot tool."""
+
+import logging
+import sys
+from typing import Optional
+
+
+def setup_logging(level: str = "INFO", log_file: Optional[str] = None, verbose: bool = False) -> None:
+    """Configure logging for the application.
+
+    Args:
+        level: Log level (DEBUG, INFO, WARN, ERROR)
+        log_file: Optional log file path
+        verbose: If True, show detailed logs; if False, suppress all but critical
+    """
+    # Convert string level to logging constant
+    numeric_level = getattr(logging, level.upper(), logging.INFO)
+
+    # In non-verbose mode, suppress all logs except CRITICAL
+    # User will only see styled Rich console output
+    if not verbose:
+        numeric_level = logging.CRITICAL
+
+    # Create formatter
+    formatter = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s", datefmt="%Y-%m-%d %H:%M:%S")
+
+    # Configure root logger
+    root_logger = logging.getLogger()
+    root_logger.setLevel(numeric_level)
+
+    # Remove existing handlers
+    for handler in root_logger.handlers[:]:
+        root_logger.removeHandler(handler)
+
+    # Console handler (only add if verbose or log_file specified)
+    if verbose or log_file:
+        console_handler = logging.StreamHandler(sys.stderr)
+        console_handler.setLevel(numeric_level)
+        console_handler.setFormatter(formatter)
+        root_logger.addHandler(console_handler)
+
+    # File handler (if specified)
+    if log_file:
+        file_handler = logging.FileHandler(log_file)
+        file_handler.setLevel(logging.DEBUG)  # Always log everything to file
+        file_handler.setFormatter(formatter)
+        root_logger.addHandler(file_handler)
+
+    # Suppress noisy third-party loggers
+    logging.getLogger("boto3").setLevel(logging.CRITICAL)
+    logging.getLogger("botocore").setLevel(logging.CRITICAL)
+    logging.getLogger("urllib3").setLevel(logging.CRITICAL)
+    logging.getLogger("s3transfer").setLevel(logging.CRITICAL)
+
+    # Suppress internal module logs unless verbose
+    if not verbose:
+        logging.getLogger("src").setLevel(logging.CRITICAL)
+        logging.getLogger("src.snapshot").setLevel(logging.CRITICAL)
+        logging.getLogger("src.snapshot.resource_collectors").setLevel(logging.CRITICAL)
+        logging.getLogger("src.snapshot.capturer").setLevel(logging.CRITICAL)
+        logging.getLogger("src.snapshot.storage").setLevel(logging.CRITICAL)
+        logging.getLogger("src.aws").setLevel(logging.CRITICAL)
+        logging.getLogger("src.aws.credentials").setLevel(logging.CRITICAL)

src/utils/paths.py

@@ -0,0 +1,51 @@
+"""Path resolution utilities for snapshot storage."""
+
+import os
+from pathlib import Path
+from typing import Optional, Union
+
+
+def get_snapshot_storage_path(custom_path: Optional[Union[str, Path]] = None) -> Path:
+    """Resolve snapshot storage path with precedence: parameter > env var > default.
+
+    Precedence order:
+    1. custom_path parameter (if provided)
+    2. AWS_INVENTORY_STORAGE_PATH environment variable (if set)
+    3. ~/.snapshots (default)
+
+    Args:
+        custom_path: Optional custom path override
+
+    Returns:
+        Resolved Path object for snapshot storage
+
+    Examples:
+        # Use default
+        >>> get_snapshot_storage_path()
+        Path.home() / '.snapshots'
+
+        # Use environment variable
+        >>> os.environ['AWS_INVENTORY_STORAGE_PATH'] = '/data/snapshots'
+        >>> get_snapshot_storage_path()
+        Path('/data/snapshots')
+
+        # Use parameter (highest priority)
+        >>> get_snapshot_storage_path('/custom/path')
+        Path('/custom/path')
+    """
+    # Priority 1: Custom path parameter (but not empty string)
+    if custom_path:
+        # Handle both str and Path types
+        if isinstance(custom_path, str):
+            if custom_path.strip():
+                return Path(custom_path).expanduser().resolve()
+        else:  # Path object
+            return custom_path.expanduser().resolve()
+
+    # Priority 2: Environment variable
+    env_path = os.getenv("AWS_INVENTORY_STORAGE_PATH")
+    if env_path:
+        return Path(env_path).expanduser().resolve()
+
+    # Priority 3: Default to ~/.snapshots
+    return Path.home() / ".snapshots"

src/utils/progress.py

@@ -0,0 +1,41 @@
+"""Progress indicator utilities using Rich library."""
+
+from contextlib import contextmanager
+
+from rich.progress import (
+    BarColumn,
+    Progress,
+    SpinnerColumn,
+    TaskProgressColumn,
+    TextColumn,
+    TimeRemainingColumn,
+)
+
+
+@contextmanager
+def create_progress():
+    """Create a Rich progress context for tracking operations.
+
+    Yields:
+        Progress instance configured for multi-task tracking
+    """
+    with Progress(
+        SpinnerColumn(),
+        TextColumn("[progress.description]{task.description}"),
+        BarColumn(),
+        TaskProgressColumn(),
+        TimeRemainingColumn(),
+    ) as progress:
+        yield progress
+
+
+def create_spinner_progress():
+    """Create a simple spinner progress for indeterminate operations.
+
+    Returns:
+        Progress instance with spinner
+    """
+    return Progress(
+        SpinnerColumn(),
+        TextColumn("[progress.description]{task.description}"),
+    )