From 77294f16f85ef4f0006b35546e66042fd5a43a9c Mon Sep 17 00:00:00 2001
From: Tony Mack <tmack@cs.princeton.edu>
Date: Thu, 31 Jul 2008 18:10:11 +0000
Subject: [PATCH] merge from trunk. changes how we obtain serial primary keys

---
 PLC/PostgreSQL.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/PLC/PostgreSQL.py b/PLC/PostgreSQL.py
index 4e056394..22c2b63f 100644
--- a/PLC/PostgreSQL.py
+++ b/PLC/PostgreSQL.py
@@ -134,6 +134,15 @@ class PostgreSQL:
         cursor.close()
         return self.rowcount
 
+    def next_id(self, table_name, primary_key):
+        sequence = "%(table_name)s_%(primary_key)s_seq" % locals()
+        sql = "SELECT nextval('%(sequence)s')" % locals()
+        rows = self.selectall(sql, hashref = False)
+        if rows:
+            return rows[0][0]
+
+        return None
+
     def last_insert_id(self, table_name, primary_key):
         if isinstance(self.lastrowid, int):
             sql = "SELECT %s FROM %s WHERE oid = %d" % \
-- 
2.47.0