test_setup.py

"""
API Resource Availability Bot - Test Script
Run this to verify your setup before starting the bot
"""
import os
import sys


def check_python_version():
    """Check Python version"""
    print("🐍 Checking Python version...")
    version = sys.version_info
    if version.major >= 3 and version.minor >= 9:
        print(f"   ✅ Python {version.major}.{version.minor}.{version.micro}")
        return True
    else:
        print(f"   ❌ Python {version.major}.{version.minor} (requires 3.9+)")
        return False

def check_dependencies():
    """Check if all dependencies are installed"""
    print("\n📦 Checking dependencies...")
    
    required = [
        'telegram',
        'httpx',
        'dns.resolver',
        'aiosqlite',
        'loguru',
        'bs4',
        'tabulate'
    ]
    
    missing = []
    
    for module in required:
        try:
            if module == 'dns.resolver':
                __import__('dns.resolver')
            elif module == 'bs4':
                __import__('bs4')
            else:
                __import__(module)
            print(f"   ✅ {module}")
        except ImportError:
            print(f"   ❌ {module} (missing)")
            missing.append(module)
    
    if missing:
        print("\n   Install missing packages: pip install -r requirements.txt")
        return False
    
    return True

def check_env_file():
    """Check if .env file exists"""
    print("\n⚙️ Checking configuration...")
    
    if os.path.exists('.env'):
        print("   ✅ .env file found")
        
        # Check if token is set
        with open('.env', 'r') as f:
            content = f.read()
            if 'TELEGRAM_BOT_TOKEN=' in content and 'your_bot_token_here' not in content:
                print("   ✅ Bot token configured")
                return True
            else:
                print("   ⚠️ Bot token not configured")
                print("   Edit .env and add your Telegram bot token")
                return False
    else:
        print("   ❌ .env file not found")
        print("   Run: copy .env.example .env")
        return False

def check_database():
    """Check if database can be initialized"""
    print("\n💾 Checking database...")
    try:
        import sqlite3
        # Try to create a test database
        conn = sqlite3.connect(':memory:')
        conn.close()
        print("   ✅ SQLite working")
        return True
    except Exception as e:
        print(f"   ❌ Database error: {e}")
        return False

def test_scanners():
    """Test if scanner modules can be imported"""
    print("\n🔍 Checking scanner modules...")
    
    try:
        # Test imports without using the classes
        from scanners import (
            AvailabilityChecker,
            DNSChecker,
            PaywallChecker,
            ProtocolChecker,
            SecurityChecker,
        )
        # Mark as used
        _ = (AvailabilityChecker, DNSChecker, ProtocolChecker, 
             PaywallChecker, SecurityChecker)
        print("   ✅ All scanner modules found")
        return True
    except ImportError as e:
        print(f"   ❌ Scanner import error: {e}")
        return False

def test_bot_import():
    """Test if bot can be imported"""
    print("\n🤖 Checking bot module...")
    
    try:
        import config  # noqa: F401
        from aggregator import ResultAggregator  # noqa: F401
        from database import Database  # noqa: F401
        # Mark as used to satisfy linter
        _ = (config, ResultAggregator, Database)
        print("   ✅ Bot modules working")
        return True
    except Exception as e:
        print(f"   ❌ Bot import error: {e}")
        return False

def main():
    """Run all checks"""
    print("=" * 50)
    print("  API Bot - Setup Verification")
    print("=" * 50)
    
    checks = [
        check_python_version(),
        check_dependencies(),
        check_env_file(),
        check_database(),
        test_scanners(),
        test_bot_import()
    ]
    
    print("\n" + "=" * 50)
    
    if all(checks):
        print("✅ All checks passed! Your bot is ready to run!")
        print("\nStart the bot with:")
        print("  python bot.py")
        print("  or")
        print("  run_bot.bat (Windows)")
        print("  ./run_bot.sh (Linux/Mac)")
    else:
        print("❌ Some checks failed. Please fix the issues above.")
        print("\nQuick fixes:")
        print("  1. Install dependencies: pip install -r requirements.txt")
        print("  2. Create .env file: copy .env.example .env")
        print("  3. Add bot token to .env file")
    
    print("=" * 50)

if __name__ == '__main__':
    main()