setup.py

"""
Setup script for commit message generator.
Creates virtual environment and installs dependencies.
"""

import os
import sys
import subprocess
import venv
from pathlib import Path

def main():
    script_dir = Path(__file__).parent.absolute()
    venv_dir = script_dir / "venv"
    
    print(f"Setting up virtual environment in {venv_dir}")
    
    # Create virtual environment
    if venv_dir.exists():
        print("Virtual environment already exists. Removing old one...")
        import shutil
        shutil.rmtree(venv_dir)
    
    print("Creating virtual environment...")
    venv.create(venv_dir, with_pip=True)
    
    # Determine the correct python executable path
    if os.name == 'nt':  # Windows
        python_exe = venv_dir / "Scripts" / "python.exe"
        pip_exe = venv_dir / "Scripts" / "pip.exe"
    else:  # Unix-like
        python_exe = venv_dir / "bin" / "python"
        pip_exe = venv_dir / "bin" / "pip"
    
    # Install requirements
    requirements_file = script_dir / "requirements.txt"
    if requirements_file.exists():
        print("Installing dependencies...")
        subprocess.run([str(pip_exe), "install", "-r", str(requirements_file)], check=True)
    else:
        print("No requirements.txt found, skipping dependency installation")
    
    print("Setup complete!")
    print(f"Virtual environment created at: {venv_dir}")
    print("\nTo use this script as a git hook:")
    print("1. Copy or symlink this directory to your project's .git/hooks/")
    print("2. Make sure the hook script is executable")

if __name__ == "__main__":
    main()