~abentley/mailtyrant/trunk

Tweak dbcache and its tests.

added added

removed removed

def __init__(self, connection):

self.connection = connection

self.reset_cache()

@classmethod

def from_path(klass, path):

connection = sqlite3.connect(path,

detect_types=sqlite3.PARSE_DECLTYPES|sqlite3.PARSE_COLNAMES)

return klass(connection)

def reset_cache(self):

c = self.connection.cursor()

try:

c.execute('''drop table mailcache''')

except sqlite3.OperationalError:

pass

except sqlite3.OperationalError, e:

if e[0] != 'no such table: mailcache':

raise

c.execute('''create table mailcache (key text, date timestamp,

'from' text, subject text)''')

c.execute('select key, date as "date [timestamp]", "from", subject'

' from mailcache where key in (?)', keys)

result = c.fetchall()

return result

return [(x[0], x[1:]) for x in result]

def set(self, key, values):

c = self.connection.cursor()

c.execute('insert into mailcache values (?, ?, ?, ?)',

(key,) + values)

def save(self):

self.connection.commit()

def get_cache():

return Cache(sqlite3.connect('cache.sqlite', detect_types=sqlite3.PARSE_DECLTYPES|sqlite3.PARSE_COLNAMES))

return Cache.from_path('cache.sqlite')