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

robust

Browse Source
This commit is contained in:
Newnius 2019-06-14 15:19:56 +08:00
parent 694dcf2b08
commit 34cbc27b38
1 changed files with 15 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
executor.py
View File

@ -8,7 +8,6 @@ from urllib import parse
PORT_NUMBER = 8000
# This class will handles any incoming request from the browser
class MyHandler(BaseHTTPRequestHandler):
# Handler for the GET requests
def do_GET(self):
@ -112,7 +111,7 @@ class MyHandler(BaseHTTPRequestHandler):
runtime='nvidia'
)
client.start(container)
msg = {"code": 0, "id": container.id}
msg = {"code": 0, "id": container['Id']}
except Exception as e:
msg = {"code": 1, "error": str(e)}
@ -131,10 +130,13 @@ class MyHandler(BaseHTTPRequestHandler):
})
container_id = form.getvalue('id')
client = docker.from_env()
container = client.containers.get(container_id)
container.stop()
msg = {"code": 0}
try:
client = docker.from_env()
container = client.containers.get(container_id)
container.stop()
msg = {"code": 0, "error": "Success"}
except Exception as e:
msg = {"code": 1, "error": str(e)}
self.send_response(200)
self.send_header('Content-type', 'application/json')
@ -151,10 +153,13 @@ class MyHandler(BaseHTTPRequestHandler):
})
container_id = form.getvalue('id')
client = docker.from_env()
container = client.containers.get(container_id)
container.remove(force=True)
msg = {"code": 0}
try:
client = docker.from_env()
container = client.containers.get(container_id)
container.remove(force=True)
msg = {"code": 0, "error": "Success"}
except Exception as e:
msg = {"code": 1, "error": str(e)}
self.send_response(200)
self.send_header('Content-type', 'application/json')