YAO/YAO-agent
1
0
Fork 0
mirror of https://github.com/newnius/YAO-agent.git synced 2025-12-15 13:56:44 +00:00
Code Issues Releases Wiki Activity

all in one

Browse Source
This commit is contained in:
Newnius
2019-12-26 18:47:25 +08:00
parent a0ff4274d0
commit 702fc8dcfb
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
main.py
View File

@@ -131,7 +131,7 @@ class MyHandler(BaseHTTPRequestHandler):
try: try:
script = " ".join([ script = " ".join([
"docker run", "docker run",
"--gpus '\'\"device=" + docker_gpus + "\"$'\''", "--gpus '\"device=" + docker_gpus + "\"'",
"--detach=True", "--detach=True",
"--hostname " + docker_name, "--hostname " + docker_name,
"--network " + docker_network, "--network " + docker_network,
@@ -143,11 +143,9 @@ class MyHandler(BaseHTTPRequestHandler):
docker_cmd docker_cmd
]) ])
print('sh -c \'' + script + '\'')
client = docker.from_env() client = docker.from_env()
container = client.containers.get('yao-agent-helper') container = client.containers.get('yao-agent-helper')
exit_code, output = container.exec_run('sh -c \'' + script + '\'') exit_code, output = container.exec_run('sh', '-c', script)
msg = {"code": 0, "id": output.decode('utf-8').rstrip('\n')} msg = {"code": 0, "id": output.decode('utf-8').rstrip('\n')}
print(msg) print(msg)