From 0bbf8d2c31d45e261db16b193bebfa407196034c Mon Sep 17 00:00:00 2001
From: Rim <info@rimmyscorner.com>
Date: Fri, 28 Feb 2025 02:21:12 -0500
Subject: [PATCH] chore(setup.py): complete code refactor

complete code refactor by Claude Sonnet 3.7
---
 setup.py | 128 +++++++++++++++++++++++++++++++++++++++++++------------
 1 file changed, 101 insertions(+), 27 deletions(-)

diff --git a/setup.py b/setup.py
index 4eb4261..05d0d15 100644
--- a/setup.py
+++ b/setup.py
@@ -1,38 +1,112 @@
+#!/usr/bin/env python3
+"""
+Setup script for Modern Warfare 2019 Advanced Statistics Tracker.
+Creates a virtual environment and installs required dependencies.
+"""
+
 import os
 import subprocess
+import sys
 import venv
+from pathlib import Path
 
-def deps_exists():
-    return os.path.exists('deps')
 
-def create_venv():
-    venv.create('venv', with_pip=True)
-    # Create activation scripts
-    with open("venv.ps1", "w") as f:
-        f.write("venv\\Scripts\\Activate.ps1")
+def check_dependencies():
+    """Check if the dependencies directory exists."""
+    deps_dir = Path('deps')
+    if not deps_dir.exists():
+        print("Error: 'deps' directory not found!")
+        print("Please ensure the dependencies directory exists before running setup.")
+        sys.exit(1)
+    return deps_dir
 
-    with open("venv.bat", "w") as f:
-        f.write("venv\\Scripts\\activate")
 
-def setup_tools():
-    subprocess.check_call([os.path.join('venv', 'Scripts', 'pip'), 'install', '--no-index', '--find-links=deps', 'pip', 'wheel', 'build', 'pyinstaller'])
+def create_virtual_environment():
+    """Create and configure a Python virtual environment."""
+    print("Creating virtual environment...")
+    venv_dir = Path('venv')
+    
+    # Create the virtual environment with pip
+    venv.create(venv_dir, with_pip=True)
+    
+    # Create activation scripts for different platforms
+    scripts = {
+        "venv.ps1": "venv\\Scripts\\Activate.ps1",
+        "venv.bat": "venv\\Scripts\\activate",
+        "venv.sh": "source venv/bin/activate"  # Added for Unix/Linux users
+    }
+    
+    for filename, content in scripts.items():
+        with open(filename, "w") as f:
+            f.write(content)
+    
+    return venv_dir
 
-def setup_api():
-    wheel_path = os.path.join('deps', 'cod_api-2.0.2-py3-none-any.whl')
-    subprocess.check_call([os.path.join('venv', 'Scripts', 'pip'), 'install', '--no-index', '--find-links=deps', wheel_path])
 
-def remove_deprecated_packages():
-    subprocess.check_call([os.path.join('venv', 'Scripts', 'pip'), 'uninstall', 'enum34', '-y'])
+def get_pip_path(venv_dir):
+    """Get the platform-specific path to pip in the virtual environment."""
+    if sys.platform == 'win32':
+        return venv_dir / 'Scripts' / 'pip'
+    return venv_dir / 'bin' / 'pip'
+
+
+def install_packages(pip_path, deps_dir):
+    """Install required packages from the local dependencies directory."""
+    print("Installing base packages...")
+    base_packages = ['pip', 'wheel', 'build', 'pyinstaller']
+    subprocess.check_call([
+        str(pip_path), 'install', 
+        '--no-index', 
+        f'--find-links={deps_dir}',
+        *base_packages
+    ])
+    
+    print("Installing COD API package...")
+    cod_api_wheel = 'cod_api-2.0.2-py3-none-any.whl'
+    subprocess.check_call([
+        str(pip_path), 'install',
+        '--no-index',
+        f'--find-links={deps_dir}',
+        str(deps_dir / cod_api_wheel)
+    ])
+
+
+def clean_environment(pip_path):
+    """Remove deprecated or conflicting packages."""
+    print("Removing deprecated packages...")
+    deprecated_packages = ['enum34']
+    subprocess.check_call([
+        str(pip_path), 'uninstall', *deprecated_packages, '-y'
+    ])
+
+
+def main():
+    """Main setup function."""
+    print("Starting setup for Modern Warfare 2019 Statistics Tracker...")
+    
+    # Check for dependencies directory
+    deps_dir = check_dependencies()
+    
+    # Create and configure the virtual environment
+    venv_dir = create_virtual_environment()
+    
+    # Get the appropriate pip path
+    pip_path = get_pip_path(venv_dir)
+    
+    # Install required packages
+    install_packages(pip_path, deps_dir)
+    
+    # Clean up environment
+    clean_environment(pip_path)
+    
+    print("\nSetup completed successfully!")
+    print(f"To activate the virtual environment, run:")
+    if sys.platform == 'win32':
+        print("  - PowerShell: .\\venv.ps1")
+        print("  - Command Prompt: venv.bat")
+    else:
+        print("  - Bash/Zsh: source venv.sh")
+
 
 if __name__ == "__main__":
-    if not deps_exists():
-        print("Error: 'deps' directory does not exist!")
-        exit(1)    
-
-    print("Creating virtual environment...")
-    create_venv()
-    print("Setting up virtual environment...")
-    setup_tools()
-    setup_api()
-    remove_deprecated_packages()
-    print("Setup completed successfully.")
+    main()
\ No newline at end of file