log_system_info()
# Step 2: Extract or copy source files if args.source and Path(args.source).exists(): source_dir = extract_installer_package(args.source) else: # Assume script is running from the extracted package folder source_dir = Path.cwd() / "files" # Modify as needed if not source_dir.exists(): log.error("No source files found. Use --source or prepare 'files' folder.") sys.exit(1) umtv2-umtpro-ultimateunisoc-v0.1-installer
# Step 6: Add to PATH if not args.no_path: add_to_path(DEFAULT_INSTALL_PATH) log_system_info() # Step 2: Extract or copy source
def add_to_path(install_path): """Add tool directory to system PATH (optional).""" try: import winreg key = winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE, r"SYSTEM\CurrentControlSet\Control\Session Manager\Environment", 0, winreg.KEY_SET_VALUE) current_path, _ = winreg.QueryValueEx(key, "Path") if str(install_path) not in current_path: new_path = current_path + ";" + str(install_path) winreg.SetValueEx(key, "Path", 0, winreg.REG_EXPAND_SZ, new_path) winreg.CloseKey(key) log.info("Added to system PATH") # Notify system about change subprocess.run(["refreshenv.exe"], shell=True, capture_output=True) except Exception as e: log.error(f"Failed to update PATH: {e}") _ = winreg.QueryValueEx(key
def run_as_admin(): """Restart script with admin rights.""" if not is_admin(): log.warning("Requesting administrator privileges...") script = os.path.abspath(sys.argv[0]) params = ' '.join([script] + sys.argv[1:]) try: ctypes.windll.shell32.ShellExecuteW(None, "runas", sys.executable, params, None, 1) except Exception as e: log.error(f"Failed to elevate: {e}") sys.exit(1) sys.exit()
def create_uninstaller(install_path): """Create uninstall registry and script.""" uninstall_script = install_path / "uninstall.py" uninstall_content = f'''#!/usr/bin/env python3 import shutil, sys, os from pathlib import Path target = Path(r"{install_path}") if input(f"Delete {{target}}? (y/N): ").lower() == 'y': shutil.rmtree(target) print("Removed", target) # Optionally remove from PATH here ''' uninstall_script.write_text(uninstall_content) log.info(f"Uninstaller created at {uninstall_script}")