basic capsul detail page

capsulflask/db_model.py

@@ -89,11 +89,52 @@ class DBModel:
       self.cursor.fetchall()
     ))
 
-  def create_vm(self, email, id, size, os):
+  def updateVm(self, email, id, ipv4):
+    self.cursor.execute("UPDATE vms SET last_seen_ipv4 = %s WHERE email = %s AND id = %s", (ipv4, email, id))
+    self.connection.commit()
+
+  def create_vm(self, email, id, size, os, ssh_public_keys):
     self.cursor.execute(""" 
       INSERT INTO vms (email, id, size, os)
       VALUES  (%s, %s, %s, %s)
       """, 
       (email, id, size, os)
     )
-    self.connection.commit()
+    
+    for ssh_public_key in ssh_public_keys:
+      self.cursor.execute(""" 
+        INSERT INTO vm_ssh_public_key (email, vm_id, ssh_public_key_name)
+        VALUES  (%s, %s, %s)
+        """, 
+        (email, id, ssh_public_key)
+      )
+    self.connection.commit()
+
+  def get_vm_detail(self, email, id):
+    self.cursor.execute(""" 
+      SELECT vms.id, vms.last_seen_ipv4, vms.last_seen_ipv6, os_images.description, vms.created, vms.deleted,
+            vm_sizes.id, vm_sizes.dollars_per_month, vm_sizes.vcpus, vm_sizes.memory_mb, vm_sizes.bandwidth_gb_per_month
+      FROM vms 
+        JOIN os_images on vms.os = os_images.id
+        JOIN vm_sizes on vms.size = vm_sizes.id
+      WHERE vms.email = %s AND vms.id = %s""", 
+      (email, id)
+    )
+    rows = self.cursor.fetchall()
+    if len(rows) == 0:
+      return None
+
+    x = rows[0]
+    vm = dict(
+      id=x[0], ipv4=x[1], ipv6=x[2], os_description=x[3], created=x[4], deleted=x[5],
+      size=x[6], dollars_per_month=x[7], vcpus=x[8], memory_mb=x[9], bandwidth_gb_per_month=x[10]
+    )
+
+    self.cursor.execute(""" 
+      SELECT ssh_public_key_name FROM vm_ssh_public_key 
+      WHERE vm_ssh_public_key.email = %s AND vm_ssh_public_key.vm_id = %s""", 
+      (email, id)
+    )
+    vm["ssh_public_keys"] = list(map( lambda x: x[0], self.cursor.fetchall() ))
+
+    return vm