Archipelago/ModuleUpdate.py

import os
import sys
import subprocess
import importlib


if sys.version_info < (3, 8, 6):
    raise RuntimeError("Incompatible Python Version. 3.8.7+ is supported.")

update_ran = hasattr(sys, "frozen") and getattr(sys, "frozen")  # don't run update if environment is frozen/compiled


def update_command():
    subprocess.call([sys.executable, '-m', 'pip', 'install', '-r', 'requirements.txt', '--upgrade'])


naming_specialties = {"PyYAML": "yaml",  # PyYAML is imported as the name yaml
                      "maseya-z3pr": "maseya"}


def update():
    global update_ran
    if not update_ran:
        update_ran = True
        path = os.path.join(os.path.dirname(sys.argv[0]), 'requirements.txt')
        if not os.path.exists(path):
            path = os.path.join(os.path.dirname(__file__), 'requirements.txt')
        with open(path) as requirementsfile:
            for line in requirementsfile.readlines():
                module, remote_version = line.split(">=")
                module = naming_specialties.get(module, module)
                try:
                    module = importlib.import_module(module)
                except:
                    import traceback
                    traceback.print_exc()
                    input(f'Required python module {module} not found, press enter to install it')
                    update_command()
                    return
                else:
                    if hasattr(module, "__version__"):
                        module_version = module.__version__
                        module = module.__name__  # also unloads the module to make it writable
                        if type(module_version) == str:
                            module_version = tuple(int(part.strip()) for part in module_version.split("."))
                        remote_version = tuple(int(part.strip()) for part in remote_version.split("."))
                        if module_version < remote_version:
                            input(f'Required python module {module} is outdated ({module_version}<{remote_version}),'
                                  ' press enter to upgrade it')
                            update_command()
                            return


if __name__ == "__main__":
    update()