From dd9eaef3b467ffe1693ceeee4a0398d191945faf Mon Sep 17 00:00:00 2001 From: Timotej Lazar Date: Tue, 12 Feb 2019 18:42:14 +0100 Subject: create_disk_images: add diskless computers to db.student_computers Using defaultdict for all_disks resulted in diskless computers missing from DB. By iterating over keys in computers instead of all_disks computers with no disk are also added (and later shown on task page). --- kpov_judge/create_disk_images.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/kpov_judge/create_disk_images.py b/kpov_judge/create_disk_images.py index 5e630be..ce79f85 100755 --- a/kpov_judge/create_disk_images.py +++ b/kpov_judge/create_disk_images.py @@ -178,7 +178,9 @@ if __name__ == '__main__': g.close() lock_fp.write("saving URLs\n") - for comp_name, disks in all_disks.items(): + for computer in computers: + comp_name = computer['name'] + disks = all_disks[comp_name] lock_fp.write('urls: '+ str(disks) + '\n') l = db.student_computers.update({ 'disk_urls': {'$exists': False}, -- cgit v1.2.1