Correction crash
[auf_roundup.git] / build / lib / roundup / backends / sessions_rdbms.py
1 #$Id: sessions_rdbms.py,v 1.8 2008-08-18 05:04:01 richard Exp $
2 """This module defines a very basic store that's used by the CGI interface
3 to store session and one-time-key information.
4
5 Yes, it's called "sessions" - because originally it only defined a session
6 class. It's now also used for One Time Key handling too.
7 """
8 __docformat__ = 'restructuredtext'
9
10 import os, time
11
12 class BasicDatabase:
13 ''' Provide a nice encapsulation of an RDBMS table.
14
15 Keys are id strings, values are automatically marshalled data.
16 '''
17 def __init__(self, db):
18 self.db = db
19 self.cursor = self.db.cursor
20
21 def clear(self):
22 self.cursor.execute('delete from %ss'%self.name)
23
24 def exists(self, infoid):
25 n = self.name
26 self.cursor.execute('select count(*) from %ss where %s_key=%s'%(n,
27 n, self.db.arg), (infoid,))
28 return int(self.cursor.fetchone()[0])
29
30 _marker = []
31 def get(self, infoid, value, default=_marker):
32 n = self.name
33 self.cursor.execute('select %s_value from %ss where %s_key=%s'%(n,
34 n, n, self.db.arg), (infoid,))
35 res = self.cursor.fetchone()
36 if not res:
37 if default != self._marker:
38 return default
39 raise KeyError('No such %s "%s"'%(self.name, infoid))
40 values = eval(res[0])
41 return values.get(value, None)
42
43 def getall(self, infoid):
44 n = self.name
45 self.cursor.execute('select %s_value from %ss where %s_key=%s'%(n,
46 n, n, self.db.arg), (infoid,))
47 res = self.cursor.fetchone()
48 if not res:
49 raise KeyError('No such %s "%s"'%(self.name, infoid))
50 return eval(res[0])
51
52 def set(self, infoid, **newvalues):
53 c = self.cursor
54 n = self.name
55 a = self.db.arg
56 c.execute('select %s_value from %ss where %s_key=%s'%(n, n, n, a),
57 (infoid,))
58 res = c.fetchone()
59 if res:
60 values = eval(res[0])
61 else:
62 values = {}
63 values.update(newvalues)
64
65 if res:
66 sql = 'update %ss set %s_value=%s where %s_key=%s'%(n, n,
67 a, n, a)
68 args = (repr(values), infoid)
69 else:
70 sql = 'insert into %ss (%s_key, %s_time, %s_value) '\
71 'values (%s, %s, %s)'%(n, n, n, n, a, a, a)
72 args = (infoid, time.time(), repr(values))
73 c.execute(sql, args)
74
75 def destroy(self, infoid):
76 self.cursor.execute('delete from %ss where %s_key=%s'%(self.name,
77 self.name, self.db.arg), (infoid,))
78
79 def updateTimestamp(self, infoid):
80 """ don't update every hit - once a minute should be OK """
81 now = time.time()
82 self.cursor.execute('''update %ss set %s_time=%s where %s_key=%s
83 and %s_time < %s'''%(self.name, self.name, self.db.arg,
84 self.name, self.db.arg, self.name, self.db.arg),
85 (now, infoid, now-60))
86
87 def clean(self):
88 ''' Remove session records that haven't been used for a week. '''
89 now = time.time()
90 week = 60*60*24*7
91 old = now - week
92 self.cursor.execute('delete from %ss where %s_time < %s'%(self.name,
93 self.name, self.db.arg), (old, ))
94
95 class Sessions(BasicDatabase):
96 name = 'session'
97
98 class OneTimeKeys(BasicDatabase):
99 name = 'otk'
100
101 # vim: set et sts=4 sw=4 :