auth

app.py

@@ -7,7 +7,6 @@ from global_state import set
 parent_dir = Path(__file__).resolve().parent
 set('project_root', parent_dir)
 
-print('\n\n\n======================')
 
 from fastapi import FastAPI, HTTPException, Response, Depends
 from fastapi.responses import HTMLResponse
@@ -22,7 +21,6 @@ from routers.users_v1 import router as users_router
 from dotenv import load_dotenv
 import os
 
-print('\n\n\n======================')
 # 当在本地使用.env 文件时加载.env文件
 load_dotenv()
 

auth.py

@@ -0,0 +1,30 @@
+from fastapi import Depends, HTTPException, status
+from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
+from global_state import get
+from db.tbs_db import TbsDb
+
+# 创建一个 HTTPBearer 实例
+security = HTTPBearer()
+
+def get_current_user_id(credentials: HTTPAuthorizationCredentials = Depends(security)):
+    token = credentials.credentials
+    token = credentials.credentials
+    # 假设你有一个函数来验证Token并返回用户ID
+    user_id = validate_token(token)
+    if user_id is None:
+        raise HTTPException(
+            status_code=status.HTTP_401_UNAUTHORIZED,
+            detail="Invalid authentication credentials",
+            headers={"WWW-Authenticate": "Bearer"},
+        )
+    return user_id
+
+def validate_token(token: str):
+    db_module_filename = f"{get('project_root')}/db/cloudflare.py"
+    query = f"SELECT * FROM users where api_key='{token}'"
+    response = TbsDb(db_module_filename, "Cloudflare").get_item(query)
+    result = response['result'][0]['results']
+    if len(result) == 0:
+        return None
+    user_id = result[0]['id']
+    return user_id

db/cloudflare.py

@@ -1,5 +1,5 @@
 import requests, os
-from dotenv import load_dotenv
+# from dotenv import load_dotenv
 
 class Cloudflare():
     def __init__(self):
@@ -27,6 +27,6 @@ class Cloudflare():
         }
         response = requests.post(url, headers=headers, json=input)
         resp_json = response.json()
-        print(resp_json)
+
         return resp_json
     

routers/users_v1.py

@@ -1,16 +1,17 @@
-from fastapi import APIRouter
+from fastapi import APIRouter, Depends
 from pydantic import BaseModel
 import uuid
 
 from global_state import get
 from db.tbs_db import TbsDb
+from auth import get_current_user_id
 
 router = APIRouter()
 
 db_module_filename = f"{get('project_root')}/db/cloudflare.py"
 
 @router.get("/users")
-async def read_users():
+async def read_users(current_user_id: int = Depends(get_current_user_id)):
     query = "SELECT * FROM users"
     response = TbsDb(db_module_filename, "Cloudflare").get_list(query)
     return response
@@ -22,7 +23,7 @@ class User(BaseModel):
     nikename: str = None
 
 @router.post("/users")
-async def create_user(user: User):
+async def create_user(user: User, current_user_id: int = Depends(get_current_user_id)): 
     username = user.username
     password = user.password
     email = user.email
@@ -36,7 +37,7 @@ async def create_user(user: User):
     return response
 
 @router.get("/users/{id}")
-async def read_user(id:int):
+async def read_user(id:int, current_user_id: int = Depends(get_current_user_id)):
     query = f"SELECT * FROM users where id={id}"
     response = TbsDb(db_module_filename, "Cloudflare").get_item(query)
     return response