Initial commit

auth/urls/auth.py

@@ -0,0 +1,68 @@
+from typing import Optional, List
+
+from fastapi import Depends, APIRouter, Body, Request
+from fastapi.security import SecurityScopes
+from pydantic import BaseModel
+
+from auth.models import UserModel, OAuth2PasswordRequestFormJson, CreateTokenModel, OAuth2OptRequest, \
+    PasswordForgottenInput
+from auth.utils import get_current_active_user, get_current_user, generate_token, reset_password
+from config import conf
+from core.app import app
+from core.utils import to_http_res
+from core.utils.router import BaseRoute, safe_route
+
+_pre = conf.server.prefix_url
+
+router = APIRouter(
+    prefix=f"{_pre}/auth",
+    route_class=BaseRoute,
+)
+
+_tag_base = ['token']
+
+
+class ValidateTokenInput(BaseModel):
+    token: str
+    scopes: Optional[List[str]] = ['account']
+
+
+@router.post('/login/otp', response_model=CreateTokenModel, tags=_tag_base)
+async def login_with_otp(request: Request, form_data: OAuth2OptRequest = Depends()):
+    client_host = request.client.host
+    return await safe_route('login', generate_token, form_data, client_host, request.headers.get('user-agent'))
+
+
+@router.post("/token", response_model=CreateTokenModel, tags=_tag_base)
+async def login_for_access_token(request: Request, form_data: OAuth2PasswordRequestFormJson = Depends()):
+    client_host = request.client.host
+    return await safe_route('login', generate_token, form_data, client_host, request.headers.get('user-agent'))
+
+
+@router.post('/login', response_model=CreateTokenModel, tags=_tag_base)
+async def login(request: Request, data: OAuth2PasswordRequestFormJson = Depends()):
+    client_host = request.client.host
+    return await safe_route('login', generate_token, data, client_host, request.headers.get('user-agent'))
+
+
+@router.post('/reset-password', response_model=bool, tags=_tag_base)
+async def reset_password_p(_input: PasswordForgottenInput):
+    return await safe_route('update', reset_password, _input)
+
+
+@router.get('/logout', response_model=bool, tags=_tag_base)
+async def logout():
+    return to_http_res(True, 'token')
+
+
+@router.get("/status", tags=_tag_base)
+async def token_status(current_user: UserModel = Depends(get_current_active_user)):
+    return to_http_res('get', {"status": "token is valid", "scope": current_user.scopes})
+
+
+@router.post('/validate-token', tags=_tag_base)
+async def validate_token(body: ValidateTokenInput = Body(...)):
+    return await safe_route('get', get_current_user, SecurityScopes(scopes=body.scopes), body.token)
+
+
+app.include_router(router)