updating virt_model with email and

2020-05-10 20:23:00 -05:00
parent 838348a0fb
commit d706ae7761
5 changed files with 93 additions and 29 deletions
--- a/capsulflask/db_model.py
+++ b/capsulflask/db_model.py
@ -30,4 +30,9 @@ class DBModel:
      self.cursor.execute("DELETE FROM login_tokens WHERE email = %s", (email, ))
      self.connection.commit()
      return email
-    return None
+    return None
+
+  def allVmIds(self,):
+    self.cursor.execute("SELECT id FROM vms")
+    return map(lambda x: x[0], self.cursor.fetchall()) 
+