More DB tweaks
Add the creation_time timestamp when creating VLANs
Minor syntax fixups
Change-Id: I488d14f435e6f97431659cdb52d081725c6999c9
diff --git a/db/db.py b/db/db.py
index 1b8da3f..fe3b893 100644
--- a/db/db.py
+++ b/db/db.py
@@ -20,6 +20,7 @@
import psycopg2
import psycopg2.extras
+import datetime
from errors import CriticalError
class VlanDB:
@@ -48,9 +49,7 @@
def create_port(self, name, switch_id, is_locked, is_trunk, current_vlan_id, base_vlan_id):
try:
- sql = "INSERT INTO port (name, switch_id, is_locked, "
- + "is_trunk, current_vlan_id, base_vlan_id) "
- + "VALUES (%s, %s, %s, %s, %s, %s) RETURNING port_id"
+ sql = "INSERT INTO port (name, switch_id, is_locked, is_trunk, current_vlan_id, base_vlan_id) VALUES (%s, %s, %s, %s, %s, %s) RETURNING port_id"
data = (name, switch_id, is_locked,
is_trunk, current_vlan_id, base_vlan_id)
self.cursor.execute(sql, data)
@@ -63,9 +62,9 @@
def create_vlan(self, name, tag, is_base_vlan):
try:
- sql = "INSERT INTO vlan (name, tag, is_base_vlan) "
- + "VALUES (%s, %s, %s) RETURNING vlan_id"
- data = (name, tag, is_base_vlan)
+ dt = datetime.datetime.now()
+ sql = "INSERT INTO vlan (name, tag, is_base_vlan, creation_time) VALUES (%s, %s, %s, %s) RETURNING vlan_id"
+ data = (name, tag, is_base_vlan, dt)
self.cursor.execute(sql, data)
vlan_id = self.cursor.fetchone()[0]
self.connection.commit()