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

new: add tqdm to scripts

Browse source
This commit is contained in:
thiswillbeyourgithub 2022-04-23 13:34:42 +02:00
parent 7e88ff8929
commit 19470cf07c
2 changed files with 9 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
pull_sleep_data.py
View file

@ -6,6 +6,7 @@ import os
import subprocess
import shlex
import re
from tqdm import tqdm
mode = "all" # download "all" files or only "latest"
@ -30,7 +31,7 @@ else:
raise Exception("Wrong value for 'mode'")
print("\n\n")
for fi in to_dl:
for fi in tqdm(to_dl):
if os.path.exists(f"./logs/sleep/{fi}"):
print(f"Skipping file {fi}: already exists")
else:

13
rm_sleep_data.py
View file

@ -4,6 +4,7 @@ import time
import subprocess
import shlex
import re
from tqdm import tqdm
print("\n\nRunning gc.collect()...")
mem_cmd = './tools/wasptool --verbose --eval \'wasp.gc.collect()\''
@ -20,19 +21,19 @@ reset_cmd = './tools/wasptool --verbose --reset'
to_rm = files
print("\n\n")
for fi in to_rm:
print(f"Removing file '{fi}'")
for fi in tqdm(to_rm):
tqdm.write(f"Removing file '{fi}'")
rm_cmd = f'./tools/wasptool --verbose --eval \'from shell import rm ; rm(\"logs/sleep/{fi}\")\''
try:
out = subprocess.check_output(shlex.split(rm_cmd))
if b"Watch reported error" in out:
raise Exception("Watch reported error")
print(f"Succesfully removed to './logs/sleep/{fi}'")
tqdm.write(f"Succesfully removed to './logs/sleep/{fi}'")
except Exception as e:
print(f"Error happened while removing {fi}")
tqdm.write(f"Error happened while removing {fi}")
print("Restarting watch.")
tqdm.write("Restarting watch.")
out = subprocess.check_output(shlex.split(reset_cmd))
time.sleep(10)
print("\n\n")
tqdm.write("\n\n")