py.lib: 1668cc57225b db/pg.py

1 # -*- coding: utf-8 -*-

2 import psycopg2

3 from psycopg2 import Error, Warning, DataError, IntegrityError, ProgrammingError

4 # DBError: Клас-родитель для всех ошибок

5 # DataError: Проблема с данными (деление на ноль, численное изначение за областью значение и пр.)

6 # IntegrityError: задеты ограничения целостности БД (уникольность, внешние ключи и пр.)

7 # ProgrammingError: ошибки в запросах (таблица не найдена, или уже существует, ошибка в синтаксисе запроса и пр.)

9 import psycopg2.extensions

10 psycopg2.extensions.register_type(psycopg2.extensions.UNICODE)

11 psycopg2.extensions.register_type(psycopg2.extensions.UNICODEARRAY)

15 class DB(object):

16 def __init__(self, **kva):

17 self._conn = psycopg2.connect(**kva)

18 self.commit = self._conn.commit

19 self.rollback = self._conn.rollback

20 self.cursor = self._conn.cursor

21 self.reset = self._conn.reset

23 def __enter__(self, *a, **kva):

24 return self._conn.__enter__(*a, **kva)

26 def __exit__(self, *a, **kva):

27 return self._conn.__exit__(*a, **kva)

29 def __call__(self, *a, **kva):

30 _cur = self.cursor()

31 _cur.execute(*a, **kva)

32 return _cur

34 def cq(self, *a, **kva):

35 try:

36 _cur = self.cursor()

37 _cur.execute(*a, **kva)

38 self.commit()

39 return _cur

40 except Exception as e:

41 self.rollback()

42 raise e

44 def close(self):

45 try:

46 self.rollback()

47 self._conn.close()

48 except:

49 pass

51 def __del__(self):

52 self.close()

54 def __repr__(self):

55 return 'DB(%s)' % self._conn.dsn

py.lib