env_creator.py

import os
import subprocess
from pathlib import Path
import platform
import argparse  # 引入 argparse 模块

class VirtualEnvManager:
    def __init__(self, base_dir):
        self.base_dir = Path(base_dir)

    def setup_virtualenv(self, env_name, custom_env_dir=None, mirror_url=None):
        requirements_path = self.base_dir / "requirements.txt"
        env_dir = Path(custom_env_dir) if custom_env_dir else self.base_dir / env_name / f"{env_name}_venv"

        if not env_dir.exists():
            env_dir.mkdir(parents=True, exist_ok=True)
            print(f"Creating virtual environment at {env_dir} for {env_name}...")
            subprocess.run(["python", "-m", "venv", str(env_dir)], check=True)
            
            # Upgrade pip with or without mirror URL
            pip_upgrade_command = ["python", "-m", "pip", "install", "--upgrade", "pip"]
            if mirror_url:
                pip_upgrade_command += ["-i", mirror_url]
            subprocess.run(pip_upgrade_command, check=True)
            
            print(requirements_path)
            if requirements_path.exists():
                print(f"Installing dependencies for {env_name} from {requirements_path}...")
                pip_command = env_dir / "bin" / "pip" if platform.system() != "Windows" else env_dir / "Scripts" / "pip"
                install_command = [str(pip_command), "install", "-r", str(requirements_path)]
                
                # Add mirror URL if provided
                if mirror_url:
                    install_command += ["-i", mirror_url]
                
                subprocess.run(install_command, check=True)
            else:
                print(f"No requirements.txt found for {env_name}. Skipping dependencies installation.")
        else:
            print(f"Virtual environment for {env_name} already exists at {env_dir}.")

        self._print_activate_instruction(env_dir)

    def _print_activate_instruction(self, env_dir):
        if platform.system() == "Windows":
            activate_script = env_dir / "Scripts" / "activate.bat"
            print(f"\nTo activate the virtual environment on Windows, run:")
        else:
            activate_script = env_dir / "bin" / "activate"
            print(f"\nTo activate the virtual environment on Unix or MacOS, run:")

        print(f"{activate_script}")

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="Virtual Environment Manager")
    parser.add_argument("base_dir", type=str, help="The base directory where virtual environments will be created")
    parser.add_argument("env_name", type=str, help="The name of the virtual environment")
    parser.add_argument("--mirror_url", type=str, default=None, help="The mirror URL for pip installations")

    args = parser.parse_args()

    manager = VirtualEnvManager(args.base_dir)
    manager.setup_virtualenv(args.env_name, mirror_url=args.mirror_url)