PeterCxy/wasp-os
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

autoreboot when downloading files + use tqdm

Browse source
This commit is contained in:
thiswillbeyourgithub 2022-04-27 22:06:43 +02:00
parent 025342dd8f
commit 2b1b69f1b4
1 changed files with 7 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
pull_sleep_data.py
View file

@ -9,6 +9,7 @@ import re
from tqdm import tqdm
mode = "all" # download "all" files or only "latest"
autoreboot = True # reboot between each download, to avoid memory issues
print("\n\nRunning gc.collect()...")
mem_cmd = './tools/wasptool --verbose --eval \'wasp.gc.collect()\''
@ -33,20 +34,20 @@ else:
print("\n\n")
for fi in tqdm(to_dl):
if os.path.exists(f"./logs/sleep/{fi}"):
print(f"Skipping file {fi}: already exists")
tqdm.write(f"Skipping file {fi}: already exists")
else:
print(f"Downloading file '{fi}'")
tqdm.write(f"Downloading file '{fi}'")
pull_cmd = f'./tools/wasptool --verbose --pull logs/sleep/{fi}'
try:
out = subprocess.check_output(shlex.split(pull_cmd))
if b"Watch reported error" in out:
raise Exception("Watch reported error")
print(f"Succesfully downloaded to './logs/sleep/{fi}'")
tqdm.write(f"Succesfully downloaded to './logs/sleep/{fi}'")
except Exception as e:
print(f"Error happened while downloading {fi}, deleting local incomplete file")
tqdm.write(f"Error happened while downloading {fi}, deleting local incomplete file")
os.system(f"rm ./logs/sleep/{fi}")
if mode == "all":
print("Restarting watch.")
if mode == "all" and autoreboot:
tqdm.write("Restarting watch.")
out = subprocess.check_output(shlex.split(reset_cmd))
time.sleep(10)