Mercurial Repository: p/roundup/code: roundup/backends/back

comparison roundup/backends/back_postgresql.py @ 8302:82a26ea1afdf

issue2551376: Fix tracebacks in item templates

author	Ralf Schlatterbeck <rsc@runtux.com>
date	Wed, 26 Feb 2025 11:24:13 +0100
parents	cfa7d43a3658
children	45ec660eb7f7

comparison

equal deleted inserted replaced

-:3ba8078843db
+:82a26ea1afdf
 #   FATAL:  database "rounduptest" does not exist\n'
 # ugh.
 #
 # Database name is any character sequence not including a " or
 # whitespace. Arguably both are allowed by:
 #
 #  https://www.postgresql.org/docs/current/sql-syntax-lexical.html#SQL-SYNTAX-IDENTIFIERS
 #
 # with suitable quoting but ... really.
 search = re.search(
 r'FATAL:\s+database\s+"([^"\s]*)"\s+does\s+not\s+exist',
 #   FATAL:  database "rounduptest" does not exist\n'
 # ugh.
 #
 # Database name is any character sequence not including a " or
 # whitespace. Arguably both are allowed by:
 #
 #  https://www.postgresql.org/docs/current/sql-syntax-lexical.html#SQL-SYNTAX-IDENTIFIERS
 #
 # with suitable quoting but ... really.
 search = re.search(
 r'FATAL:\s+database\s+"([^"\s]*)"\s+does\s+not\s+exist',
 # reset the sequences
 for cn in self.classes:
 self.cursor.execute('DROP SEQUENCE _%s_ids' % cn)
 self.cursor.execute('CREATE SEQUENCE _%s_ids' % cn)
+def getnode (self, classname, nodeid, fetch_multilinks=True):
+""" For use of savepoint see 'Class' below """
+self.sql('savepoint sp')
+try:
+getnode = rdbms_common.Database.getnode
+return getnode(self, classname, nodeid, fetch_multilinks)
+except psycopg2.errors.DataError as err:
+self.sql('rollback to savepoint sp')
+raise hyperdb.HyperdbValueError(str (err).split('\n')[0])
 class PostgresqlClass:
 order_by_null_values = '(%s is not NULL)'
 case_insensitive_like = 'ILIKE'
 class Class(PostgresqlClass, rdbms_common.Class):
-pass
+""" We re-raise database-specific data errors as HyperdbValueError
+Note that we re-use the savepoint so that at most one savepoint
+is used.
+"""
+def filter(self, *args, **kw):
+self.db.sql('savepoint sp')
+try:
+return rdbms_common.Class.filter(self, *args, **kw)
+except psycopg2.errors.DataError as err:
+self.db.sql('rollback to savepoint sp')
+raise hyperdb.HyperdbValueError(str (err).split('\n')[0])
+def filter_iter(self, *args, **kw):
+self.db.sql('savepoint sp')
+try:
+return rdbms_common.Class.filter_iter(self, *args, **kw)
+except psycopg2.errors.DataError as err:
+self.db.sql('rollback to savepoint sp')
+raise hyperdb.HyperdbValueError(str (err).split('\n')[0])
+def is_retired(self, nodeid):
+self.db.sql('savepoint sp')
+try:
+return rdbms_common.Class.is_retired(self, nodeid)
+except psycopg2.errors.DataError as err:
+self.db.sql('rollback to savepoint sp')
+raise hyperdb.HyperdbValueError (str (err).split('\n')[0])
 class IssueClass(PostgresqlClass, rdbms_common.IssueClass):
 pass

Mercurial > p > roundup > code

comparison roundup/backends/back_postgresql.py @ 8302:82a26ea1afdf