From: Tony Mack <tmack@cs.princeton.edu>
Date: Tue, 3 Jun 2008 02:27:02 +0000 (+0000)
Subject: new method next_id(...) returns next primary key id
X-Git-Tag: PLCAPI-4.3-1~33
X-Git-Url: http://git.onelab.eu/?a=commitdiff_plain;h=bb508a5a3a28e71913beaa983c4e02177afc7ace;p=plcapi.git

new method next_id(...) returns next primary key id
---

diff --git a/PLC/PostgreSQL.py b/PLC/PostgreSQL.py
index 597077a..5c12257 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" % \