This commit is contained in:
vvzvlad 2023-04-18 01:48:06 +07:00
parent 68a9054b24
commit ad94bd4493

View File

@ -174,8 +174,13 @@ def create_autogit_systemd(c):
enabled = c.run(f'systemctl is-enabled {service} || true', hide=True).stdout.strip() enabled = c.run(f'systemctl is-enabled {service} || true', hide=True).stdout.strip()
print(f"{service}: {active}, {enabled}") print(f"{service}: {active}, {enabled}")
def mark_original_restored(c, mark):
if mark == "original":
c.run("rm /mnt/data/etc/vestasync/restored", warn=True)
c.run("touch /mnt/data/etc/vestasync/original", warn=True)
if mark == "restored":
c.run("touch /mnt/data/etc/vestasync/restored", warn=True)
c.run("rm /mnt/data/etc/vestasync/original", warn=True)
def reboot(c): def reboot(c):
c.run("reboot > /dev/null 2>&1", warn=True) c.run("reboot > /dev/null 2>&1", warn=True)
@ -277,6 +282,7 @@ def device_install(c):
ppush_the_repo(c) ppush_the_repo(c)
create_automac_systemd(c) create_automac_systemd(c)
create_autogit_systemd(c) create_autogit_systemd(c)
mark_original_restored(c, "original")
reboot(c) reboot(c)
print(f"Install vestasync complete (hostname {hostname}), rebooting target device..\n") print(f"Install vestasync complete (hostname {hostname}), rebooting target device..\n")
@ -312,6 +318,7 @@ def device_restore():
ppush_the_repo(c) ppush_the_repo(c)
create_autogit_systemd(c) create_autogit_systemd(c)
create_automac_systemd(c) create_automac_systemd(c)
mark_original_restored(c, "restored")
if args.user_cmd is not None: if args.user_cmd is not None:
run_user_cmd(c, args.user_cmd_file) run_user_cmd(c, args.user_cmd_file)
reboot(c) reboot(c)