self_update.py

from time import sleep
import os
# import shutil
from modules import load_config, dots_show, internet_check, get_ota_version, Bcolors


def make_directories_accessible(config):
    markers_dir = f'/home/{config.user}/.ota_markers'
    ota_dir = f'/home/{config.user}/RH_Install-Manager'
    os.system(f"sudo chmod -R 777 {markers_dir} > /dev/null 2>&1") if os.stat(markers_dir).st_uid == 0 else None
    os.system(f"sudo chmod -R 777 {ota_dir} > /dev/null 2>&1") if os.stat(ota_dir).st_uid == 0 else None


def self_update(config):
    internet_flag = internet_check()
    if not internet_flag:
        print(f"\t{Bcolors.RED}Looks like you don't have internet connection.{Bcolors.ENDC}")
        sleep(0.5)
        os.system("./scripts/net_fixer.sh")
        print("Trying again...")
        internet_flag = internet_check()
    if not internet_flag:  # don't change to 'elif' - it is second check after a repair!
        print(f"\t{Bcolors.RED}Looks like you still don't have internet connection. Update canceled.{Bcolors.ENDC}")
        sleep(2)
    else:
        print(f"\t\t{Bcolors.GREEN}Internet connection - OK{Bcolors.ENDC}\n")
        sleep(1.5)
        make_directories_accessible(config)
        old_version_name = get_ota_version(True)
        print(f"\n\n\n\t Please wait: updating process from version {old_version_name}\n\n")
        dots_show(2)
        # if config:  # if config is not empty, then the file exited to load.
        #     shutil.copyfile('~/RH_Install-Manager/updater-config.json', '~/.ota_markers/updater-config.json')
        if config.beta_tester is True:
            source = 'main'
            print("This will be the 'beta' update - may be changed in config wizard.\n")
        elif config.beta_tester is False:
            source = 'stable'
        else:
            source = config.beta_tester
        os.system(f"./scripts/self_updater.sh {source}")
        # if config:  # if the config variable is not empty, then the config file must have existed.
        #   shutil.copyfile("~/.ota_markers/old_RH_Install-Manager/updater-config.json", "~/RH_Install-Manager/updater-config.json")
        new_version_name = get_ota_version(True)
        print(f"""
    
    RotorHazard Manager updated to version {new_version_name}
    
            You can check update-notes.

            """)
        sleep(1)
        make_directories_accessible(config)
        os.system("cp ~/.ota_markers/old_RH_Install-Manager/updater-config.json ~/RH_Install-Manager/updater-config.json")
        # it had some bug with shutil - can be changed when resolved
        if new_version_name != old_version_name:
            os.system("echo OTA was updated > ~/.ota_markers/.was_updated_new")
        else:
            os.system("echo OTA was not updated > ~/.ota_markers/.was_updated_old")


def main():
    config = load_config()
    self_update(config)


if __name__ == "__main__":
    main()