From cba83e780a9350eb45eb1c2f09fc87f5d389e1ac Mon Sep 17 00:00:00 2001 From: jaseg Date: Sun, 20 May 2018 15:54:03 +0200 Subject: Move documentation and simulation stuff to its own subdir --- doc/spectrum_progress.py | 36 ++++++++++++++++++++++++++++++++++++ 1 file changed, 36 insertions(+) create mode 100644 doc/spectrum_progress.py (limited to 'doc/spectrum_progress.py') diff --git a/doc/spectrum_progress.py b/doc/spectrum_progress.py new file mode 100644 index 0000000..35a9d65 --- /dev/null +++ b/doc/spectrum_progress.py @@ -0,0 +1,36 @@ +#!/usr/bin/env python3 +import sqlite3 +import argparse +import time + +import tqdm + +if __name__ == '__main__': + parser = argparse.ArgumentParser() + parser.add_argument('-d', '--database', nargs='?', default='spectra.sqlite3') + parser.add_argument('-u', '--update-delay', nargs='?', type=float, default=1.0) + parser.add_argument('max_step', nargs='?', type=int, default=250) + args = parser.parse_args() + + db = sqlite3.connect(args.database) + def current_step(): + step, = db.execute( + 'SELECT MAX(step) FROM measurements WHERE run_id = (SELECT MAX(run_id) FROM runs)' + ).fetchone() + return int(step) + + def step_gen(): + while True: + step = current_step() + yield step + if step >= args.max_step: + break + time.sleep(args.update_delay) + + bar = tqdm.tqdm(total=args.max_step) + while True: + try: + for step in step_gen(): + bar.update(step - bar.n) + except: + time.sleep(args.update_delay) -- cgit