feat: mcp server and arguments

Author: 7086cmd

main.py

@@ -15,6 +15,7 @@
     groups_v2_router,
     activity_statistics_router,
     times_router,
+    mcp_router,
 )
 from database import close_mongo_connection, connect_to_mongo
 import socketio
@@ -138,9 +139,10 @@ async def shutdown_event():
 app.include_router(
     activity_statistics_router.router, prefix="/api/v2/statistics/activities"
 )
+app.include_router(mcp_router.router, prefix="/api/v2/mcp")
 
 
-@app.router.get("/api/")
+@app.get("/api/")
 async def home():
     return {
         "status": "ok",

routers/activities_v2_router.py

@@ -3,7 +3,7 @@
 from datetime import datetime
 from typing import Literal, cast
 from bson import ObjectId
-from fastapi import APIRouter, Depends, HTTPException
+from fastapi import APIRouter, Depends, HTTPException, Query
 from fastapi.responses import JSONResponse
 from pydantic import BaseModel
 from database import db
@@ -57,13 +57,13 @@ async def create_activity_v2(
 
 @router.get("")
 async def get_activities_v2(
-    page: int = 1,
-    perpage: int = 10,
-    search: str = "",
-    sort: str = "_id",
-    asc: bool = False,
-    regex: bool = False,
-    activity_type: str = "all",
+    page: int = Query(1, ge=1, description="Page number for pagination"),
+    perpage: int = Query(10, ge=1, le=100, description="Number of items per page"),
+    search: str = Query("", description="Search string for filtering activities"),
+    sort: str = Query("_id", description="Sort field"),
+    asc: bool = Query(False, description="Sort order (ascending or descending)"),
+    regex: bool = Query(False, description="Whether to use regex for searching"),
+    activity_type: str = Query("all", description="Type of activity to filter by"),
     user=Depends(get_current_user),
 ):
     """
@@ -191,12 +191,12 @@ async def delete_activity_v2(
 @router.get("/{activity_id}/members")
 async def get_activity_members_v2(
     activity_id: str,
-    page: int = 1,
-    perpage: int = 10,
-    search: str = "",
-    sort: str = "_id",
-    asc: bool = True,
-    regex: bool = False,
+    page: int = Query(1, ge=1, description="Page number for pagination"),
+    perpage: int = Query(10, ge=1, le=100, description="Number of items per page"),
+    search: str = Query("", description="Search string for filtering members"),
+    sort: str = Query("_id", description="Sort field"),
+    asc: bool = Query(True, description="Sort order (ascending or descending)"),
+    regex: bool = Query(False, description="Whether to use regex for searching"),
     user=Depends(get_current_user),
 ):
     """

routers/groups_router.py

@@ -7,7 +7,7 @@
 from typings.export import ExportFormat
 from typings.group import Group
 from bson import ObjectId
-from fastapi import APIRouter, HTTPException, Depends
+from fastapi import APIRouter, HTTPException, Depends, Query
 from database import db
 from pydantic import BaseModel
 
@@ -48,10 +48,9 @@ async def create_group(payload: Group, user=Depends(get_current_user)):
 
 @router.get("")
 async def get_groups(
-    page: int = 1,
-    perpage: int = 10,
-    type="all",
-    search="",
+    page: int = Query(1, ge=1, description="Page number for pagination"),
+    perpage: int = Query(10, ge=1, le=100, description="Number of items per page"),
+    search: str = Query("", description="Search term for group names"),
     user=Depends(get_current_user),
 ):
     """
@@ -61,23 +60,13 @@ async def get_groups(
     if len(user["per"]) == 0:
         raise HTTPException(status_code=403, detail="Permission denied")
 
-    if type == "all":
-        target = ["permission", "class"]
-    elif type == "permission":
-        target = ["permission"]
-    elif type == "class":
-        target = ["class"]
-    else:
-        raise HTTPException(status_code=400, detail="Invalid type")
-
     count = await db.zvms.groups.count_documents(
         {"name": {"$regex": search, "$options": "i"}}
     )
 
     pipeline = [
         {
             "$match": {
-                "type": {"$in": target},
                 "name": {"$regex": search, "$options": "i"},
             },
         },
@@ -148,9 +137,9 @@ async def update_group_name(
 @router.get("/{group_id}/activities")
 async def get_class_activities(
     group_id: str,
-    page: int = 1,
-    perpage: int = 10,
-    query: str = "",
+    page: int = Query(1, ge=1, description="Page number for pagination"),
+    perpage: int = Query(10, ge=1, le=100, description="Number of items per page"),
+    query: str = Query("", description="Search query for activities"),
     user=Depends(get_current_user),
 ):
     """
@@ -190,10 +179,10 @@ async def get_class_activities(
 @router.get("/{group_id}/user")
 async def get_users_in_class(
     group_id: str,
-    page: int = 1,
-    perpage: int = 10,
-    search: str = "",
-    pwdm: bool = False,
+    page: int = Query(1, ge=1, description="Page number for pagination"),
+    perpage: int = Query(10, ge=1, le=100, description="Number of items per page"),
+    search: str = Query("", description="Search term for user names"),
+    pwdm: bool = Query(False, description="Include password mode information"),
     user=Depends(get_current_user),
 ):
     """
@@ -238,14 +227,14 @@ async def get_users_in_class(
 @router.get("/{group_id}/time")
 async def get_user_times_in_class(
     group_id: str,
-    page: int = 1,
-    perpage: int = 10,
-    exceeding: bool = True,
-    shortage: bool = False,
-    start: Optional[str] = None,
-    end: Optional[str] = None,
-    search: str = "",
-    allow_cache: bool = True,
+    page: int = Query(1, ge=1, description="Page number for pagination"),
+    perpage: int = Query(10, ge=1, le=100, description="Number of items per page"),
+    exceeding: bool = Query(True, description="Include exceeding time calculations"),
+    shortage: bool = Query(False, description="Include shortage calculations"),
+    start: Optional[str] = Query(None, description="Start date filter"),
+    end: Optional[str] = Query(None, description="End date filter"),
+    search: str = Query("", description="Search term for user names"),
+    allow_cache: bool = Query(True, description="Allow cached time calculations"),
     user=Depends(get_current_user),
 ):
     """

routers/groups_v2_router.py

@@ -3,7 +3,7 @@
 from typing import Optional
 
 from bson import ObjectId
-from fastapi import APIRouter, Depends
+from fastapi import APIRouter, Depends, Query
 
 from config import (
     BASE_OFF_CAMPUS,
@@ -26,12 +26,12 @@
 @router.get("/{group_id}/activities")
 async def get_group_activities_v2(
     group_id: str,
-    page: int = 1,
-    perpage: int = 10,
-    search: str = "",
-    sort: str = "_id",
-    regex: bool = True,
-    asc: bool = False,
+    page: int = Query(1, ge=1, description="Page number for pagination"),
+    perpage: int = Query(10, ge=1, le=100, description="Number of items per page"),
+    search: str = Query("", description="Search keyword"),
+    sort: str = Query("_id", description="Sort by field"),
+    regex: bool = Query(True, description="Use regex for search"),
+    asc: bool = Query(False, description="Sort in ascending order"),
     user=Depends(get_current_user),
 ):
     """
@@ -120,12 +120,12 @@ async def get_group_activities_v2(
 @router.get("/{group_id}/time")
 async def read_users(
     group_id: str,
-    query: str = "",
-    page: int = 1,
-    perpage: int = 5,
-    allow_cache: bool = True,
-    sort: str = "id",
-    asc: bool = True,
+    query: str = Query("", description="Search query for user name or ID"),
+    page: int = Query(1, ge=1, description="Page number for pagination"),
+    perpage: int = Query(5, ge=1, le=100, description="Number of items per page"),
+    allow_cache: bool = Query(True, description="Allow cached data"),
+    sort: str = Query("id", description="Sort by field"),
+    asc: bool = Query(True, description="Sort in ascending order"),
     user: Optional[str] = Depends(get_current_user),
 ):
     """
@@ -201,16 +201,16 @@ async def read_users(
 @router.get("/{group_id}/users")
 async def get_group_users_v2(
     group_id: str,
-    page: int = 1,
-    perpage: int = 10,
-    search: str = "",
-    sort: str = "id",
-    regex: bool = True,
-    asc: bool = True,
-    exceeding: bool = True,
-    shortage: bool = False,
-    start: Optional[str] = None,
-    end: Optional[str] = None,
+    page: int = Query(1, ge=1, description="Page number for pagination"),
+    perpage: int = Query(10, ge=1, le=100, description="Number of items per page"),
+    search: str = Query("", description="Search keyword"),
+    sort: str = Query("id", description="Sort by field"),
+    regex: bool = Query(True, description="Use regex for search"),
+    asc: bool = Query(True, description="Sort in ascending order"),
+    exceeding: bool = Query(True, description="Include exceeding time"),
+    shortage: bool = Query(False, description="Include shortage time"),
+    start: Optional[str] = Query(None, description="Start date filter"),
+    end: Optional[str] = Query(None, description="End date filter"),
     user=Depends(get_current_user),
 ):
     """

routers/logs_router.py

@@ -1,6 +1,6 @@
 from typing import Any
 
-from fastapi import APIRouter, Depends, HTTPException
+from fastapi import APIRouter, Depends, HTTPException, Query
 
 from config import MAX_QUERY_COUNT
 from database import db
@@ -11,10 +11,10 @@
 
 @router.get("")
 async def read_logs(
-    performer: str = "",
-    page: int = -1,
-    perpage: int = 10,
-    query: str = "",
+    performer: str = Query("", description="Filter by performer/user"),
+    page: int = Query(-1, ge=-1, description="Page number (-1 for no pagination)"),
+    perpage: int = Query(10, ge=1, le=100, description="Number of items per page"),
+    query: str = Query("", max_length=MAX_QUERY_COUNT, description="Search query"),
     user=Depends(get_current_user),
 ):
     if "admin" not in user["per"]: