mirror of
https://github.com/newnius/YAO-agent.git
synced 2025-06-06 05:21:55 +00:00
109 lines
2.7 KiB
Python
109 lines
2.7 KiB
Python
#!/usr/bin/python
|
|
from http.server import BaseHTTPRequestHandler, HTTPServer
|
|
import cgi
|
|
import docker
|
|
import json
|
|
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):
|
|
req = parse.urlparse(self.path)
|
|
query = parse.parse_qs(req.query)
|
|
|
|
if req.path == "/ping":
|
|
# Open the static file requested and send it
|
|
self.send_response(200)
|
|
self.send_header('Content-type', 'application/json')
|
|
self.end_headers()
|
|
self.wfile.write(bytes("pong", "utf-8"))
|
|
|
|
elif req.path == "/logs":
|
|
id = query['id'][0]
|
|
client = docker.from_env()
|
|
container = client.containers.get(id)
|
|
|
|
msg = {'code': 0, 'logs': container.logs().decode()}
|
|
self.send_response(200)
|
|
self.send_header('Content-type', 'application/json')
|
|
self.end_headers()
|
|
self.wfile.write(bytes(json.dumps(msg), "utf-8"))
|
|
|
|
else:
|
|
self.send_error(404, 'File Not Found: %s' % self.path)
|
|
|
|
# Handler for the POST requests
|
|
def do_POST(self):
|
|
if self.path == "/create":
|
|
form = cgi.FieldStorage(
|
|
fp=self.rfile,
|
|
headers=self.headers,
|
|
environ={
|
|
'REQUEST_METHOD': 'POST',
|
|
'CONTENT_TYPE': self.headers['Content-Type'],
|
|
})
|
|
docker_image = form["image"].value
|
|
docker_cmd = form["cmd"].value
|
|
|
|
print(docker_image)
|
|
print(docker_cmd)
|
|
|
|
client = docker.from_env()
|
|
container = client.containers.run(
|
|
image=docker_image,
|
|
command=docker_cmd,
|
|
environment={"key": "value"},
|
|
runtime="nvidia",
|
|
detach=True
|
|
)
|
|
|
|
msg = {"code": 0, "id": container.id}
|
|
|
|
self.send_response(200)
|
|
self.send_header('Content-type', 'application/json')
|
|
self.end_headers()
|
|
self.wfile.write(bytes(json.dumps(msg), "utf-8"))
|
|
|
|
elif self.path == "/stop":
|
|
form = cgi.FieldStorage(
|
|
fp=self.rfile,
|
|
headers=self.headers,
|
|
environ={
|
|
'REQUEST_METHOD': 'POST',
|
|
'CONTENT_TYPE': self.headers['Content-Type'],
|
|
})
|
|
id = form["id"].value
|
|
|
|
client = docker.from_env()
|
|
container = client.containers.get(id)
|
|
container.stop()
|
|
msg = {"code": 0}
|
|
|
|
self.send_response(200)
|
|
self.send_header('Content-type', 'application/json')
|
|
self.end_headers()
|
|
self.wfile.write(bytes(json.dumps(msg), "utf-8"))
|
|
else:
|
|
self.send_error(404, 'File Not Found: %s' % self.path)
|
|
|
|
|
|
try:
|
|
# Create a web server and define the handler to manage the
|
|
# incoming request
|
|
server = HTTPServer(('', PORT_NUMBER), MyHandler)
|
|
print('Started httpserver on port ', PORT_NUMBER)
|
|
|
|
# Wait forever for incoming htto requests
|
|
server.serve_forever()
|
|
|
|
except KeyboardInterrupt:
|
|
print('^C received, shutting down the web server')
|
|
|
|
|
|
server.socket.close()
|