376 lines
13 KiB
Python
376 lines
13 KiB
Python
"""
|
||
管理员仪表盘相关 API
|
||
"""
|
||
from fastapi import APIRouter, Depends, HTTPException, Query
|
||
from sqlalchemy.ext.asyncio import AsyncSession
|
||
from sqlalchemy import select, func, and_, or_
|
||
from datetime import datetime, date
|
||
from typing import Optional
|
||
import os
|
||
import glob
|
||
import json
|
||
|
||
from app.core.database import get_db
|
||
from app.core.deps import get_current_user
|
||
from app.core.config import settings
|
||
from app.models.user import User
|
||
from app.models.project import Project, ProjectMember
|
||
from app.models.log import OperationLog
|
||
from app.core.enums import OperationType, ResourceType
|
||
from app.schemas.response import success_response
|
||
|
||
router = APIRouter()
|
||
|
||
|
||
@router.get("/stats", response_model=dict)
|
||
async def get_dashboard_stats(
|
||
current_user: User = Depends(get_current_user),
|
||
db: AsyncSession = Depends(get_db)
|
||
):
|
||
"""获取仪表盘统计数据（仅管理员）"""
|
||
|
||
# 检查是否为超级管理员
|
||
if current_user.is_superuser != 1:
|
||
raise HTTPException(status_code=403, detail="仅管理员可访问")
|
||
|
||
# 统计用户数
|
||
user_count_result = await db.execute(select(func.count(User.id)))
|
||
user_count = user_count_result.scalar()
|
||
|
||
# 统计项目数
|
||
project_count_result = await db.execute(select(func.count(Project.id)))
|
||
project_count = project_count_result.scalar()
|
||
|
||
# 统计文档数（所有项目中的 .md 文件）
|
||
document_count = 0
|
||
if os.path.exists(settings.PROJECTS_PATH):
|
||
for project_dir in os.listdir(settings.PROJECTS_PATH):
|
||
project_path = os.path.join(settings.PROJECTS_PATH, project_dir)
|
||
if os.path.isdir(project_path):
|
||
md_files = glob.glob(os.path.join(project_path, "**/*.md"), recursive=True)
|
||
document_count += len(md_files)
|
||
|
||
# 获取最近创建的用户
|
||
recent_users_result = await db.execute(
|
||
select(User)
|
||
.order_by(User.created_at.desc())
|
||
.limit(5)
|
||
)
|
||
recent_users = recent_users_result.scalars().all()
|
||
recent_users_data = [
|
||
{
|
||
"id": user.id,
|
||
"username": user.username,
|
||
"email": user.email,
|
||
"created_at": user.created_at.isoformat() if user.created_at else None,
|
||
}
|
||
for user in recent_users
|
||
]
|
||
|
||
# 获取最近创建的项目（包含所有者信息）
|
||
recent_projects_result = await db.execute(
|
||
select(Project, User)
|
||
.join(User, Project.owner_id == User.id)
|
||
.order_by(Project.created_at.desc())
|
||
.limit(5)
|
||
)
|
||
recent_projects_rows = recent_projects_result.all()
|
||
recent_projects_data = [
|
||
{
|
||
"id": project.id,
|
||
"name": project.name,
|
||
"description": project.description,
|
||
"owner_name": owner.username,
|
||
"created_at": project.created_at.isoformat() if project.created_at else None,
|
||
}
|
||
for project, owner in recent_projects_rows
|
||
]
|
||
|
||
return success_response(
|
||
data={
|
||
"stats": {
|
||
"user_count": user_count,
|
||
"project_count": project_count,
|
||
"document_count": document_count,
|
||
},
|
||
"recent_users": recent_users_data,
|
||
"recent_projects": recent_projects_data,
|
||
}
|
||
)
|
||
|
||
|
||
@router.get("/personal-stats", response_model=dict)
|
||
async def get_personal_stats(
|
||
current_user: User = Depends(get_current_user),
|
||
db: AsyncSession = Depends(get_db)
|
||
):
|
||
"""获取个人桌面统计数据"""
|
||
|
||
# 统计个人项目数
|
||
personal_projects_count_result = await db.execute(
|
||
select(func.count(Project.id)).where(Project.owner_id == current_user.id)
|
||
)
|
||
personal_projects_count = personal_projects_count_result.scalar()
|
||
|
||
# 统计参加项目数（协作项目）
|
||
shared_projects_count_result = await db.execute(
|
||
select(func.count(Project.id))
|
||
.join(ProjectMember, Project.id == ProjectMember.project_id)
|
||
.where(ProjectMember.user_id == current_user.id)
|
||
.where(Project.owner_id != current_user.id)
|
||
)
|
||
shared_projects_count = shared_projects_count_result.scalar()
|
||
|
||
# 统计个人文档数（个人项目中的 .md 文件）
|
||
document_count = 0
|
||
personal_projects_result = await db.execute(
|
||
select(Project).where(Project.owner_id == current_user.id)
|
||
)
|
||
personal_projects = personal_projects_result.scalars().all()
|
||
|
||
for project in personal_projects:
|
||
project_path = os.path.join(settings.PROJECTS_PATH, project.storage_key)
|
||
if os.path.exists(project_path) and os.path.isdir(project_path):
|
||
md_files = glob.glob(os.path.join(project_path, "**/*.md"), recursive=True)
|
||
document_count += len(md_files)
|
||
|
||
# 获取最近的个人项目
|
||
recent_personal_projects_result = await db.execute(
|
||
select(Project)
|
||
.where(Project.owner_id == current_user.id)
|
||
.order_by(Project.created_at.desc())
|
||
.limit(5)
|
||
)
|
||
recent_personal_projects = recent_personal_projects_result.scalars().all()
|
||
recent_personal_projects_data = [
|
||
{
|
||
"id": project.id,
|
||
"name": project.name,
|
||
"description": project.description,
|
||
"created_at": project.created_at.isoformat() if project.created_at else None,
|
||
}
|
||
for project in recent_personal_projects
|
||
]
|
||
|
||
# 获取最近的分享项目（从 project_members 表）
|
||
recent_shared_projects_result = await db.execute(
|
||
select(Project, ProjectMember)
|
||
.join(ProjectMember, Project.id == ProjectMember.project_id)
|
||
.where(ProjectMember.user_id == current_user.id)
|
||
.where(Project.owner_id != current_user.id)
|
||
.order_by(ProjectMember.joined_at.desc())
|
||
.limit(5)
|
||
)
|
||
recent_shared_projects_rows = recent_shared_projects_result.all()
|
||
recent_shared_projects_data = [
|
||
{
|
||
"id": project.id,
|
||
"name": project.name,
|
||
"description": project.description,
|
||
"role": member.role,
|
||
"joined_at": member.joined_at.isoformat() if member.joined_at else None,
|
||
}
|
||
for project, member in recent_shared_projects_rows
|
||
]
|
||
|
||
return success_response(
|
||
data={
|
||
"user_info": {
|
||
"id": current_user.id,
|
||
"username": current_user.username,
|
||
"email": current_user.email,
|
||
"created_at": current_user.created_at.isoformat() if current_user.created_at else None,
|
||
},
|
||
"stats": {
|
||
"personal_projects_count": personal_projects_count,
|
||
"shared_projects_count": shared_projects_count,
|
||
"document_count": document_count,
|
||
},
|
||
"recent_personal_projects": recent_personal_projects_data,
|
||
"recent_shared_projects": recent_shared_projects_data,
|
||
}
|
||
)
|
||
|
||
|
||
@router.get("/document-activity-dates", response_model=dict)
|
||
async def get_document_activity_dates(
|
||
year: int = Query(..., description="年份"),
|
||
month: int = Query(..., description="月份"),
|
||
current_user: User = Depends(get_current_user),
|
||
db: AsyncSession = Depends(get_db)
|
||
):
|
||
"""获取指定月份有文档操作的日期列表"""
|
||
# 计算月份的开始和结束日期
|
||
start_date = datetime(year, month, 1)
|
||
if month == 12:
|
||
end_date = datetime(year + 1, 1, 1)
|
||
else:
|
||
end_date = datetime(year, month + 1, 1)
|
||
|
||
# 查询该用户在指定月份内的文档操作日志
|
||
# 文档操作包括：创建文件、保存文件、删除文件、重命名文件、移动文件
|
||
document_operations = [
|
||
OperationType.CREATE_FILE,
|
||
OperationType.SAVE_FILE,
|
||
OperationType.DELETE_FILE,
|
||
OperationType.RENAME_FILE,
|
||
OperationType.MOVE_FILE,
|
||
]
|
||
|
||
result = await db.execute(
|
||
select(func.date(OperationLog.created_at).label('activity_date'), func.count(OperationLog.id).label('count'))
|
||
.where(
|
||
and_(
|
||
OperationLog.user_id == current_user.id,
|
||
OperationLog.operation_type.in_(document_operations),
|
||
OperationLog.created_at >= start_date,
|
||
OperationLog.created_at < end_date
|
||
)
|
||
)
|
||
.group_by(func.date(OperationLog.created_at))
|
||
.order_by(func.date(OperationLog.created_at))
|
||
)
|
||
|
||
activity_dates = result.all()
|
||
dates_data = [
|
||
{
|
||
"date": activity_date.isoformat(),
|
||
"count": count
|
||
}
|
||
for activity_date, count in activity_dates
|
||
]
|
||
|
||
return success_response(data={"dates": dates_data})
|
||
|
||
|
||
@router.get("/document-activity", response_model=dict)
|
||
async def get_document_activity(
|
||
date_str: str = Query(..., description="日期（YYYY-MM-DD）"),
|
||
current_user: User = Depends(get_current_user),
|
||
db: AsyncSession = Depends(get_db)
|
||
):
|
||
"""获取指定日期的文档操作日志（按项目+文件+操作类型聚合）"""
|
||
# 解析日期
|
||
try:
|
||
target_date = datetime.strptime(date_str, "%Y-%m-%d").date()
|
||
except ValueError:
|
||
raise HTTPException(status_code=400, detail="日期格式错误，应为 YYYY-MM-DD")
|
||
|
||
# 计算日期范围
|
||
start_datetime = datetime.combine(target_date, datetime.min.time())
|
||
end_datetime = datetime.combine(target_date, datetime.max.time())
|
||
|
||
# 查询该用户在指定日期的文档操作日志
|
||
document_operations = [
|
||
OperationType.CREATE_FILE,
|
||
OperationType.SAVE_FILE,
|
||
OperationType.DELETE_FILE,
|
||
OperationType.RENAME_FILE,
|
||
OperationType.MOVE_FILE,
|
||
]
|
||
|
||
result = await db.execute(
|
||
select(OperationLog)
|
||
.where(
|
||
and_(
|
||
OperationLog.user_id == current_user.id,
|
||
OperationLog.operation_type.in_(document_operations),
|
||
OperationLog.created_at >= start_datetime,
|
||
OperationLog.created_at <= end_datetime
|
||
)
|
||
)
|
||
.order_by(OperationLog.created_at.desc())
|
||
)
|
||
|
||
logs = result.scalars().all()
|
||
|
||
# 操作类型中文映射
|
||
operation_map = {
|
||
OperationType.CREATE_FILE: "创建",
|
||
OperationType.SAVE_FILE: "保存",
|
||
OperationType.DELETE_FILE: "删除",
|
||
OperationType.RENAME_FILE: "重命名",
|
||
OperationType.MOVE_FILE: "移动",
|
||
}
|
||
|
||
# 聚合日志：按 (project_id, file_path, operation_type) 分组
|
||
aggregated = {}
|
||
|
||
for log in logs:
|
||
# 解析 detail 字段获取文件路径和项目ID
|
||
detail = json.loads(log.detail) if log.detail else {}
|
||
project_id = detail.get('project_id')
|
||
file_path = detail.get('path') or detail.get('file_path') or detail.get('old_path')
|
||
|
||
# 创建聚合键
|
||
key = (project_id, file_path, log.operation_type)
|
||
|
||
if key not in aggregated:
|
||
aggregated[key] = {
|
||
'project_id': project_id,
|
||
'file_path': file_path,
|
||
'operation_type': log.operation_type,
|
||
'count': 0,
|
||
'first_time': log.created_at,
|
||
'last_time': log.created_at,
|
||
}
|
||
|
||
aggregated[key]['count'] += 1
|
||
# 更新最早和最晚时间
|
||
if log.created_at < aggregated[key]['first_time']:
|
||
aggregated[key]['first_time'] = log.created_at
|
||
if log.created_at > aggregated[key]['last_time']:
|
||
aggregated[key]['last_time'] = log.created_at
|
||
|
||
# 构建返回数据，包含项目信息
|
||
logs_data = []
|
||
for key, agg in aggregated.items():
|
||
project_id = agg['project_id']
|
||
file_path = agg['file_path']
|
||
operation_type = agg['operation_type']
|
||
|
||
# 获取项目信息
|
||
project_name = None
|
||
project_storage_key = None
|
||
if project_id:
|
||
project_result = await db.execute(
|
||
select(Project).where(Project.id == project_id)
|
||
)
|
||
project = project_result.scalar_one_or_none()
|
||
if project:
|
||
project_name = project.name
|
||
project_storage_key = project.storage_key
|
||
|
||
# 检查文件是否存在（仅针对非删除操作）
|
||
file_exists = False
|
||
if project_storage_key and file_path and operation_type != OperationType.DELETE_FILE:
|
||
full_path = os.path.join(settings.PROJECTS_PATH, project_storage_key, file_path)
|
||
file_exists = os.path.exists(full_path) and os.path.isfile(full_path)
|
||
|
||
# 生成描述文本
|
||
operation_text = operation_map.get(operation_type, operation_type)
|
||
if agg['count'] > 1:
|
||
description = f"{operation_text} {agg['count']} 次"
|
||
else:
|
||
description = operation_text
|
||
|
||
logs_data.append({
|
||
"id": f"{project_id}_{file_path}_{operation_type}", # 唯一ID
|
||
"operation_type": description,
|
||
"operation_count": agg['count'],
|
||
"project_id": project_id,
|
||
"project_name": project_name or "未知项目",
|
||
"file_path": file_path or "未知文件",
|
||
"file_exists": file_exists,
|
||
"first_time": agg['first_time'].isoformat() if agg['first_time'] else None,
|
||
"last_time": agg['last_time'].isoformat() if agg['last_time'] else None,
|
||
"created_at": agg['last_time'].isoformat() if agg['last_time'] else None, # 用最后操作时间排序
|
||
})
|
||
|
||
# 按最后操作时间降序排序
|
||
logs_data.sort(key=lambda x: x['last_time'] if x['last_time'] else '', reverse=True)
|
||
|
||
return success_response(data={"logs": logs_data})
|
||
|
||
|