github/sprockets-postgres
1
0
Fork 0
mirror of https://github.com/sprockets/sprockets-postgres.git synced 2024-12-27 20:24:44 +00:00
Code Issues Projects Releases Packages Wiki Activity

Make on_error optional for PostgresConnector

Browse source
This commit is contained in:
Andrew Rabert 2020-09-16 17:20:50 -04:00
parent 69b806f9b3
commit e57646df00
1 changed files with 7 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
sprockets_postgres.py
View file

@ -110,7 +110,7 @@ class PostgresConnector:
"""
def __init__(self,
cursor: aiopg.Cursor,
on_error: typing.Callable,
on_error: typing.Optional[typing.Callable] = None,
on_duration: typing.Optional[typing.Callable] = None,
timeout: Timeout = None):
self.cursor = cursor
@ -246,9 +246,10 @@ class PostgresConnector:
try:
await method(**kwargs)
except (asyncio.TimeoutError, psycopg2.Error) as err:
exc = self._on_error(metric_name, err)
if exc:
raise exc
if self._on_error:
err = self._on_error(metric_name, err)
if err:
raise err
else:
results = await self._query_results()
if self._on_duration:
@ -313,7 +314,8 @@ class ApplicationMixin:
@contextlib.asynccontextmanager
async def postgres_connector(self,
on_error: typing.Callable,
on_error: typing.Optional[
typing.Callable] = None,
on_duration: typing.Optional[
typing.Callable] = None,
timeout: Timeout = None,