forked from tpd94/CDRM-Project
104 lines
2.9 KiB
Python
104 lines
2.9 KiB
Python
import sys
|
|
import os
|
|
import subprocess
|
|
import venv
|
|
|
|
|
|
def version_check():
|
|
major_version = sys.version_info.major
|
|
minor_version = sys.version_info.minor
|
|
|
|
if major_version >= 3:
|
|
if minor_version >= 12:
|
|
return
|
|
else:
|
|
exit("Python version 3.12 or higher is required")
|
|
else:
|
|
exit("Python 2 detected, Python version 3.12 or higher is required")
|
|
|
|
|
|
def pip_check():
|
|
try:
|
|
import pip
|
|
|
|
return
|
|
except ImportError:
|
|
exit("Pip is not installed")
|
|
|
|
|
|
def venv_check():
|
|
# Check if we're already inside a virtual environment
|
|
if hasattr(sys, "real_prefix") or (
|
|
hasattr(sys, "base_prefix") and sys.base_prefix != sys.prefix
|
|
):
|
|
return
|
|
|
|
venv_path = os.path.join(os.getcwd(), "cdrm-venv")
|
|
venv_python = (
|
|
os.path.join(venv_path, "bin", "python")
|
|
if os.name != "nt"
|
|
else os.path.join(venv_path, "Scripts", "python.exe")
|
|
)
|
|
|
|
# If venv already exists, restart script using its Python
|
|
if os.path.exists(venv_path):
|
|
subprocess.call([venv_python] + sys.argv)
|
|
sys.exit()
|
|
|
|
# Ask user for permission to create a virtual environment
|
|
answer = ""
|
|
while not answer or answer[0].upper() not in {"Y", "N"}:
|
|
answer = input(
|
|
"Program is not running from a venv. To maintain compatibility and dependencies, this program must be run from one.\n"
|
|
"Would you like me to create one for you? (Y/N): "
|
|
)
|
|
|
|
if answer[0].upper() == "Y":
|
|
print("Creating virtual environment...")
|
|
venv.create(venv_path, with_pip=True)
|
|
subprocess.call([venv_python] + sys.argv)
|
|
sys.exit()
|
|
else:
|
|
print("Exiting program. Please run it from a virtual environment next time.")
|
|
sys.exit(1)
|
|
|
|
|
|
def requirements_check():
|
|
try:
|
|
import pywidevine
|
|
import pyplayready
|
|
import flask
|
|
import flask_cors
|
|
import yaml
|
|
import mysql.connector
|
|
|
|
return
|
|
except ImportError:
|
|
while True:
|
|
user_input = (
|
|
input("Missing packages. Do you want to install them? (Y/N): ")
|
|
.strip()
|
|
.upper()
|
|
)
|
|
if user_input == "Y":
|
|
print("Installing packages from requirements.txt...")
|
|
subprocess.check_call(
|
|
[sys.executable, "-m", "pip", "install", "-r", "requirements.txt"]
|
|
)
|
|
print("Installation complete.")
|
|
break
|
|
elif user_input == "N":
|
|
print("Dependencies required, please install them and run again.")
|
|
sys.exit()
|
|
else:
|
|
print("Invalid input. Please enter 'Y' to install or 'N' to exit.")
|
|
|
|
|
|
def run_python_checks():
|
|
if getattr(sys, "frozen", False): # Check if running from PyInstaller
|
|
return
|
|
version_check()
|
|
pip_check()
|
|
venv_check()
|
|
requirements_check()
|