Fix #35: catch import error when not having optional DB depedencies

This commit is contained in:
François Voron
2019-10-31 21:43:10 +01:00
parent 658161518a
commit 60c815e425

View File

@ -1,6 +1,14 @@
from fastapi_users.db.base import BaseUserDatabase # noqa: F401
from fastapi_users.db.mongodb import MongoDBUserDatabase # noqa: F401
from fastapi_users.db.sqlalchemy import ( # noqa: F401
SQLAlchemyBaseUserTable,
SQLAlchemyUserDatabase,
)
try:
from fastapi_users.db.mongodb import MongoDBUserDatabase # noqa: F401
except ImportError: # pragma: no cover
pass
try:
from fastapi_users.db.sqlalchemy import ( # noqa: F401
SQLAlchemyBaseUserTable,
SQLAlchemyUserDatabase,
)
except ImportError: # pragma: no cover
pass