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

bugfix

Browse Source
This commit is contained in:
Newnius 2019-03-20 10:48:28 +08:00
parent 54f20c0ece
commit 5c90e5a94f
1 changed files with 55 additions and 57 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
server.py
View File

@ -24,25 +24,24 @@ class MyHandler(BaseHTTPRequestHandler):
self.wfile.write(bytes("pong", "utf-8")) self.wfile.write(bytes("pong", "utf-8"))
elif req.path == "/logs": elif req.path == "/logs":
id = query['id'][0]
try: try:
container_id = query['id'][0]
client = docker.from_env() client = docker.from_env()
container = client.containers.get(id) container = client.containers.get(container_id)
msg = {'code': 0, 'logs': str(container.logs().decode())} msg = {'code': 0, 'logs': str(container.logs().decode())}
except Exception as e: except Exception as e:
msg = {'code': 0, 'error': e} msg = {'code': 0, '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')
self.end_headers() self.end_headers()
self.wfile.write(bytes(json.dumps(msg), "utf-8")) self.wfile.write(bytes(json.dumps(msg), "utf-8"))
elif req.path == "/status": elif req.path == "/status":
id = query['id'][0] container_id = query['id'][0]
client = docker.from_env() client = docker.from_env()
container = client.containers.list(all=True, filters={'id': id}) container = client.containers.list(all=True, filters={'id': container_id})
if len(container) > 0: if len(container) > 0:
container = container[0] container = container[0]
print(container.image.attrs)
status = { status = {
'id': container.short_id, 'id': container.short_id,
'image': container.attrs['Config']['Image'], 'image': container.attrs['Config']['Image'],
@ -65,9 +64,8 @@ class MyHandler(BaseHTTPRequestHandler):
else: else:
self.send_error(404, 'File Not Found: %s' % self.path) self.send_error(404, 'File Not Found: %s' % self.path)
# Handler for the POST requests
# Handler for the POST requests def do_POST(self):
def do_POST(self):
if self.path == "/create": if self.path == "/create":
form = cgi.FieldStorage( form = cgi.FieldStorage(
fp=self.rfile, fp=self.rfile,
@ -90,7 +88,7 @@ def do_POST(self):
) )
msg = {"code": 0, "id": container.id} msg = {"code": 0, "id": container.id}
except Exception as e: except Exception as e:
msg = {"code": 1, "error": 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')
@ -105,10 +103,10 @@ def do_POST(self):
'REQUEST_METHOD': 'POST', 'REQUEST_METHOD': 'POST',
'CONTENT_TYPE': self.headers['Content-Type'], 'CONTENT_TYPE': self.headers['Content-Type'],
}) })
id = form["id"].value container_id = form["id"].value
client = docker.from_env() client = docker.from_env()
container = client.containers.get(id) container = client.containers.get(container_id)
container.stop() container.stop()
msg = {"code": 0} msg = {"code": 0}
@ -124,9 +122,9 @@ try:
# Create a web server and define the handler to manage the # Create a web server and define the handler to manage the
# incoming request # incoming request
server = HTTPServer(('', PORT_NUMBER), MyHandler) server = HTTPServer(('', PORT_NUMBER), MyHandler)
print('Started httpserver on port ', PORT_NUMBER) print('Started http server on port ', PORT_NUMBER)
# Wait forever for incoming htto requests # Wait forever for incoming http requests
server.serve_forever() server.serve_forever()
except KeyboardInterrupt: except KeyboardInterrupt: