1 files changed, 21 insertions, 1 deletions
diff --git a/python/common.py b/python/common.py
index a194b09b..f0cc3fdf 100644
--- a/python/common.py
+++ b/python/common.py
@@ -10,4 +10,24 @@ conn = psycopg2.connect(database="ET",
                         user="et_user",
                         password="Serenitas1",
                         host="debian")
-cursor = conn.cursor()
+
+def query_db(sqlstr, **kwargs):
+    c = conn.cursor()
+    params = kwargs.get('params', None)
+    one = kwargs.get('one', True)
+    try:
+        if params:
+            c.execute(sqlstr, params)
+        else:
+            c.execute(sqlstr)
+    except psycopg2.Error:
+        c.close()
+        conn.rollback()
+        return None
+
+    if one:
+        result = c.fetchone()
+    else:
+        result = c.fetchall()
+    c.close()
+    return result