update

server.py

@@ -0,0 +1,108 @@
+#!/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()