config.py

import os
import sys
import logging
from pathlib import Path
from typing import Optional, Dict, Any

ROOT = Path(__file__).resolve().parent

# Configure logging
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
)
logger = logging.getLogger(__name__)


def load_environment() -> Dict[str, Any]:
    """Load and parse environment variables from .env file."""
    try:
        from dotenv import load_dotenv
        load_dotenv(ROOT / '.env')
    except ImportError:
        logger.warning("python-dotenv not installed; skipping .env loading")

    return {
        'token': os.getenv('DISCORD_TOKEN'),
        'my_id': os.getenv('MY_ID'),
        'oauth_client_id': os.getenv('OAUTH_CLIENT_ID'),
        'oauth_client_secret': os.getenv('OAUTH_CLIENT_SECRET'),
        'json_folder': Path(os.getenv('JSON_FOLDER', './friends_data')),
        'cache_dir': Path(os.getenv('CACHE_DIR', './icon_cache')),
        'canvas_size': os.getenv('CANVAS_SIZE', '50,50'),
        'dpi': int(os.getenv('DPI', '300')),
        'avatar_pixels': int(os.getenv('AVATAR_PIXELS', '256')),
        'avatar_zoom': float(os.getenv('AVATAR_ZOOM', '0.18')),
        'peer_weight': float(os.getenv('PEER_WEIGHT', '1.5')),
        'hero_weight': float(os.getenv('HERO_WEIGHT', '2.5')),
        'heatmap_grid': int(os.getenv('HEATMAP_GRID', '100')),
        'min_delay': float(os.getenv('MIN_DELAY', '1.5')),
        'max_delay': float(os.getenv('MAX_DELAY', '4.0')),
        'batch_size': int(os.getenv('BATCH_SIZE', '30')),
        'batch_pause': float(os.getenv('BATCH_PAUSE', '60')),
        'refresh_days': int(os.getenv('REFRESH_DAYS', '7')),
    }


def validate_config(config: Dict[str, Any], required_keys: list) -> bool:
    """Validate that required keys exist and are non-empty."""
    missing = [k for k in required_keys if not config.get(k)]
    if missing:
        logger.error(f"Missing required configuration: {', '.join(missing)}")
        return False
    return True


def ensure_env_file() -> bool:
    """Ensure .env file exists, with helpful error message if not."""
    env_path = ROOT / '.env'
    if env_path.exists():
        return True

    logger.error(f"No .env file found at {env_path}")
    example_path = ROOT / '.env.example'
    if example_path.exists():
        logger.error(f"Please create .env from the example at {example_path}")
    logger.error("Set DISCORD_TOKEN and MY_ID at minimum.")
    return False