useless: fix syntax warning

storage: shorten code
This commit is contained in:
2020-03-16 22:04:29 +00:00
parent e03f5d0a43
commit 80cfa2f815
2 changed files with 5 additions and 7 deletions

View File

@@ -15,28 +15,26 @@ class DBConn:
def _reconnect(self):
self._con = psycopg2.connect(*self.args, **self.kwargs)
return self._con
def commit(self, *args, **kwargs):
try:
return self._con.commit(*args, **kwargs)
except psycopg2.InterfaceError:
self._reconnect()
return self._con.commit(*args, **kwargs)
return self._reconnect().commit(*args, **kwargs)
def rollback(self, *args, **kwargs):
try:
return self._con.rollback(*args, **kwargs)
except psycopg2.InterfaceError:
self._reconnect()
return self._con.rollback(*args, **kwargs)
return self._reconnect().rollback(*args, **kwargs)
@contextlib.contextmanager
def cursor(self, *args, **kwargs):
try:
yield self._con.cursor(cursor_factory=DictCursor, *args, **kwargs)
except psycopg2.InterfaceError:
self._reconnect()
yield self._con.cursor(cursor_factory=DictCursor, *args, **kwargs)
yield self._reconnect().cursor(cursor_factory=DictCursor, *args, **kwargs)
class Storage(Plugin):