File size: 3,330 Bytes
2469150 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 |
#!/usr/bin/env python3
"""
Virtual Environment Setup Script for FRED ML
Creates and configures a virtual environment for development
"""
import os
import sys
import subprocess
import venv
from pathlib import Path
def create_venv(venv_path: str = ".venv") -> bool:
"""Create a virtual environment"""
try:
print(f"Creating virtual environment at {venv_path}...")
venv.create(venv_path, with_pip=True)
print("✅ Virtual environment created successfully")
return True
except Exception as e:
print(f"❌ Failed to create virtual environment: {e}")
return False
def install_requirements(venv_path: str = ".venv") -> bool:
"""Install requirements in the virtual environment"""
try:
# Determine the pip path
if os.name == 'nt': # Windows
pip_path = os.path.join(venv_path, "Scripts", "pip")
else: # Unix/Linux/macOS
pip_path = os.path.join(venv_path, "bin", "pip")
print("Installing requirements...")
subprocess.run([pip_path, "install", "-r", "requirements.txt"], check=True)
print("✅ Requirements installed successfully")
return True
except subprocess.CalledProcessError as e:
print(f"❌ Failed to install requirements: {e}")
return False
except Exception as e:
print(f"❌ Unexpected error installing requirements: {e}")
return False
def activate_venv_instructions(venv_path: str = ".venv"):
"""Print activation instructions"""
print("\n📋 Virtual Environment Setup Complete!")
print("=" * 50)
if os.name == 'nt': # Windows
activate_script = os.path.join(venv_path, "Scripts", "activate")
print(f"To activate the virtual environment, run:")
print(f" {activate_script}")
else: # Unix/Linux/macOS
activate_script = os.path.join(venv_path, "bin", "activate")
print(f"To activate the virtual environment, run:")
print(f" source {activate_script}")
print("\nOr use the provided Makefile target:")
print(" make venv-activate")
print("\nTo deactivate, simply run:")
print(" deactivate")
def main():
"""Main setup function"""
print("🏗️ FRED ML - Virtual Environment Setup")
print("=" * 40)
venv_path = ".venv"
# Check if virtual environment already exists
if os.path.exists(venv_path):
print(f"⚠️ Virtual environment already exists at {venv_path}")
response = input("Do you want to recreate it? (y/N): ").lower().strip()
if response == 'y':
import shutil
shutil.rmtree(venv_path)
print("Removed existing virtual environment")
else:
print("Using existing virtual environment")
activate_venv_instructions(venv_path)
return
# Create virtual environment
if not create_venv(venv_path):
sys.exit(1)
# Install requirements
if not install_requirements(venv_path):
print("⚠️ Failed to install requirements, but virtual environment was created")
print("You can manually install requirements after activation")
# Print activation instructions
activate_venv_instructions(venv_path)
if __name__ == "__main__":
main() |