py.lib

     1.1 --- a/db/pg.py	Sun Jun 05 15:29:59 2022 +0300
     1.2 +++ b/db/pg.py	Wed Jul 06 23:32:51 2022 +0300
     1.3 @@ -3,15 +3,15 @@
     1.4  from psycopg2 import Error, Warning, DataError, IntegrityError, ProgrammingError
     1.5  # DBError: Клас-родитель для всех ошибок
     1.6  # DataError: Проблема с данными (деление на ноль, численное изначение за областью значение и пр.)
     1.7 -# IntegrityError: задеты ограничения целостности БД (уникольность, внешние ключи и пр.)
     1.8 +# IntegrityError: задеты ограничения целостности БД (уникальность, внешние ключи и пр.)
     1.9  # ProgrammingError: ошибки в запросах (таблица не найдена, или уже существует, ошибка в синтаксисе запроса и пр.)
    1.10  
    1.11 +
    1.12  import psycopg2.extensions
    1.13  psycopg2.extensions.register_type(psycopg2.extensions.UNICODE)
    1.14  psycopg2.extensions.register_type(psycopg2.extensions.UNICODEARRAY)
    1.15  
    1.16  
    1.17 -
    1.18  class DB(object):
    1.19  	def __init__(self, **kva):
    1.20  		self._conn = psycopg2.connect(**kva)
    1.21 @@ -29,6 +29,7 @@
    1.22  	def __call__(self, *a, **kva):
    1.23  		_cur = self.cursor()
    1.24  		_cur.execute(*a, **kva)
    1.25 +
    1.26  		return _cur
    1.27  		
    1.28  	def cq(self, *a, **kva):
    1.29 @@ -37,6 +38,7 @@
    1.30  			_cur.execute(*a, **kva)
    1.31  			self.commit()
    1.32  			return _cur
    1.33 +
    1.34  		except Exception as e:
    1.35  			self.rollback()
    1.36  			raise e
    1.37 @@ -45,6 +47,7 @@
    1.38  		try:
    1.39  			self.rollback()
    1.40  			self._conn.close()
    1.41 +
    1.42  		except:
    1.43  			pass		
    1.44