psycopg/psycopg2
1
1
Fork 0
mirror of https://github.com/psycopg/psycopg2.git synced 2024-11-25 18:33:44 +03:00
Code Issues Packages Projects Releases Wiki Activity

Use errors module to catch a specific postgres error

Browse Source
This commit is contained in:
Daniele Varrazzo 2019-03-16 19:07:27 +00:00
parent b0119fef81
commit e8135ee2cf
1 changed files with 3 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/testutils.py
View File

@ -34,6 +34,7 @@ from functools import wraps
from ctypes.util import find_library
import psycopg2
import psycopg2.errors
import psycopg2.extensions
from psycopg2.compat import text_type
@ -372,12 +373,8 @@ def skip_if_no_superuser(f):
def skip_if_no_superuser_(self):
try:
return f(self)
except psycopg2.ProgrammingError as e:
import psycopg2.errorcodes
if e.pgcode == psycopg2.errorcodes.INSUFFICIENT_PRIVILEGE:
except psycopg2.errors.InsufficientPrivilege:
self.skipTest("skipped because not superuser")
else:
raise
return skip_if_no_superuser_