Ticket #8575: update_sequence.patch

trac/db/sqlite_backend.py

@@ -315 +315 @@
 
     def get_last_id(self, cursor, table, column='id'):
         return cursor.lastrowid
+    
+    def update_sequence(self, cursor, table, column='id'):
+        # sqlite handles sequence updates automagically
+        # http://www.sqlite.org/autoinc.html
+        pass

trac/db/tests/api.py

@@ -320 +320 @@
         id2 = self.db.get_last_id(c, 'report')
         self.assertEqual(id1 + 1, id2)
 
+    def test_update_sequence(self):
+        c = self.db.cursor()
+        q = "INSERT INTO report (id, author) VALUES (42, 'anonymous')"
+        c.execute(q)
+        self.db.update_sequence(c, "report")
+        id1 = self.db.get_last_id(c, 'report')
+        self.assertEqual(id1, 42)
 
+
+
 def suite():
     suite = unittest.TestSuite()
     suite.addTest(unittest.makeSuite(ParseConnectionStringTestCase, 'test'))

trac/db/mysql_backend.py

@@ -253 +253 @@
     def get_last_id(self, cursor, table, column='id'):
         return cursor.lastrowid
 
+    def update_sequence(self, cursor, table, column='id'):
+        # mysql handles this automagically, verified in 
+        # mysql  Ver 14.14 Distrib 5.1.41, for debian-linux-gnu (x86_64) 
+        # using readline 6.1
+        pass
+
     def rollback(self):
         self.cnx.ping()
         try:

trac/db/postgres_backend.py

@@ -235 +235 @@
         cursor.execute("SELECT CURRVAL('%s_%s_seq')" % (table, column))
         return cursor.fetchone()[0]
 
+    def update_sequence(self, cursor, table, column='id'):
+        sql = "SELECT setval('%s_%s_seq', (SELECT MAX(id) FROM %s))" % \
+            (table, column, table)
+        cursor.execute(sql)
+
     def cursor(self):
         return IterableCursor(self.cnx.cursor(), self.log)