Fix import error when OAuth is not installed

This commit is contained in:
François Voron
2020-06-02 07:57:42 +02:00
parent 8431a4b1b2
commit 6ff5adfdf2
2 changed files with 11 additions and 3 deletions

View File

@ -1,19 +1,23 @@
from typing import Any, Callable, Dict, Optional, Sequence, Type
from fastapi import APIRouter, Request
from httpx_oauth.oauth2 import BaseOAuth2
from fastapi_users import models
from fastapi_users.authentication import Authenticator, BaseAuthentication
from fastapi_users.db import BaseUserDatabase
from fastapi_users.router import (
get_auth_router,
get_oauth_router,
get_register_router,
get_reset_password_router,
get_users_router,
)
try:
from httpx_oauth.oauth2 import BaseOAuth2
from fastapi_users.router import get_oauth_router
except ModuleNotFoundError: # pragma: no cover
BaseOAuth2 = Type
class FastAPIUsers:
"""

View File

@ -1,6 +1,10 @@
from fastapi_users.router.auth import get_auth_router # noqa: F401
from fastapi_users.router.common import ErrorCode # noqa: F401
from fastapi_users.router.oauth import get_oauth_router # noqa: F401
from fastapi_users.router.register import get_register_router # noqa: F401
from fastapi_users.router.reset import get_reset_password_router # noqa: F401
from fastapi_users.router.users import get_users_router # noqa: F401
try:
from fastapi_users.router.oauth import get_oauth_router # noqa: F401
except ModuleNotFoundError: # pragma: no cover
pass