From ee6233326edbe4fe3d3eece6f25ca4e4f5788d22 Mon Sep 17 00:00:00 2001
From: David Sommerseth <davids@redhat.com>
Date: Thu, 30 Jul 2009 17:06:59 +0200
Subject: database: Added SELECT() and GetValue() methods

---
 server/database.py | 70 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 70 insertions(+)

(limited to 'server/database.py')

diff --git a/server/database.py b/server/database.py
index 8413b4f..fbdad9d 100644
--- a/server/database.py
+++ b/server/database.py
@@ -115,8 +115,37 @@ class Database(object):
             else:
                 results.append(True)
 
+        curs.close()
         return results
 
+    def SELECT(self, table, fields, joins=None, where=None):
+        curs = self.conn.cursor()
+
+        # Query
+        sql = "SELECT %s FROM %s %s %s" % (
+            ",".join(fields),
+            table,
+            joins and "%s" % joins or "",
+            where and "WHERE %s" % " AND ".join(["%s = %%(%s)s" % (k,k) for (k,v) in where.items()] or "")
+            )
+        curs.execute(sql, where)
+
+        # Extract field names
+        fields = []
+        for fn in curs.description:
+            fields.append(fn[0])
+
+        # Extract records
+        records = []
+        for dbrec in curs.fetchall():
+            values = []
+            for val in dbrec:
+                values.append(val)
+            records.append(values)
+
+        curs.close()
+        return {"table": table, "fields": fields, "records": records}
+
     def COMMIT(self):
         # Commit the work
         self.conn.commit()
@@ -124,3 +153,44 @@ class Database(object):
     def ROLLBACK(self):
         # Abort / rollback the current work
         self.conn.rollback()
+
+
+    def GetValue(self, dbres, recidx, field):
+        "Helper function to easy extract a field from a record set"
+
+        # Check that input data good
+        if type(dbres) is not types.DictType:
+            raise AttributeError,'Database result parameter is not a Python dict'
+
+        try:
+            dbres['table']
+            dbres['fields']
+            dbres['records']
+        except KeyError, err:
+            raise KeyError, "Database result parameter do not contain a required element: %s", str(err)
+
+        if type(dbres['fields']) is not types.ListType:
+            raise AttributeError,"The 'fields' element is not a list of fields"
+
+        if type(dbres['records']) is not types.ListType:
+            raise AttributeError,"The 'records' element is not a list of fields"
+
+        # Return None when we're going out of boundaries
+        if recidx >= len(dbres['records']):
+            return None
+
+        if type(field) == types.StringType:
+            # Find the field index of the field name in the records set
+            try:
+                fidx = dbres['fields'].index(field)
+            except ValueError:
+                raise Exception, "Field '%s' is not found in the database result" % field
+        elif type(field) == types.IntType:
+            # If the field value is integer, assume it is the numeric field id
+            if field >= len(dbres['fields']):
+                raise Exception, "Field id '%i' is too high.  No field available" % field
+            fidx = field
+
+        # Return the value
+        return dbres['records'][recidx][fidx]
+
-- 
cgit