51 lines
2.0 KiB
Python
51 lines
2.0 KiB
Python
import os
|
|
import sys
|
|
import subprocess
|
|
import importlib
|
|
|
|
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
|
|
|
|
|
|
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):
|
|
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()
|