set method doesn't include user set timestamp if update

set() is supposed to update the record with the key if it already
exists.

It does update the value marshalled data blob. However it doesn't
update the timestamp column for rdbms tables if provided.

This change updates the x_time column with the provided __timestamp or
preserves the original timestamp.

Author: rouilj

roundup/backends/sessions_rdbms.py

@@ -60,19 +60,27 @@ def set(self, infoid, **newvalues):
         c = self.cursor
         n = self.name
         a = self.db.arg
-        c.execute('select %s_value from %ss where %s_key=%s'%(n, n, n, a),
+        c.execute('select %s_value, %s_time from %ss where %s_key=%s'% \
+                  (n, n, n, n, a),
             (infoid,))
         res = c.fetchone()
         if res:
             values = eval(res[0])
+            timestamp = res[1]
         else:
             values = {}
         values.update(newvalues)
-
         if res:
-            sql = 'update %ss set %s_value=%s where %s_key=%s'%(n, n,
-                a, n, a)
-            args = (repr(values), infoid)
+            if '__timestamp' in newvalues:
+                try:
+                    # __timestamp must be representable as a float. Check it.
+                    timestamp = float(newvalues['__timestamp'])
+                except ValueError:
+                    pass
+
+            sql = ('update %ss set %s_value=%s, %s_time=%s '
+                       'where %s_key=%s'%(n, n, a, n, a, n, a))
+            args = (repr(values), timestamp, infoid)
         else:
             if '__timestamp' in newvalues:
                 try: