97 lines
3.0 KiB
Python
97 lines
3.0 KiB
Python
# 认证路由 - 使用字段编码
|
|
from fastapi import APIRouter, Depends, HTTPException, status
|
|
from fastapi.security import OAuth2PasswordRequestForm
|
|
from sqlalchemy.orm import Session
|
|
from app.core.database import get_db
|
|
from app.core.auth import verify_password, create_access_token, get_password_hash
|
|
from app.models.models import User
|
|
from app.schemas.schemas import Token, UserCreate, UserResponse
|
|
|
|
router = APIRouter(prefix="/api/auth", tags=["认证"])
|
|
|
|
|
|
@router.post("/register", response_model=UserResponse)
|
|
def register(user_data: UserCreate, db: Session = Depends(get_db)):
|
|
"""用户注册"""
|
|
# 检查用户名是否已存在
|
|
existing_user = db.query(User).filter(User.f01_01_name == user_data.f01_01_name).first()
|
|
if existing_user:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_400_BAD_REQUEST,
|
|
detail="f01_01_name: 用户名已存在"
|
|
)
|
|
|
|
# 检查邮箱是否已存在
|
|
if user_data.email:
|
|
existing_email = db.query(User).filter(User.email == user_data.email).first()
|
|
if existing_email:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_400_BAD_REQUEST,
|
|
detail="邮箱已被注册"
|
|
)
|
|
|
|
# 创建用户
|
|
import uuid
|
|
hashed_password = get_password_hash(user_data.password)
|
|
user = User(
|
|
f99_90_id=str(uuid.uuid4()),
|
|
f99_91_user_id=str(uuid.uuid4()), # 生成唯一 user_id
|
|
f01_01_name=user_data.f01_01_name,
|
|
email=user_data.email,
|
|
phone=user_data.phone,
|
|
avatar=user_data.avatar,
|
|
address=user_data.address,
|
|
bio=user_data.bio,
|
|
password=hashed_password,
|
|
role="user"
|
|
)
|
|
|
|
db.add(user)
|
|
db.commit()
|
|
db.refresh(user)
|
|
|
|
return user
|
|
|
|
|
|
@router.post("/login", response_model=Token)
|
|
def login(
|
|
form_data: OAuth2PasswordRequestForm = Depends(),
|
|
db: Session = Depends(get_db)
|
|
):
|
|
"""用户登录"""
|
|
# 查找用户
|
|
user = db.query(User).filter(User.f01_01_name == form_data.username).first()
|
|
if not user:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED,
|
|
detail="E00011: 用户名或密码错误",
|
|
headers={"WWW-Authenticate": "Bearer"},
|
|
)
|
|
|
|
# 验证密码
|
|
if not verify_password(form_data.password, user.password):
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED,
|
|
detail="E00011: 用户名或密码错误",
|
|
headers={"WWW-Authenticate": "Bearer"},
|
|
)
|
|
|
|
# 生成 token
|
|
access_token = create_access_token(data={"sub": user.f99_90_id})
|
|
|
|
return {
|
|
"access_token": access_token,
|
|
"token_type": "bearer"
|
|
}
|
|
|
|
|
|
@router.get("/me", response_model=UserResponse)
|
|
def get_current_user_info(
|
|
current_user: User = Depends(lambda: None)
|
|
):
|
|
"""获取当前用户信息"""
|
|
raise HTTPException(
|
|
status_code=status.HTTP_501_NOT_IMPLEMENTED,
|
|
detail="请使用正确的依赖注入"
|
|
)
|