update

2025-12-15 13:56:44 +00:00 · 2019-04-13 19:39:29 +08:00
parent 21b1d2edd7
commit d2889614ee
6 changed files with 200 additions and 192 deletions
--- a/executor.py
+++ b/executor.py
@@ -1,52 +1,137 @@
+#!/usr/bin/python
+from http.server import BaseHTTPRequestHandler, HTTPServer
+import cgi
 import docker
+import json
+from urllib import parse
+
+PORT_NUMBER = 8000


-def run():
-	client = docker.from_env()
-	try:
-		print(client.containers.run(image="alpine", command="nvid", environment={"KEY": "value"}))
-		# print(client.containers.run(image="nvidia/cuda:9.0-base", command="nvidia-smi", environment={"KEY": "value"}, runtime="nvidia"))
-	except Exception as e:
-		print(e.__class__.__name__, e)
+# 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":
+			try:
+				container_id = query['id'][0]
+				client = docker.from_env()
+				container = client.containers.get(container_id)
+				msg = {'code': 0, 'logs': str(container.logs().decode())}
+			except Exception as e:
+				msg = {'code': 0, 'error': str(e)}
+			self.send_response(200)
+			self.send_header('Content-type', 'application/json')
+			self.end_headers()
+			self.wfile.write(bytes(json.dumps(msg), "utf-8"))
+
+		elif req.path == "/status":
+			container_id = query['id'][0]
+			client = docker.from_env()
+			container = client.containers.list(all=True, filters={'id': container_id})
+			if len(container) > 0:
+				container = container[0]
+				status = {
+					'id': container.short_id,
+					'image': container.attrs['Config']['Image'],
+					'image_digest': container.attrs['Image'],
+					'command': container.attrs['Config']['Cmd'],
+					'created_at': container.attrs['Created'],
+					'finished_at': container.attrs['State']['FinishedAt'],
+					'status': container.status
+				}
+				if status['command'] is not None:
+					status['command'] = ' '.join(container.attrs['Config']['Cmd'])
+				msg = {'code': 0, 'status': status}
+			else:
+				msg = {'code': 1, 'error': "container not exist"}
+			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_name = form["name"].value
+			docker_cmd = form["cmd"].value
+			docker_workspace = form["workspace"].value
+			docker_gpus = form["gpus"].value
+
+			try:
+				client = docker.from_env()
+				container = client.containers.run(
+					image=docker_image,
+					hostname=docker_name,
+					command=docker_cmd,
+					environment={"repo": docker_workspace, "NVIDIA_VISIBLE_DEVICES": docker_gpus},
+					runtime="nvidia",
+					detach=True
+				)
+				msg = {"code": 0, "id": container.id}
+			except Exception as e:
+				msg = {"code": 1, "error": str(e)}
+
+			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'],
+				})
+			container_id = form["id"].value
+
+			client = docker.from_env()
+			container = client.containers.get(container_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)


-def run_in_background():
-	client = docker.from_env()
-	container = client.containers.run("alpine", ["echo", "hello", "world"], detach=True)
-	print(container.id)
+try:
+	# Create a web server and define the handler to manage the
+	# incoming request
+	server = HTTPServer(('', PORT_NUMBER), MyHandler)
+	print('Started http server on port ', PORT_NUMBER)

+	# Wait forever for incoming http requests
+	server.serve_forever()

-def list_containers():
-	client = docker.from_env()
-	for container in client.containers.list():
-		print(container.id)
+except KeyboardInterrupt:
+	print('^C received, shutting down the web server')

-
-def get_logs(id):
-	try:
-		client = docker.from_env()
-		container = client.containers.get(id)
-		print(container.logs().decode())
-	except Exception as e:
-		print(e)
-
-
-def get_status(id):
-	client = docker.from_env()
-	container = client.containers.list(all=True, filters={'id': id})
-	status = {}
-	if len(container) > 0:
-		container= container[0]
-		status['id'] = container.short_id
-		status['image'] = container.attrs['Config']['Image']
-		status['image_digest'] = container.attrs['Image']
-		status['command'] = container.attrs['Config']['Cmd']
-		status['createdAt'] = container.attrs['Created']
-		status['finishedAt'] = container.attrs['State']['FinishedAt']
-		status['status'] = container.status
-		if status['command'] is not None:
-			status['command'] = ' '.join(container.attrs['Config']['Cmd'])
-	print(status)
-
-
-get_status('')
+server.socket.close()