diff options
-rwxr-xr-x | kpov_judge/create_disk_images.py | 4 |
1 files changed, 3 insertions, 1 deletions
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}, |