updated

wpcodevo · wpcodevo · commit e38159011c5a · 2022-07-09T16:07:37.000Z
diff --git a/app/oauth2.py b/app/oauth2.py
@@ -1,9 +1,30 @@
+import base64
+from typing import List
 from fastapi import Depends, HTTPException, status
 from fastapi_jwt_auth import AuthJWT
+from pydantic import BaseModel
 
 from . import models
 from .database import get_db
 from sqlalchemy.orm import Session
+from .config import settings
+
+
+class Settings(BaseModel):
+    authjwt_algorithm: str = settings.JWT_ALGORITHM
+    authjwt_decode_algorithms: List[str] = [settings.JWT_ALGORITHM]
+    authjwt_token_location: set = {'cookies', 'headers'}
+    authjwt_access_cookie_key: str = 'access_token'
+    authjwt_refresh_cookie_key: str = 'refresh_token'
+    authjwt_public_key: str = base64.b64decode(
+        settings.JWT_PUBLIC_KEY).decode('utf-8')
+    authjwt_private_key: str = base64.b64decode(
+        settings.JWT_PRIVATE_KEY).decode('utf-8')
+
+
+@AuthJWT.load_config
+def get_config():
+    return Settings()
 
 
 class NotVerified(Exception):
diff --git a/app/routers/auth.py b/app/routers/auth.py
@@ -1,8 +1,6 @@
-import base64
 from datetime import timedelta
-from typing import List
 from fastapi import APIRouter, Request, Response, status, Depends, HTTPException
-from pydantic import BaseModel, EmailStr
+from pydantic import EmailStr
 
 from app import oauth2
 from .. import schemas, models, utils
@@ -12,23 +10,6 @@
 from ..config import settings
 
 
-class Settings(BaseModel):
-    authjwt_algorithm: str = settings.JWT_ALGORITHM
-    authjwt_decode_algorithms: List[str] = [settings.JWT_ALGORITHM]
-    authjwt_token_location: set = {'cookies', 'headers'}
-    authjwt_access_cookie_key: str = 'access_token'
-    authjwt_refresh_cookie_key: str = 'refresh_token'
-    authjwt_public_key: str = base64.b64decode(
-        settings.JWT_PUBLIC_KEY).decode('utf-8')
-    authjwt_private_key: str = base64.b64decode(
-        settings.JWT_PRIVATE_KEY).decode('utf-8')
-
-
-@AuthJWT.load_config
-def get_config():
-    return Settings()
-
-
 router = APIRouter()
 ACCESS_TOKEN_EXPIRES_IN = settings.ACCESS_TOKEN_EXPIRES_IN
 REFRESH_TOKEN_EXPIRES_IN = settings.REFRESH_TOKEN_EXPIRES_IN
