From 86aaeead3019a5a2713e772073f4b94d84daa491 Mon Sep 17 00:00:00 2001 From: Newnius Date: Mon, 6 Jul 2020 10:54:31 +0800 Subject: [PATCH] update, robust --- agent.py | 22 +++++++++++----------- 1 file changed, 11 insertions(+), 11 deletions(-) diff --git a/agent.py b/agent.py index be78425..479ea63 100644 --- a/agent.py +++ b/agent.py @@ -275,19 +275,19 @@ class MyHandler(BaseHTTPRequestHandler): }) docker_image = form.getvalue('image') docker_name = form.getvalue('name') - docker_cmd = form.getvalue('cmd') + docker_cmd = form.getvalue('cmd', '') docker_workspace = form.getvalue('workspace') - docker_gpus = form.getvalue('gpus') - docker_mem_limit = form.getvalue('mem_limit') - docker_cpu_limit = form.getvalue('cpu_limit') + docker_gpus = form.getvalue('gpus', '') + docker_mem_limit = form.getvalue('mem_limit', 4096) + docker_cpu_limit = form.getvalue('cpu_limit', 4) docker_network = form.getvalue('network') - docker_wait = form.getvalue('should_wait') - docker_output = form.getvalue('output_dir') - docker_hdfs_address = form.getvalue('hdfs_address') - docker_hdfs_dir = form.getvalue('hdfs_dir') - docker_gpu_mem = form.getvalue('gpu_mem') - dfs_src = form.getvalue('dfs_src') - dfs_dst = form.getvalue('dfs_dst') + docker_wait = form.getvalue('should_wait', '0') + docker_output = form.getvalue('output_dir', '') + docker_hdfs_address = form.getvalue('hdfs_address', '') + docker_hdfs_dir = form.getvalue('hdfs_dir', '') + docker_gpu_mem = form.getvalue('gpu_mem', 8192) + dfs_src = form.getvalue('dfs_src', '') + dfs_dst = form.getvalue('dfs_dst', '') token = generate_token(16) if len(dfs_src) > 0: