GPU status endpoints

gpu_cluster/controllers/container_controller.py

@@ -37,7 +37,6 @@ def launch_container(self, c_id):
     '''
     STATUS QUERIERS
     '''
-
     def verify_launch(self, c_id):
         instance = db_session.query(Instance).filter_by(cid = c_id).first() 
         return True if instance.launched == True else False

gpu_cluster/routes/cluster_api.py

@@ -4,6 +4,8 @@
 from ..database import db_session
 from ..models import Instance
 from flask import Flask, jsonify, request, abort 
+from nvdocker import NVDockerClient
+import socket
 
 class ClusterAPI():
     def __init__(self, controller):
@@ -29,11 +31,20 @@ def confirm_launch(self):
 
         if launched == False:
             return jsonify({"error" : "non-existant container"})
-        return jsonify({"verified" : "confirmed"})    
-
+        return jsonify({"verified" : "confirmed"})
+    
     def kill_container(self):
         pass
 
+    def status(self):
+        hostname = socket.gethostname()
+        available_gpu = NVDockerClient.least_used_gpu()
+        response  = {
+            "hostname" : hostname,
+            "gpu" : available_gpu
+        }
+        return jsonify(response)
+
     def register_routes(self, app):
         app.add_url_rule('/create_container', 'create_container', self.create_container, methods=['POST'])
         app.add_url_rule('/confirm',          'confirm',          self.confirm_launch,   methods=['POST'])