初始版本

2025-12-26 13:42:22 +08:00
parent ddb90d6c20
commit b495bc1dca
43 changed files with 2179 additions and 20 deletions
@@ -0,0 +1,4 @@
+"""
+API 路由模块
+"""
+
@@ -0,0 +1,4 @@
+"""
+API v1 路由
+"""
+
@@ -0,0 +1,14 @@
+"""
+API v1 路由聚合
+"""
+from fastapi import APIRouter
+from app.api.api_v1.endpoints import auth, todos, posts, users
+
+api_router = APIRouter()
+
+# 注册各个功能模块的路由
+api_router.include_router(auth.router, prefix="/auth", tags=["认证"])
+api_router.include_router(users.router, prefix="/users", tags=["用户"])
+api_router.include_router(todos.router, prefix="/todos", tags=["待办事项"])
+api_router.include_router(posts.router, prefix="/posts", tags=["博客"])
+
@@ -0,0 +1,4 @@
+"""
+API 端点模块
+"""
+
@@ -0,0 +1,142 @@
+"""
+博客文章相关 API 端点
+"""
+from fastapi import APIRouter, Depends, HTTPException, status
+from sqlmodel import Session, select
+from typing import List
+
+from app.db.session import get_db
+from app.models.post import Post
+from app.models.user import User
+from app.schemas.post import Post as PostSchema, PostCreate, PostUpdate
+from app.api.api_v1.endpoints.users import get_current_user
+
+router = APIRouter()
+
+
+@router.post("/", response_model=PostSchema, status_code=status.HTTP_201_CREATED)
+def create_post(
+    post_in: PostCreate,
+    current_user: User = Depends(get_current_user),
+    db: Session = Depends(get_db)
+):
+    """创建博客文章"""
+    # 检查 slug 是否已存在（同一用户）
+    statement = select(Post).where(
+        Post.slug == post_in.slug,
+        Post.user_id == current_user.id
+    )
+    existing_post = db.exec(statement).first()
+    if existing_post:
+        raise HTTPException(
+            status_code=status.HTTP_400_BAD_REQUEST,
+            detail="该 slug 已存在"
+        )
+    
+    db_post = Post(
+        title=post_in.title,
+        slug=post_in.slug,
+        content=post_in.content,
+        user_id=current_user.id
+    )
+    db.add(db_post)
+    db.commit()
+    db.refresh(db_post)
+    return db_post
+
+
+@router.get("/", response_model=List[PostSchema])
+def read_posts(
+    skip: int = 0,
+    limit: int = 100,
+    current_user: User = Depends(get_current_user),
+    db: Session = Depends(get_db)
+):
+    """获取当前用户的博客文章列表"""
+    statement = (
+        select(Post)
+        .where(Post.user_id == current_user.id)
+        .offset(skip)
+        .limit(limit)
+    )
+    posts = db.exec(statement).all()
+    return posts
+
+
+@router.get("/{post_id}", response_model=PostSchema)
+def read_post(
+    post_id: int,
+    current_user: User = Depends(get_current_user),
+    db: Session = Depends(get_db)
+):
+    """获取单个博客文章"""
+    statement = select(Post).where(
+        Post.id == post_id,
+        Post.user_id == current_user.id
+    )
+    post = db.exec(statement).first()
+    if post is None:
+        raise HTTPException(
+            status_code=status.HTTP_404_NOT_FOUND,
+            detail="文章不存在"
+        )
+    return post
+
+
+@router.put("/{post_id}", response_model=PostSchema)
+def update_post(
+    post_id: int,
+    post_in: PostUpdate,
+    current_user: User = Depends(get_current_user),
+    db: Session = Depends(get_db)
+):
+    """更新博客文章"""
+    statement = select(Post).where(
+        Post.id == post_id,
+        Post.user_id == current_user.id
+    )
+    post = db.exec(statement).first()
+    if post is None:
+        raise HTTPException(
+            status_code=status.HTTP_404_NOT_FOUND,
+            detail="文章不存在"
+        )
+    
+    if post_in.title is not None:
+        post.title = post_in.title
+    if post_in.slug is not None:
+        post.slug = post_in.slug
+    if post_in.content is not None:
+        post.content = post_in.content
+    
+    from datetime import datetime
+    post.updated_at = datetime.now()
+    
+    db.add(post)
+    db.commit()
+    db.refresh(post)
+    return post
+
+
+@router.delete("/{post_id}", status_code=status.HTTP_204_NO_CONTENT)
+def delete_post(
+    post_id: int,
+    current_user: User = Depends(get_current_user),
+    db: Session = Depends(get_db)
+):
+    """删除博客文章"""
+    statement = select(Post).where(
+        Post.id == post_id,
+        Post.user_id == current_user.id
+    )
+    post = db.exec(statement).first()
+    if post is None:
+        raise HTTPException(
+            status_code=status.HTTP_404_NOT_FOUND,
+            detail="文章不存在"
+        )
+    
+    db.delete(post)
+    db.commit()
+    return None
+
@@ -0,0 +1,123 @@
+"""
+待办事项相关 API 端点
+"""
+from fastapi import APIRouter, Depends, HTTPException, status
+from sqlmodel import Session, select
+from typing import List
+
+from app.db.session import get_db
+from app.models.todo import Todo
+from app.models.user import User
+from app.schemas.todo import Todo as TodoSchema, TodoCreate, TodoUpdate
+from app.api.api_v1.endpoints.users import get_current_user
+
+router = APIRouter()
+
+
+@router.post("/", response_model=TodoSchema, status_code=status.HTTP_201_CREATED)
+def create_todo(
+    todo_in: TodoCreate,
+    current_user: User = Depends(get_current_user),
+    db: Session = Depends(get_db)
+):
+    """创建待办事项"""
+    db_todo = Todo(
+        title=todo_in.title,
+        user_id=current_user.id
+    )
+    db.add(db_todo)
+    db.commit()
+    db.refresh(db_todo)
+    return db_todo
+
+
+@router.get("/", response_model=List[TodoSchema])
+def read_todos(
+    skip: int = 0,
+    limit: int = 100,
+    current_user: User = Depends(get_current_user),
+    db: Session = Depends(get_db)
+):
+    """获取当前用户的待办事项列表"""
+    statement = (
+        select(Todo)
+        .where(Todo.user_id == current_user.id)
+        .offset(skip)
+        .limit(limit)
+    )
+    todos = db.exec(statement).all()
+    return todos
+
+
+@router.get("/{todo_id}", response_model=TodoSchema)
+def read_todo(
+    todo_id: int,
+    current_user: User = Depends(get_current_user),
+    db: Session = Depends(get_db)
+):
+    """获取单个待办事项"""
+    statement = select(Todo).where(
+        Todo.id == todo_id,
+        Todo.user_id == current_user.id
+    )
+    todo = db.exec(statement).first()
+    if todo is None:
+        raise HTTPException(
+            status_code=status.HTTP_404_NOT_FOUND,
+            detail="待办事项不存在"
+        )
+    return todo
+
+
+@router.put("/{todo_id}", response_model=TodoSchema)
+def update_todo(
+    todo_id: int,
+    todo_in: TodoUpdate,
+    current_user: User = Depends(get_current_user),
+    db: Session = Depends(get_db)
+):
+    """更新待办事项"""
+    statement = select(Todo).where(
+        Todo.id == todo_id,
+        Todo.user_id == current_user.id
+    )
+    todo = db.exec(statement).first()
+    if todo is None:
+        raise HTTPException(
+            status_code=status.HTTP_404_NOT_FOUND,
+            detail="待办事项不存在"
+        )
+    
+    if todo_in.title is not None:
+        todo.title = todo_in.title
+    if todo_in.done is not None:
+        todo.done = todo_in.done
+    
+    db.add(todo)
+    db.commit()
+    db.refresh(todo)
+    return todo
+
+
+@router.delete("/{todo_id}", status_code=status.HTTP_204_NO_CONTENT)
+def delete_todo(
+    todo_id: int,
+    current_user: User = Depends(get_current_user),
+    db: Session = Depends(get_db)
+):
+    """删除待办事项"""
+    statement = select(Todo).where(
+        Todo.id == todo_id,
+        Todo.user_id == current_user.id
+    )
+    todo = db.exec(statement).first()
+    if todo is None:
+        raise HTTPException(
+            status_code=status.HTTP_404_NOT_FOUND,
+            detail="待办事项不存在"
+        )
+    
+    db.delete(todo)
+    db.commit()
+    return None
+
@@ -0,0 +1,63 @@
+"""
+用户相关 API 端点
+"""
+from fastapi import APIRouter, Depends, HTTPException, status
+from sqlmodel import Session, select
+from typing import List
+
+from app.db.session import get_db
+from app.models.user import User
+from app.schemas.user import User as UserSchema
+from app.core.security import decode_access_token
+from fastapi.security import OAuth2PasswordBearer
+
+router = APIRouter()
+oauth2_scheme = OAuth2PasswordBearer(tokenUrl="/api/v1/auth/login")
+
+
+async def get_current_user(
+    token: str = Depends(oauth2_scheme),
+    db: Session = Depends(get_db)
+) -> User:
+    """获取当前登录用户"""
+    payload = decode_access_token(token)
+    if payload is None:
+        raise HTTPException(
+            status_code=status.HTTP_401_UNAUTHORIZED,
+            detail="无效的认证凭据",
+            headers={"WWW-Authenticate": "Bearer"},
+        )
+    
+    username: str = payload.get("sub")
+    if username is None:
+        raise HTTPException(
+            status_code=status.HTTP_401_UNAUTHORIZED,
+            detail="无效的认证凭据",
+            headers={"WWW-Authenticate": "Bearer"},
+        )
+    
+    statement = select(User).where(User.username == username)
+    user = db.exec(statement).first()
+    if user is None:
+        raise HTTPException(
+            status_code=status.HTTP_401_UNAUTHORIZED,
+            detail="用户不存在",
+            headers={"WWW-Authenticate": "Bearer"},
+        )
+    
+    return user
+
+
+@router.get("/me", response_model=UserSchema)
+async def read_users_me(current_user: User = Depends(get_current_user)):
+    """获取当前用户信息"""
+    return current_user
+
+
+@router.get("/", response_model=List[UserSchema])
+def read_users(skip: int = 0, limit: int = 100, db: Session = Depends(get_db)):
+    """获取用户列表（示例端点）"""
+    statement = select(User).offset(skip).limit(limit)
+    users = db.exec(statement).all()
+    return users
+