8ee0b319ee
These are for use by jenkins for gating jobs based off of devstack. Change-Id: I6688eeae905b3cedd6585a2b8ae88d6e92d38c36
58 lines
2.0 KiB
Python
58 lines
2.0 KiB
Python
import sqlite3
|
|
import os
|
|
import time
|
|
|
|
class VMDatabase(object):
|
|
def __init__(self, path=os.path.expanduser("~/vm.db")):
|
|
if not os.path.exists(path):
|
|
conn = sqlite3.connect(path)
|
|
c = conn.cursor()
|
|
c.execute('''create table machines
|
|
(id int, name text, ip text, change_number, patch_number, build_number, created int, user text)''')
|
|
conn.commit()
|
|
c.close()
|
|
self.conn = sqlite3.connect(path)
|
|
|
|
def addMachine(self, mid, name, ip, change, patch, build):
|
|
c = self.conn.cursor()
|
|
c.execute("insert into machines (id, name, ip, change_number, patch_number, build_number, created) values (?, ?, ?, ?, ?, ?, ?)",
|
|
(mid, name, ip, change, patch, build, int(time.time())))
|
|
self.conn.commit()
|
|
c.close()
|
|
|
|
def delMachine(self, mid):
|
|
c = self.conn.cursor()
|
|
c.execute("delete from machines where id=?", (mid,))
|
|
self.conn.commit()
|
|
c.close()
|
|
|
|
def setMachineUser(self, mid, user):
|
|
c = self.conn.cursor()
|
|
c.execute("update machines set user=? where id=?", (user, mid))
|
|
self.conn.commit()
|
|
c.close()
|
|
|
|
def getMachines(self):
|
|
c = self.conn.cursor()
|
|
c.execute("select * from machines")
|
|
names = [col[0] for col in c.description]
|
|
data = [dict(zip(names, row)) for row in c]
|
|
c.close()
|
|
return data
|
|
|
|
def getMachine(self, change, patch, build):
|
|
c = self.conn.cursor()
|
|
c.execute("select * from machines where change_number=? and patch_number=? and build_number=?", (change, patch, build))
|
|
names = [col[0] for col in c.description]
|
|
data = [row for row in c]
|
|
c.close()
|
|
return dict(zip(names, data[0]))
|
|
|
|
if __name__=='__main__':
|
|
db = VMDatabase()
|
|
db.addMachine(1, 'foo', '1.2.3.4', 88, 2, 1)
|
|
db.setMachineUser(1, 'jeblair')
|
|
print db.getMachines()
|
|
print db.getMachine(88,2,1)
|
|
db.delMachine(1)
|