2026-05-29 16:20:30 +08:00
|
|
|
|
from datetime import date
|
|
|
|
|
|
from ninja_extra import api_controller, ControllerBase, route
|
|
|
|
|
|
import json
|
|
|
|
|
|
from typing import Literal
|
|
|
|
|
|
from apps.project.models import Project
|
|
|
|
|
|
from django.db import transaction
|
|
|
|
|
|
from django.contrib.auth import get_user_model
|
|
|
|
|
|
import requests
|
|
|
|
|
|
from utils.chen_response import ChenResponse
|
|
|
|
|
|
from django.db.models import Q
|
|
|
|
|
|
from ninja import Schema
|
|
|
|
|
|
|
|
|
|
|
|
Users = get_user_model()
|
|
|
|
|
|
|
|
|
|
|
|
class AIPostSchema(Schema):
|
|
|
|
|
|
question: str
|
|
|
|
|
|
project_type: Literal["cpu", "fpga"]
|
|
|
|
|
|
|
|
|
|
|
|
# AI测试接口
|
|
|
|
|
|
@api_controller("/local_doc_qa", tags=['AI测试接口'])
|
|
|
|
|
|
class AITestController(ControllerBase):
|
|
|
|
|
|
"""AI测试接口：自定义延迟"""
|
|
|
|
|
|
|
|
|
|
|
|
@route.post("/testing_item")
|
|
|
|
|
|
def ai_return(self, item: AIPostSchema):
|
|
|
|
|
|
target_url = "http://192.168.0.63:8777/api/local_doc_qa/testing_item"
|
|
|
|
|
|
payload = {
|
|
|
|
|
|
"question": item.question,
|
|
|
|
|
|
"model_name": "qwen3.5", # 可能会变
|
|
|
|
|
|
"project_type": item.project_type,
|
|
|
|
|
|
"streaming": False,
|
|
|
|
|
|
"user_focus_points": "",
|
|
|
|
|
|
}
|
|
|
|
|
|
try:
|
|
|
|
|
|
resp = requests.post(target_url, json=payload, timeout=120, headers={"Content-Type": "application/json"})
|
|
|
|
|
|
resp.raise_for_status()
|
|
|
|
|
|
return resp.json()
|
|
|
|
|
|
except requests.RequestException:
|
|
|
|
|
|
# 调用失败，返回 502
|
|
|
|
|
|
return ChenResponse(data={}, message="调用大模型接口失败，请联系管理员", code=502, status=502)
|
|
|
|
|
|
|
|
|
|
|
|
# 这是其他common内容接口
|
|
|
|
|
|
@api_controller("/system", tags=['通用接口'])
|
|
|
|
|
|
class CommonController(ControllerBase):
|
|
|
|
|
|
"""通用接口类：工作台内的信息"""
|
|
|
|
|
|
|
|
|
|
|
|
@route.get("/getNoticeList")
|
|
|
|
|
|
def get_notice(self, pageSize, orderBy, orderType):
|
|
|
|
|
|
item_list = []
|
|
|
|
|
|
item1 = {"title": "测试管理平台V0.0.2测试发布", "created_at": "2023-09-23",
|
|
|
|
|
|
"content": "测试管理平台V0.0.2发布，正在进行内部测试.."}
|
|
|
|
|
|
item_list.append(item1)
|
|
|
|
|
|
item2 = {"title": "测试管理平台更新公共", "created_at": "2024-06-17",
|
|
|
|
|
|
"content": "<p>1.修改大纲和报告模版<p><p>2.修复多个bug<p>"}
|
|
|
|
|
|
item_list.append(item2)
|
|
|
|
|
|
return item_list
|
|
|
|
|
|
|
|
|
|
|
|
@route.get('/workplace/statistics')
|
|
|
|
|
|
@transaction.atomic
|
|
|
|
|
|
def get_statistics(self):
|
|
|
|
|
|
# 查询用户数量，进行的项目，项目总数，已完成项目数
|
|
|
|
|
|
user_count = Users.objects.count()
|
|
|
|
|
|
project_qs = Project.objects.all()
|
|
|
|
|
|
project_count = project_qs.count()
|
|
|
|
|
|
project_done_count = project_qs.filter(step='3').count()
|
|
|
|
|
|
project_processing_count = project_qs.filter(Q(step='1') | Q(step='2')).count()
|
|
|
|
|
|
return ChenResponse(data={'pcount': project_count, 'ucount': user_count,
|
|
|
|
|
|
'pdcount': project_done_count, 'ppcount': project_processing_count})
|
|
|
|
|
|
|
|
|
|
|
|
@route.get('/statistics/chart')
|
|
|
|
|
|
@transaction.atomic
|
|
|
|
|
|
def get_chart(self):
|
|
|
|
|
|
"""该接口返回当前年份下，每月的项目统计，返回横坐标12个月的字符串以及12个月数据"""
|
|
|
|
|
|
current_year = date.today().year
|
|
|
|
|
|
month_list = []
|
|
|
|
|
|
# 构造数组，里面是字典
|
|
|
|
|
|
for i in range(12):
|
|
|
|
|
|
month_dict = {'month': i + 1, 'count': 0}
|
|
|
|
|
|
month_list.append(month_dict)
|
|
|
|
|
|
project_qs = Project.objects.all()
|
|
|
|
|
|
for project in project_qs:
|
|
|
|
|
|
for m in month_list:
|
|
|
|
|
|
if m['month'] == project.beginTime.month and project.beginTime.year == current_year:
|
|
|
|
|
|
m['count'] += 1
|
|
|
|
|
|
return ChenResponse(status=200, code=200, data=month_list)
|
