mirror of
https://github.com/ChuckBuilds/LEDMatrix.git
synced 2026-04-10 21:03:01 +00:00
110 lines
3.8 KiB
Python
110 lines
3.8 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Clean setup script for LED Matrix Web Interface V2
|
|
Removes existing virtual environment and creates a fresh one with all dependencies
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
import subprocess
|
|
import logging
|
|
import shutil
|
|
from pathlib import Path
|
|
|
|
# Setup logging
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format='%(asctime)s - %(levelname)s - %(message)s'
|
|
)
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
def main():
|
|
"""Main function to set up a clean virtual environment."""
|
|
# Change to script directory
|
|
script_dir = Path(__file__).parent
|
|
os.chdir(script_dir)
|
|
|
|
venv_path = script_dir / 'venv_web_v2'
|
|
|
|
# Remove existing virtual environment if it exists
|
|
if venv_path.exists():
|
|
logger.info("Removing existing virtual environment...")
|
|
try:
|
|
shutil.rmtree(venv_path)
|
|
logger.info("Existing virtual environment removed")
|
|
except Exception as e:
|
|
logger.error(f"Failed to remove existing virtual environment: {e}")
|
|
return
|
|
|
|
# Create new virtual environment
|
|
logger.info("Creating new virtual environment...")
|
|
try:
|
|
subprocess.check_call([
|
|
sys.executable, '-m', 'venv', str(venv_path)
|
|
])
|
|
logger.info("Virtual environment created successfully")
|
|
except subprocess.CalledProcessError as e:
|
|
logger.error(f"Failed to create virtual environment: {e}")
|
|
return
|
|
|
|
# Get virtual environment Python and pip paths
|
|
if os.name == 'nt': # Windows
|
|
venv_python = venv_path / 'Scripts' / 'python.exe'
|
|
venv_pip = venv_path / 'Scripts' / 'pip.exe'
|
|
else: # Unix/Linux
|
|
venv_python = venv_path / 'bin' / 'python'
|
|
venv_pip = venv_path / 'bin' / 'pip'
|
|
|
|
# Upgrade pip first
|
|
logger.info("Upgrading pip...")
|
|
try:
|
|
subprocess.check_call([
|
|
str(venv_pip), 'install', '--upgrade', 'pip'
|
|
], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
|
|
logger.info("Pip upgraded successfully")
|
|
except subprocess.CalledProcessError as e:
|
|
logger.error(f"Failed to upgrade pip: {e}")
|
|
return
|
|
|
|
# Install dependencies
|
|
logger.info("Installing dependencies...")
|
|
try:
|
|
subprocess.check_call([
|
|
str(venv_pip), 'install', '-r', 'requirements_web_v2.txt'
|
|
], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
|
|
logger.info("Dependencies installed successfully")
|
|
except subprocess.CalledProcessError as e:
|
|
logger.error(f"Failed to install dependencies: {e}")
|
|
return
|
|
|
|
# Install rgbmatrix module from local source
|
|
logger.info("Installing rgbmatrix module...")
|
|
try:
|
|
rgbmatrix_path = script_dir / 'rpi-rgb-led-matrix-master' / 'bindings' / 'python'
|
|
subprocess.check_call([
|
|
str(venv_pip), 'install', '-e', str(rgbmatrix_path)
|
|
], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
|
|
logger.info("rgbmatrix module installed successfully")
|
|
except subprocess.CalledProcessError as e:
|
|
logger.error(f"Failed to install rgbmatrix module: {e}")
|
|
return
|
|
|
|
# Verify key packages are installed
|
|
logger.info("Verifying installation...")
|
|
test_packages = ['flask', 'freetype', 'PIL']
|
|
for package in test_packages:
|
|
try:
|
|
subprocess.check_call([
|
|
str(venv_python), '-c', f'import {package}'
|
|
], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
|
|
logger.info(f"✓ {package} is available")
|
|
except subprocess.CalledProcessError:
|
|
logger.error(f"✗ {package} is NOT available")
|
|
|
|
logger.info("Setup completed successfully!")
|
|
logger.info("You can now run the web interface with:")
|
|
logger.info(" sudo python3 run_web_v2_simple.py")
|
|
|
|
if __name__ == '__main__':
|
|
main() |