setup.py

#!/usr/bin/env python
"""
Setup script for Data Lake API project.
This script helps with initial project setup and configuration.
"""

import os
import sys
import subprocess
from pathlib import Path


def run_command(command, description):
    """Run a command and handle errors."""
    print(f"🔄 {description}...")
    try:
        result = subprocess.run(command, shell=True, check=True, capture_output=True, text=True)
        print(f"✅ {description} completed successfully")
        return True
    except subprocess.CalledProcessError as e:
        print(f"❌ {description} failed: {e.stderr}")
        return False


def check_python_version():
    """Check if Python version is compatible."""
    version = sys.version_info
    if version.major < 3 or (version.major == 3 and version.minor < 8):
        print("❌ Python 3.8+ is required")
        return False
    print(f"✅ Python {version.major}.{version.minor}.{version.micro} is compatible")
    return True


def check_dependencies():
    """Check if required system dependencies are installed."""
    dependencies = {
        'python': 'python --version',
        'pip': 'pip --version',
        'git': 'git --version'
    }
    
    for name, command in dependencies.items():
        if not run_command(command, f"Checking {name}"):
            return False
    return True


def setup_virtual_environment():
    """Create and activate virtual environment."""
    venv_path = Path("venv")
    
    if venv_path.exists():
        print("✅ Virtual environment already exists")
        return True
    
    if not run_command("python -m venv venv", "Creating virtual environment"):
        return False
    
    print("📝 To activate the virtual environment:")
    print("   Windows: venv\\Scripts\\activate")
    print("   macOS/Linux: source venv/bin/activate")
    return True


def install_requirements():
    """Install Python requirements."""
    if not Path("requirements.txt").exists():
        print("❌ requirements.txt not found")
        return False
    
    return run_command("pip install -r requirements.txt", "Installing Python dependencies")


def setup_database():
    """Provide database setup instructions."""
    print("\n🗄️  Database Setup Required:")
    print("1. Install PostgreSQL if not already installed")
    print("2. Create a database named 'api_project'")
    print("3. Update database settings in api_project/api_project/settings.py")
    print("4. Run migrations: python manage.py migrate")
    print("5. Create superuser: python manage.py createsuperuser")


def main():
    """Main setup function."""
    print("🚀 Data Lake API Setup Script")
    print("=" * 40)
    
    # Check system requirements
    if not check_python_version():
        sys.exit(1)
    
    if not check_dependencies():
        print("❌ Please install missing dependencies")
        sys.exit(1)
    
    # Setup virtual environment
    if not setup_virtual_environment():
        sys.exit(1)
    
    # Install requirements
    print("\n📦 Installing Python dependencies...")
    print("⚠️  Make sure to activate your virtual environment first!")
    response = input("Have you activated the virtual environment? (y/n): ")
    if response.lower() != 'y':
        print("Please activate the virtual environment and run this script again")
        sys.exit(1)
    
    if not install_requirements():
        sys.exit(1)
    
    # Database setup instructions
    setup_database()
    
    print("\n🎉 Setup completed successfully!")
    print("\n📋 Next steps:")
    print("1. Activate virtual environment")
    print("2. Set up PostgreSQL database")
    print("3. Update settings.py with database credentials")
    print("4. Run: python manage.py migrate")
    print("5. Run: python manage.py createsuperuser")
    print("6. Run: python manage.py runserver")
    print("7. Visit: http://127.0.0.1:8000/")


if __name__ == "__main__":
    main()