check_python.py

#!/usr/bin/env python3
"""
Python environment validation script.
Checks for required Python modules and common issues.
"""
import sys
import subprocess
import platform

def check_lzma():
    """Check if Python has lzma support."""
    try:
        import lzma
        return True, None
    except ImportError as e:
        return False, str(e)

def check_required_modules():
    """Check for critical required modules."""
    issues = []
    
    # Check lzma (critical for PyTorch Lightning)
    lzma_ok, lzma_error = check_lzma()
    if not lzma_ok:
        issues.append({
            'module': 'lzma',
            'error': lzma_error,
            'critical': True,
            'fix': get_lzma_fix_instructions()
        })
    
    return issues

def get_lzma_fix_instructions():
    """Get platform-specific fix instructions for lzma."""
    system = platform.system().lower()
    
    if system == 'darwin':  # macOS
        return """Fix Python lzma support on macOS:

1. Ensure xz is installed:
   brew install xz

2. Rebuild Python with lzma support:
   export LDFLAGS="-L$(brew --prefix xz)/lib"
   export CPPFLAGS="-I$(brew --prefix xz)/include"
   export PKG_CONFIG_PATH="$(brew --prefix xz)/lib/pkgconfig:$PKG_CONFIG_PATH"
   pyenv install --force 3.11.6

   Or run: /tmp/fix_python_lzma.sh

3. Verify: python3 -c "import lzma; print('OK')"
"""
    elif system == 'linux':
        return """Fix Python lzma support on Linux:

1. Install xz development libraries:
   sudo apt-get install liblzma-dev  # Debian/Ubuntu
   sudo yum install xz-devel          # RHEL/CentOS

2. Rebuild Python or reinstall Python with lzma support
"""
    else:
        return """Fix Python lzma support:

Rebuild Python with lzma/xz library support.
Check your Python distribution documentation for details.
"""

def main():
    """Run all checks."""
    print("🔍 Checking Python environment...")
    print("")
    
    issues = check_required_modules()
    
    if not issues:
        print("✅ All checks passed!")
        return 0
    
    print("❌ Found issues:")
    print("")
    
    for issue in issues:
        severity = "CRITICAL" if issue['critical'] else "WARNING"
        print(f"[{severity}] Missing module: {issue['module']}")
        print(f"   Error: {issue['error']}")
        print("")
        print("   Fix instructions:")
        print(issue['fix'])
        print("")
    
    return 1

if __name__ == '__main__':
    sys.exit(main())