[python] sqlite och classor och definitioner

#!/usr/bin/env python

""" sql_conn.py - a simple wrapper for python sqlite3 """

import sqlite3

class SQLiteConnection():
    def __init__(self, database_name):
        self.conn = sqlite3.connect(database_name)
        self.conn.row_factory = sqlite3.Row
        self.cc = self.conn.cursor()
    
    # Returns a list of dicts
    def _to_dict(self, rows, cursor):
        """ Internal function for creating a nice list of dicts """
        ls = []
        for row in rows:
            di = {}
            for n in range(len(row)):
                di[cursor[n][0]] = row[n]
            ls.append(di)
        return ls
    
    def query(self, sql, values=[]):
        """ Performs a query on the db. Don't forget to commit! """
        try:
            self.cc.execute(sql, values)
            if sql.startswith('SELECT'):
                return self._to_dict(self.cc.fetchall(), self.cc.description)
            return None
        except:
            print("Could not complete query... No idea why.")
            return None
    
    def get_last_insert(self):
        """ Returns the last thing inserted into the db """
        result = self.cc.execute(
            'SELECT * FROM test WHERE x==(?)', (self.cc.lastrowid,))
        return self._to_dict(self.cc.fetchall(), self.cc.description)[0]
    
    def commit(self):
        """ Commit! """
        self.conn.commit()
    
    def close(self):
        """ Closes the db connection """
        self.cc.close()

###############################################################################
# Usage example                                                               #
###############################################################################

# Create connection with db. If no db is found one will be created
db = SQLiteConnection("databas.sqlite")

# Create a table in the db
db.query("CREATE TABLE test(name TEXT, age INT)")

# Insert some stuff
db.query("INSERT INTO test(name, age) VALUES (?,?)", ("Dagon", 1349))
db.query("INSERT INTO test(name, age) VALUES (?,?)", ("Hydra", 1337))
db.query("INSERT INTO test(name, age) VALUES (?,?)", ("Satan", 666))
db.commit()

# Get the stuff back out again
creatures = db.query("SELECT * FROM test")
for creature in creatures:
    print("Name:%s Age:%d" % (creature["name"], creature["age"]))

class foo:
    bar = "foobar"
    def foo(self):
        return self.bar

def addToSqlString(self, q):
    """ This function just adds to the sqlstring the value of q """
    self.sqlString = self.sqlString + " " + q + ", "