deliveryman-api/app/api/deps.py
2025-01-02 16:33:39 +08:00

13 lines
414 B
Python

from fastapi import Depends, HTTPException
from sqlalchemy.orm import Session
from app.models.database import get_db
from app.models.user import UserDB
async def get_current_user(
phone: str,
db: Session = Depends(get_db)
) -> UserDB:
user = db.query(UserDB).filter(UserDB.phone == phone).first()
if not user:
raise HTTPException(status_code=401, detail="用户未登录")
return user