stellaris-dlc-unlocker/Libs/LauncherReinstall.py

import os
from shutil import rmtree, move
from subprocess import Popen
from PyQt5 import QtCore
from time import sleep
from glob import glob
from re import search
from pathlib import Path


class ReinstallThread(QtCore.QThread):
    # progress_signal = QtCore.pyqtSignal(int)
    error_signal = QtCore.pyqtSignal(Exception)
    continue_reinstall = QtCore.pyqtSignal(str)

    def __init__(self, msi_path, paradox_folder1, paradox_folder2, paradox_folder3, paradox_folder4,
                 launcher_downloaded, downloaded_launcher_dir, user_logon_name):
        super().__init__()
        self.msi_path = msi_path
        self.user_logon_name = user_logon_name
        self.paradox_folder1 = paradox_folder1
        self.paradox_folder2 = paradox_folder2
        self.paradox_folder3 = paradox_folder3
        self.paradox_folder4 = paradox_folder4
        self.launcher_downloaded = launcher_downloaded
        self.downloaded_launcher_dir = downloaded_launcher_dir

    def run(self):
        user_home = os.path.join("C:\\Users", self.user_logon_name)
        if Path(self.paradox_folder1) == Path(self.msi_path):
            self.paradox_folder1 = os.path.join(user_home, "AppData", "Local", "Programs", "Paradox Interactive",
                                                "launcher")
        latest_file = None
        latest_version = (0, 0)

        def extract_version(filename):
            match = search(r'launcher-installer-windows_(\d+\.\d+)', filename)
            if match:
                return tuple(map(int, match.group(1).split('.')))
            return None

        msi_files = glob(os.path.join(self.msi_path, "launcher-installer-windows_*.msi"))
        if self.launcher_downloaded:
            print('Alt unlock. Deleting all other launches')
            msi_files = glob(os.path.join(self.msi_path, "launcher-installer-windows*.msi"))
            try:
                for file_path in msi_files:
                    try:
                        os.remove(file_path)
                        print(f"Delete {file_path}")
                    except Exception as e:
                        print(f"Unable to delete {file_path}: {e}")
            except:
                pass
            try:
                move(self.downloaded_launcher_dir, self.msi_path)
                print(f"Launcher moved: {self.msi_path}")
            except Exception as e:
                print(f"Unable to move launcher: {e}")
                self.error_signal.emit(e)
            msi_files = glob(os.path.join(self.msi_path, "launcher-installer-windows_*.msi"))

        if msi_files:
            msi_path = msi_files[0]

        else:
            msi_path = os.path.join(self.msi_path, "launcher-installer-windows.msi")
            if os.path.exists(msi_path):
                pass
            else:
                self.error_signal.emit('launcher_installer not found!')
        print(f'Game path: {self.msi_path}')
        print(f'Launcher Path: {msi_path}\nPath exists: {os.path.exists(msi_path)}')
        print(f'Deleting launcher...')
        try:

            self.paradox_remove(self.paradox_folder1, self.paradox_folder2, self.paradox_folder3, self.paradox_folder4)

            uninstall = Popen(['cmd.exe', '/c', 'msiexec', '/uninstall',
                               msi_path, '/quiet'], shell=True)
            # output, error = uninstall.communicate()
            # if uninstall.returncode != 0:
            #     print("Произошла ошибка при удалении:", error.decode())

            uninstall.wait()
            # self.progress_signal.emit(33)
            sleep(1)
            print(f'Installing launcher...')
            install = Popen(
                ['cmd.exe', '/c', 'msiexec', '/package', msi_path,
                 '/quiet', 'CREATE_DESKTOP_SHORTCUT=0'], shell=True)
            # output, error = install.communicate()
            # if uninstall.returncode != 0:
            #     print("Произошла ошибка при установке:", error.decode())
            install.wait()
            self.continue_reinstall.emit(self.paradox_folder1)
        except Exception as e:
            self.error_signal.emit(e)

    @staticmethod
    def paradox_remove(paradox_folder1, paradox_folder2, paradox_folder3, paradox_folder4):
        # user_home = os.path.expanduser("~")
        # paradox_folder1 = os.path.join(user_home, "AppData", "Local", "Programs", "Paradox Interactive")
        # paradox_folder2 = os.path.join(user_home, "AppData", "Local", "Paradox Interactive")
        # paradox_folder3 = os.path.join(user_home, "AppData", "Roaming", "Paradox Interactive")
        # paradox_folder4 = os.path.join(user_home, "AppData", "Roaming", "paradox-launcher-v2")
        try:
            if os.path.exists(paradox_folder1):
                print(f'Removing {paradox_folder1}')
                rmtree(paradox_folder1)
        except Exception as e:
            print(f'Cant delete {e}')
            pass

        try:
            if os.path.exists(paradox_folder2):
                print(f'Removing {paradox_folder2}')
                rmtree(paradox_folder2)
        except Exception as e:
            print(f'Cant delete {e}')
            pass

        try:
            if os.path.exists(paradox_folder3):
                print(f'Removing {paradox_folder3}')
                rmtree(paradox_folder3)
        except Exception as e:
            print(f'Cant delete {e}')
            pass

        try:
            if os.path.exists(paradox_folder4):
                print(f'Removing {paradox_folder4}')
                rmtree(paradox_folder4)
        except Exception as e:
            print(f'Cant delete {e}')
            pass