Register add, refactor

app/api/v1/endpoints.py

@@ -0,0 +1,40 @@
+from app.schemas.user import UserSchema
+from app.schemas.token import TokenSchema
+
+from app.db.demo import users_db
+
+from app.utils.jwt_utlis import encode_jwt
+from app.utils.validate_utils import validate_auth_user, get_current_auth_user, validate_register_user
+
+from fastapi import APIRouter, Depends
+
+router = APIRouter(prefix="/auth")
+
+@router.post("/login/", response_model=TokenSchema)
+def auth_user_with_password(
+        user: UserSchema = Depends(validate_auth_user)
+):
+    jwt_payload = {
+        "username": user.username
+    }
+    token = encode_jwt(jwt_payload)
+    return TokenSchema(
+        access_token=token,
+        token_type="Bearer"
+    )
+
+@router.post("/register/")
+def sign_up(
+        user: UserSchema = Depends(validate_register_user)
+):
+    users_db[user.username] = user
+
+    return {"Message": "User added",
+            "Username": user.username}
+
+@router.get("/validate/")
+def validate_user(
+        user: UserSchema = Depends(get_current_auth_user)
+):
+    return {"Message": "Validate pass",
+            "Username": user.username}