diff options
author | jaseg <git@jaseg.net> | 2018-05-05 11:56:54 +0200 |
---|---|---|
committer | jaseg <git@jaseg.net> | 2018-05-05 11:56:54 +0200 |
commit | c95e8296bf08b39f1198a8075b651e0da24934d1 (patch) | |
tree | 18d6f1ced22f81c95bdb1fd48db95fdd060164bd /firmware/spectrum_progress.py | |
parent | c641bbc090721d11dae9f4cf412320a37b1170ec (diff) | |
download | olsndot-c95e8296bf08b39f1198a8075b651e0da24934d1.tar.gz olsndot-c95e8296bf08b39f1198a8075b651e0da24934d1.tar.bz2 olsndot-c95e8296bf08b39f1198a8075b651e0da24934d1.zip |
Add spectrum measurement stuff
Diffstat (limited to 'firmware/spectrum_progress.py')
-rw-r--r-- | firmware/spectrum_progress.py | 32 |
1 files changed, 32 insertions, 0 deletions
diff --git a/firmware/spectrum_progress.py b/firmware/spectrum_progress.py new file mode 100644 index 0000000..a4eb4b2 --- /dev/null +++ b/firmware/spectrum_progress.py @@ -0,0 +1,32 @@ +#!/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)+1 + + 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) + for step in step_gen(): + bar.update(step - bar.n) |