YAO/YAO-agent
1
0
Fork 0
mirror of https://github.com/newnius/YAO-agent.git synced 2025-06-06 05:21:55 +00:00
Code Issues Releases Wiki Activity

update, add dfs

Browse Source
This commit is contained in:
Newnius 2020-06-02 21:54:19 +08:00
parent 22ee845aad
commit 48fab2bc90
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
agent.py
View File

@ -185,8 +185,16 @@ class MyHandler(BaseHTTPRequestHandler):
docker_hdfs_address = form.getvalue('hdfs_address')
docker_hdfs_dir = form.getvalue('hdfs_dir')
docker_gpu_mem = form.getvalue('gpu_mem')
docker_dfs_src = form.getvalue('dfs_src')
docker_dfs_dst = form.getvalue('dfs_dst')
token = generate_token(16)
try:
os.mkdir(docker_dfs_src)
except OSError as e:
print("Creation of the directory %s failed" % docker_dfs_src)
print(e)
try:
script = " ".join([
"docker run",
@ -204,6 +212,7 @@ class MyHandler(BaseHTTPRequestHandler):
"--env hdfs_address=" + docker_hdfs_address,
"--env hdfs_dir=" + docker_hdfs_dir,
"--env gpu_mem=" + docker_gpu_mem,
"--mount type=bind,src=" + docker_dfs_src + ",dst=" + docker_dfs_dst,
docker_image,
docker_cmd
])