jbdod/notes-ker-lann
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[WEI] Display status bar with tqdm

Browse source 
Signed-off-by: Yohann D'ANELLO <ynerant@crans.org>
This commit is contained in:
Yohann D'ANELLO 2021-09-16 20:46:34 +02:00
parent e89383e3f4
commit 1ef25924a0
No known key found for this signature in database
GPG key ID: 3A75C55819C8CF85
2 changed files with 39 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

16
apps/wei/management/commands/wei_algorithm.py
View file

@ -24,7 +24,14 @@ class Command(BaseCommand):
sid = transaction.savepoint()
algorithm = CurrentSurvey.get_algorithm_class()()
algorithm.run_algorithm()
try:
from tqdm import tqdm
display_tqdm = True
except ImportError:
display_tqdm = False
algorithm.run_algorithm(display_tqdm=display_tqdm)
output = options['output']
registrations = algorithm.get_registrations()
@ -34,8 +41,13 @@ class Command(BaseCommand):
for bus, members in per_bus.items():
output.write(bus.name + "\n")
output.write("=" * len(bus.name) + "\n")
order = -1
for r in members:
output.write(r.user.username + "\n")
survey = CurrentSurvey(r)
for order, (b, _score) in enumerate(survey.ordered_buses()):
if b == bus:
break
output.write(f"{r.user.username} ({order + 1})\n")
output.write("\n")
if not options['doit']: