Wrong imports whupps
This commit is contained in:
@@ -4,7 +4,7 @@ import os
|
||||
from typing import Any, AsyncGenerator, Callable, TypeVar
|
||||
|
||||
from sqlalchemy import Boolean, Column, Date, DateTime, ForeignKey, Integer, String
|
||||
from sqlalchemy.exc import DBAPIError, InternalServerError
|
||||
from sqlalchemy.exc import DBAPIError
|
||||
from sqlalchemy.ext.asyncio import AsyncEngine, AsyncSession, create_async_engine, async_sessionmaker
|
||||
from sqlalchemy.orm import declarative_base, relationship
|
||||
|
||||
@@ -146,7 +146,7 @@ class ResilientAsyncSession:
|
||||
try:
|
||||
async with self.async_sessionmaker() as session:
|
||||
return await func(session, *args, **kwargs)
|
||||
except (InternalServerError, DBAPIError) as e:
|
||||
except DBAPIError as e:
|
||||
last_error = e
|
||||
error_msg = str(e).lower()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user