e3ff55b4db
- 完整撰写玉宗珠宝设计大师项目README，介绍项目概况及核心功能 - 说明用户认证系统实现及优势，包含JWT鉴权和密码加密细节 - 详细描述品类管理系统，支持多流程类型和多种玉石品类 - 说明设计图生成方案及技术，包含Pillow生成示例及字体支持 - 介绍设计管理功能，支持分页浏览、预览、下载和删除设计 - 个人信息管理模块说明，涵盖昵称、手机号、密码的安全修改 - 绘制业务流程图和关键数据流图，清晰展现系统架构与数据流 - 提供详细API调用链路及参数说明，涵盖用户、品类、设计接口 - 列明技术栈及版本，包含前后端框架、ORM、认证、加密等工具 - 展示目录结构，标明后端与前端项目布局 - 规划本地开发环境与启动步骤，包括数据库初始化及运行命令 - 说明服务器部署流程和Nginx配置方案 - 详细数据库表结构说明及环境变量配置指导 - 汇总常用开发及测试命令，方便开发调试与部署管理
64 lines
1.8 KiB
Python
64 lines
1.8 KiB
Python
"""
|
||
认证路由
|
||
提供用户注册、登录和获取当前用户信息的 API
|
||
"""
|
||
from fastapi import APIRouter, Depends, HTTPException, status
|
||
from sqlalchemy.orm import Session
|
||
|
||
from ..database import get_db
|
||
from ..schemas.user import UserCreate, UserLogin, UserResponse, Token
|
||
from ..services.auth_service import register_user, authenticate_user
|
||
from ..utils.deps import get_current_user
|
||
from ..utils.security import create_access_token
|
||
from ..models.user import User
|
||
|
||
|
||
router = APIRouter(prefix="/api/auth", tags=["认证"])
|
||
|
||
|
||
@router.post("/register", response_model=UserResponse)
|
||
def register(user_data: UserCreate, db: Session = Depends(get_db)):
|
||
"""
|
||
用户注册
|
||
|
||
创建新用户账号，用户名必须唯一
|
||
"""
|
||
try:
|
||
user = register_user(db, user_data)
|
||
return user
|
||
except ValueError as e:
|
||
raise HTTPException(
|
||
status_code=status.HTTP_400_BAD_REQUEST,
|
||
detail=str(e)
|
||
)
|
||
|
||
|
||
@router.post("/login", response_model=Token)
|
||
def login(user_data: UserLogin, db: Session = Depends(get_db)):
|
||
"""
|
||
用户登录
|
||
|
||
验证用户名和密码，返回 JWT access token
|
||
"""
|
||
user = authenticate_user(db, user_data.username, user_data.password)
|
||
if not user:
|
||
raise HTTPException(
|
||
status_code=status.HTTP_401_UNAUTHORIZED,
|
||
detail="用户名或密码错误",
|
||
headers={"WWW-Authenticate": "Bearer"},
|
||
)
|
||
|
||
# 生成 JWT token，sub 字段存储用户 ID
|
||
access_token = create_access_token(data={"sub": str(user.id)})
|
||
return Token(access_token=access_token, token_type="bearer")
|
||
|
||
|
||
@router.get("/me", response_model=UserResponse)
|
||
def get_me(current_user: User = Depends(get_current_user)):
|
||
"""
|
||
获取当前登录用户信息
|
||
|
||
需要认证，从 token 中解析用户身份
|
||
"""
|
||
return current_user
|