YAO/YAO-agent
1
0
Fork 0
mirror of https://github.com/newnius/YAO-agent.git synced 2025-06-07 13:51:56 +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 PORT_NUMBER = 8000
# This class will handles any incoming request from the browser
class MyHandler(BaseHTTPRequestHandler): class MyHandler(BaseHTTPRequestHandler):
# Handler for the GET requests # Handler for the GET requests
def do_GET(self): def do_GET(self):
@ -112,7 +111,7 @@ class MyHandler(BaseHTTPRequestHandler):
runtime='nvidia' runtime='nvidia'
) )
client.start(container) client.start(container)
msg = {"code": 0, "id": container.id} msg = {"code": 0, "id": container['Id']}
except Exception as e: except Exception as e:
msg = {"code": 1, "error": str(e)} msg = {"code": 1, "error": str(e)}
@ -131,10 +130,13 @@ class MyHandler(BaseHTTPRequestHandler):
}) })
container_id = form.getvalue('id') container_id = form.getvalue('id')
client = docker.from_env() try:
container = client.containers.get(container_id) client = docker.from_env()
container.stop() container = client.containers.get(container_id)
msg = {"code": 0} container.stop()
msg = {"code": 0, "error": "Success"}
except Exception as e:
msg = {"code": 1, "error": str(e)}
self.send_response(200) self.send_response(200)
self.send_header('Content-type', 'application/json') self.send_header('Content-type', 'application/json')
@ -151,10 +153,13 @@ class MyHandler(BaseHTTPRequestHandler):
}) })
container_id = form.getvalue('id') container_id = form.getvalue('id')
client = docker.from_env() try:
container = client.containers.get(container_id) client = docker.from_env()
container.remove(force=True) container = client.containers.get(container_id)
msg = {"code": 0} 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_response(200)
self.send_header('Content-type', 'application/json') self.send_header('Content-type', 'application/json')