#!/usr/local/bin/python3

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()\''
subprocess.check_output(shlex.split(mem_cmd))

print("\n\nListing remote files...")
ls_cmd = './tools/wasptool --verbose --eval \'from shell import ls ; ls(\"/flash/logs/sleep/\")\''
out = subprocess.check_output(shlex.split(ls_cmd)).decode()
files = re.findall(r"\d*\.csv", out)
print(f"Found files {', '.join(files)}")

reset_cmd = './tools/wasptool --verbose --reset'

to_rm = files

print("\n\n")
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")
        tqdm.write(f"Succesfully removed to './logs/sleep/{fi}'")
    except Exception as e:
        tqdm.write(f"Error happened while removing {fi}")

    tqdm.write("Restarting watch.")
    out = subprocess.check_output(shlex.split(reset_cmd))
    time.sleep(10)

    tqdm.write("\n\n")