diff options
Diffstat (limited to 'kpov_judge/test_prepare_disks.py')
-rwxr-xr-x | kpov_judge/test_prepare_disks.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/kpov_judge/test_prepare_disks.py b/kpov_judge/test_prepare_disks.py index 7cfc5b5..bdb3f89 100755 --- a/kpov_judge/test_prepare_disks.py +++ b/kpov_judge/test_prepare_disks.py @@ -146,6 +146,7 @@ if __name__ == '__main__': print("Usage: " + sys.argv[0]) print("Run prepare_disks on running computers over ssh") print("The task name and params are read from ~/.kpov_params.yaml") + yaml_config_file = os.path.expanduser("~/.kpov_params.yaml") with open(yaml_config_file) as f: params = yaml.load(f) @@ -157,12 +158,15 @@ if __name__ == '__main__': http_auth(task_url, params['username'], params['password']) req = urllib.request.Request("{task_url}/{task_name}/task.py".format(**params)) source = urllib.request.urlopen(req).read() + if not source: + raise Exception('no such task: {}'.format(task_name)) d = {} exec(compile(source, 'task.py', 'exec'), globals(), d) computers, prepare_disks = d['computers'], d['prepare_disks'] except Exception as e: print(e) - exit(1) + exit(1) + templates = dict() sshguestfs_params = params.get('sshguestfs_params', dict()) task_sshguestfs_params = sshguestfs_params.get(task_name, dict()) |