498 lines
22 KiB
Python
498 lines
22 KiB
Python
from pathlib import Path
|
||
from datetime import date
|
||
from typing import List
|
||
from shutil import copytree, rmtree
|
||
from django.shortcuts import get_object_or_404
|
||
from django.db import transaction
|
||
from ninja_extra import api_controller, ControllerBase, route
|
||
from ninja_extra.permissions import IsAuthenticated
|
||
from ninja_jwt.authentication import JWTAuth
|
||
from apps.user.models import Users
|
||
from utils.chen_pagination import MyPagination
|
||
from ninja.pagination import paginate
|
||
from ninja.errors import HttpError
|
||
from ninja import Query
|
||
from utils.chen_response import ChenResponse
|
||
from utils.chen_crud import create, multi_delete_project
|
||
from apps.project.models import Project, Round, ProjectSoftSummary, StuctSortData, StaticSoftItem, StaticSoftHardware, DynamicSoftTable, \
|
||
DynamicHardwareTable, ProjectDynamicDescription, EvaluateData, EnvAnalysis
|
||
from apps.project.schemas.project import ProjectRetrieveSchema, ProjectFilterSchema, ProjectCreateInput, \
|
||
DeleteSchema, SoftSummarySchema, DataSchema, StaticDynamicData, EnvAnalysisSchema
|
||
from utils.util import get_str_dict
|
||
# 时间处理模块
|
||
from apps.project.tool.timeList import time_return_to
|
||
# 反射工具
|
||
from utils.smallTools.interfaceTools import conditionNoneToBlank
|
||
|
||
media_path = Path.cwd() / 'media'
|
||
base_document_path = Path.cwd() / 'conf/base_document'
|
||
|
||
@api_controller("/testmanage/project", auth=JWTAuth(), permissions=[IsAuthenticated], tags=['项目表相关'])
|
||
class ProjectController(ControllerBase):
|
||
@route.get("/index", response=List[ProjectRetrieveSchema])
|
||
@paginate(MyPagination)
|
||
def list_project(self, filters: ProjectFilterSchema = Query(...)):
|
||
conditionNoneToBlank(filters)
|
||
# 处理时间范围
|
||
start_time = self.context.request.GET.get('searchOnlyTimeRange[0]')
|
||
if start_time is None:
|
||
start_time = "2000-01-01"
|
||
end_time = self.context.request.GET.get('searchOnlyTimeRange[1]')
|
||
if end_time is None:
|
||
end_time = '9999-01-01'
|
||
date_list = [start_time, end_time]
|
||
# 前端返回的member
|
||
member_list = []
|
||
for key, value in self.context.request.GET.items():
|
||
if key.find('member') != -1:
|
||
member_list.append(self.context.request.GET[key])
|
||
qs = Project.objects.filter(
|
||
ident__icontains=filters.ident, name__icontains=filters.name,
|
||
beginTime__range=date_list, duty_person__icontains=filters.duty_person,
|
||
security_level__icontains=filters.security_level,
|
||
report_type__icontains=filters.report_type, step__icontains=filters.step,
|
||
member__contains=member_list, secret__icontains=filters.secret).order_by(
|
||
"-create_datetime")
|
||
# 对软件类型进行处理
|
||
if filters.soft_type != '':
|
||
qs = qs.filter(soft_type=filters.soft_type)
|
||
|
||
# ~~role:查询项目的负责人和成员：普通用户只能看到自己参加的项目~~
|
||
final_qs = []
|
||
auth_info: Users = self.context.request.auth
|
||
if auth_info:
|
||
if auth_info.role != 'admin':
|
||
for proj in qs:
|
||
if proj.duty_person == auth_info.name or auth_info.name in proj.member:
|
||
final_qs.append(proj)
|
||
return final_qs
|
||
return qs
|
||
|
||
@route.get("/findOneById/{int:project_id}", response=ProjectRetrieveSchema)
|
||
@transaction.atomic
|
||
def get_project_by_id(self, project_id: int):
|
||
project_obj = get_object_or_404(Project, id=project_id)
|
||
return project_obj
|
||
|
||
@route.post("/save")
|
||
@transaction.atomic
|
||
def create_project(self, data: ProjectCreateInput):
|
||
data_dict = data.dict()
|
||
ident_qucover = Project.objects.filter(ident=data.dict()['ident'])
|
||
if ident_qucover:
|
||
return ChenResponse(code=400, status=400, message="项目标识重复，请重新设置")
|
||
qs = create(self.context.request, data_dict, Project)
|
||
# 创建项目时候自动添加第一轮测试
|
||
if qs:
|
||
Round.objects.create(project_id=qs.id, key='0', level='0', title='第1轮测试', name='第1轮测试',
|
||
remark='第一轮测试', ident=''.join([qs.ident, '-R1']))
|
||
# 在新增项目时，将/conf/base_document 移动到 /media/{项目ident}/下面
|
||
src_dir = base_document_path
|
||
dist_dir = media_path / qs.ident
|
||
try:
|
||
copytree(src_dir, dist_dir) # shutil模块直接是复制并命名，如果命名文件存在则抛出FileExists异常
|
||
except PermissionError:
|
||
return ChenResponse(code=500, status=500,
|
||
message="错误，检查是否打开了服务器的conf中的文档，关闭后重试")
|
||
except FileExistsError:
|
||
return ChenResponse(code=500, status=500, message='文件标识已存在或输入为空格，请修改')
|
||
except FileNotFoundError:
|
||
return ChenResponse(code=500, status=500, message='文件不存在，请检查')
|
||
return ChenResponse(code=200, status=200, message="添加项目成功，并添加第一轮测试")
|
||
return ChenResponse(code=400, status=400, message="未添加任何项目")
|
||
|
||
@route.put("/update/{project_id}")
|
||
@transaction.atomic
|
||
def update_project(self, project_id: int, payload: ProjectCreateInput):
|
||
# 判断标识是否是被允许的字符串
|
||
project = self.get_object_or_exception(Project, id=project_id)
|
||
old_ident = project.ident
|
||
# 更新操作
|
||
for attr, value in payload.dict().items():
|
||
setattr(project, attr, value)
|
||
project.save()
|
||
new_ident = project.ident
|
||
# 如果新ident不等于老ident，则做 1.更新文件夹名称 2.更新所有轮次中的ident
|
||
if new_ident != old_ident:
|
||
try:
|
||
Path(media_path / old_ident).rename(media_path / project.ident)
|
||
# 同时要更改round和dut的标识
|
||
for r in project.pField.all():
|
||
r.ident = r.ident.replace(old_ident, new_ident)
|
||
r.save()
|
||
for d in project.pdField.all():
|
||
d.ident = d.ident.replace(old_ident, new_ident)
|
||
d.save()
|
||
except PermissionError:
|
||
return ChenResponse(code=500, status=500, message="错误，请关闭文件资源管理器再试")
|
||
except FileExistsError:
|
||
return ChenResponse(code=500, status=500, message='文件标识已存在或输入为空格，请修改')
|
||
except FileNotFoundError:
|
||
return ChenResponse(code=500, status=500, message='文件不存在，请检查')
|
||
return ChenResponse(code=200, status=200, message="项目更新成功")
|
||
|
||
@route.delete("/delete")
|
||
@transaction.atomic
|
||
def delete(self, data: DeleteSchema):
|
||
idents = multi_delete_project(data.ids, Project)
|
||
# 查询media所属项目文件夹，并删除
|
||
for ident in idents:
|
||
project_media_path = media_path / ident
|
||
try:
|
||
rmtree(project_media_path)
|
||
except FileNotFoundError:
|
||
return ChenResponse(status=400, code=400, message='项目模版目录可能不存在，可能之前已删除')
|
||
return ChenResponse(message="删除成功！")
|
||
|
||
# 看板页面接口
|
||
@route.get('/board')
|
||
@transaction.atomic
|
||
def board(self, id: int):
|
||
project_obj = get_object_or_404(Project, id=id)
|
||
# 1.项目阶段直接转字符串
|
||
step_str = get_str_dict(project_obj.step, 'step')
|
||
# 2.返回时间信息
|
||
# 3.返回人员信息
|
||
# 4.返回研制方信息
|
||
# 5.返回用例信息
|
||
case_qs = project_obj.pcField.all()
|
||
exe_count = 0 # 已执行数量
|
||
noexe_count = 0 # 未执行数量
|
||
partexe_count = 0 # 部分执行数量
|
||
## 5.1 计算已执行的用例数 -> 所以的都通过/未通过才算执行，否则部分执行
|
||
for case in case_qs:
|
||
steps = case.step.all()
|
||
steps_count = steps.count() # 步骤总数
|
||
passed_steps_count = steps.filter(passed='1').count()
|
||
notPassed_steps_count = steps.filter(passed='2').count()
|
||
notExe_steps_count = steps_count - passed_steps_count - notPassed_steps_count
|
||
if notExe_steps_count > 0:
|
||
# 步骤全是未执行，则用例未执行
|
||
if notExe_steps_count == steps_count:
|
||
noexe_count += 1
|
||
else:
|
||
partexe_count += 1
|
||
else:
|
||
exe_count += 1
|
||
|
||
# 6.计算问题单数
|
||
problems = project_obj.projField.all()
|
||
close_count = 0
|
||
open_count = 0
|
||
for problem in problems:
|
||
if problem.status != '1':
|
||
open_count += 1
|
||
else:
|
||
close_count += 1
|
||
|
||
# 7.将时间提取 todo:后续将计算的事件放入该页面
|
||
timers = {'round_time': []}
|
||
rounds = project_obj.pField.all()
|
||
timers['start_time'] = project_obj.beginTime # type:ignore
|
||
timers['end_time'] = project_obj.endTime # type:ignore
|
||
for round in rounds:
|
||
round_number = int(round.key) + 1
|
||
timers['round_time'].append({
|
||
'name': f'第{round_number}轮次',
|
||
'start': round.beginTime,
|
||
'end': round.endTime
|
||
})
|
||
|
||
# 8.提取所有需求下面测试项、用例数量
|
||
# 9.提取测试类型下面测试项数量、用例数量
|
||
data_list = []
|
||
for round in rounds:
|
||
round_dict = {'name': f'第{int(round.key) + 1}轮次', 'desings': [], 'method_demand': {},
|
||
'method_case': {}}
|
||
designs = round.dsField.all()
|
||
for design in designs:
|
||
design_dict = {
|
||
'name': design.name,
|
||
'demand_count': design.dtField.count(),
|
||
'case_count': design.dcField.count()
|
||
}
|
||
round_dict['desings'].append(design_dict)
|
||
demands = round.rtField.all()
|
||
for demand in demands:
|
||
test_type = get_str_dict(demand.testType, 'testType')
|
||
if test_type not in round_dict['method_demand']:
|
||
round_dict['method_demand'][test_type] = 1
|
||
else:
|
||
round_dict['method_demand'][test_type] += 1
|
||
cases = round.rcField.all()
|
||
for case in cases:
|
||
testDemand = case.test
|
||
case_type = get_str_dict(testDemand.testType, 'testType')
|
||
if case_type not in round_dict['method_case']:
|
||
round_dict['method_case'][case_type] = 1
|
||
else:
|
||
round_dict['method_case'][case_type] += 1
|
||
data_list.append(round_dict)
|
||
|
||
return {
|
||
'ident': project_obj.ident,
|
||
'name': project_obj.name,
|
||
'step': step_str,
|
||
'title_info': {
|
||
'时间': {
|
||
'开始时间': project_obj.beginTime,
|
||
'结束时间': project_obj.endTime,
|
||
'到现在时间': f"{(date.today() - project_obj.beginTime).days}天",
|
||
},
|
||
'人员': {
|
||
'负责人': project_obj.duty_person,
|
||
'成员数': len(project_obj.member),
|
||
},
|
||
'开发方信息': {
|
||
'联系人': project_obj.dev_contact,
|
||
'电话': project_obj.dev_contact_phone,
|
||
'邮箱': project_obj.dev_email
|
||
},
|
||
'用例数': {
|
||
'总数': case_qs.count(),
|
||
'已执行': exe_count,
|
||
'未执行': noexe_count,
|
||
'部分执行': partexe_count,
|
||
},
|
||
'问题数': {
|
||
'总数': problems.count(),
|
||
'已闭环': close_count,
|
||
'未闭环': open_count,
|
||
}
|
||
},
|
||
'time_line': timers,
|
||
'statistics': data_list,
|
||
}
|
||
|
||
# 看板页面的生成文档时间接口
|
||
@route.get('/document_time_show')
|
||
@transaction.atomic
|
||
def document_time_show(self, id: int):
|
||
time = time_return_to(id)
|
||
return time
|
||
|
||
# [变] 项目级信息前端告警数据获取
|
||
@route.get("/project_info_status/")
|
||
@transaction.atomic
|
||
def project_info_status(self, id: int):
|
||
project_obj = self.get_project_by_id(id)
|
||
|
||
# 统一配置每个状态的检查逻辑
|
||
status_configs = {
|
||
"soft_summary": {
|
||
"model": ProjectSoftSummary,
|
||
"check": lambda qs: qs.exists() and qs.first().data_schemas.exists()
|
||
},
|
||
"interface_image": {
|
||
"model": StuctSortData,
|
||
"check": lambda qs: qs.exists()
|
||
},
|
||
"static_soft_item": {
|
||
"model": StaticSoftItem,
|
||
"check": lambda qs: qs.exists()
|
||
},
|
||
"static_soft_hardware": {
|
||
"model": StaticSoftHardware,
|
||
"check": lambda qs: qs.exists()
|
||
},
|
||
"dynamic_soft_item": {
|
||
"model": DynamicSoftTable,
|
||
"check": lambda qs: qs.exists()
|
||
},
|
||
"dynamic_soft_hardware": {
|
||
"model": DynamicHardwareTable,
|
||
"check": lambda qs: qs.exists()
|
||
},
|
||
"dynamic_des": {
|
||
"model": ProjectDynamicDescription,
|
||
"check": lambda qs: qs.exists() and qs.first().data_schemas.exists()
|
||
},
|
||
"evaluate_data": {
|
||
"model": EvaluateData,
|
||
"check": lambda qs: qs.exists()
|
||
},
|
||
"env_analysis": {
|
||
"model": EnvAnalysis,
|
||
"check": lambda qs: qs.exists()
|
||
}
|
||
}
|
||
|
||
all_status = {}
|
||
for status_key, config in status_configs.items():
|
||
qs = config["model"].objects.filter(project=project_obj)
|
||
all_status[status_key] = config["check"](qs)
|
||
return ChenResponse(status=200, code=20000, data=all_status, message='查询成功')
|
||
|
||
# [变] 封装结构化数据新增-修改（针对project - OneToOne - DataSchemas形式）
|
||
@classmethod
|
||
def bulk_create_data_schemas(cls, parent_obj, datas: list[DataSchema]):
|
||
"""
|
||
批量创建结构化排序数据 (自动类型推断)
|
||
Args:
|
||
parent_obj: 父级对象，可以是 ProjectSoftSummary 或 Project 的实例
|
||
datas (list[DataSchema]): 数据模式对象列表
|
||
"""
|
||
# 动态确定所属父model
|
||
field_name = None # type:ignore
|
||
if isinstance(parent_obj, ProjectSoftSummary):
|
||
field_name = 'soft_summary'
|
||
elif isinstance(parent_obj, Project):
|
||
field_name = 'project'
|
||
elif isinstance(parent_obj, ProjectDynamicDescription):
|
||
field_name = 'dynamic_description'
|
||
else:
|
||
raise HttpError(400, "添加的数据未在系统内，请联系管理员")
|
||
|
||
data_list = []
|
||
for data in datas:
|
||
new_data = StuctSortData(
|
||
type=data.type,
|
||
fontnote=data.fontnote,
|
||
content=data.content,
|
||
)
|
||
setattr(new_data, field_name, parent_obj)
|
||
data_list.append(new_data)
|
||
StuctSortData.objects.bulk_create(data_list)
|
||
|
||
# 封装只有model不同 -修改和新增dataSchemas（针对project - OneToOne - DataSchemas形式）
|
||
@classmethod
|
||
def create_or_modify_data_schemas(cls, id: int, model, data):
|
||
project_obj = get_object_or_404(Project, pk=id)
|
||
qs = model.objects.filter(project=project_obj)
|
||
if qs.exists():
|
||
obj = qs.first()
|
||
# 如果存在则修改：先删除再创建
|
||
obj.data_schemas.all().delete()
|
||
cls.bulk_create_data_schemas(obj, data)
|
||
else:
|
||
parent_obj = model.objects.create(project=project_obj)
|
||
cls.bulk_create_data_schemas(parent_obj, data)
|
||
|
||
# ~~~软件概述-新增和修改~~~
|
||
@route.post('/soft_summary/')
|
||
@transaction.atomic
|
||
def soft_summary(self, payload: SoftSummarySchema):
|
||
self.create_or_modify_data_schemas(payload.id, ProjectSoftSummary, payload.data)
|
||
|
||
# ~~~动态环境描述-新增和修改~~~
|
||
@route.post('/dynamic_description/')
|
||
@transaction.atomic
|
||
def dynamic_description(self, payload: SoftSummarySchema):
|
||
self.create_or_modify_data_schemas(payload.id, ProjectDynamicDescription, payload.data)
|
||
|
||
@classmethod
|
||
def get_res_from_info(cls, project_obj: Project, model) -> list[dict] | None:
|
||
"""model: 当前一对一模型，直接获取结构化数据信息数组返回"""
|
||
qs = model.objects.filter(project=project_obj)
|
||
if qs.exists():
|
||
obj = qs.first()
|
||
ds_qs = obj.data_schemas.all()
|
||
data_list = [{
|
||
"type": item.type,
|
||
"content": item.content,
|
||
"fontnote": item.fontnote,
|
||
} for item in ds_qs]
|
||
return data_list
|
||
return None
|
||
|
||
# ~~~软件概述-获取到前端展示~~~
|
||
@route.get("/get_soft_summary/", response=list[DataSchema])
|
||
@transaction.atomic
|
||
def get_soft_summary(self, id: int):
|
||
project_obj = self.get_project_by_id(id)
|
||
data_list = self.get_res_from_info(project_obj, ProjectSoftSummary)
|
||
if data_list:
|
||
return ChenResponse(status=200, code=20000, data=data_list)
|
||
return ChenResponse(status=200, code=20000, data=[])
|
||
|
||
# ~~~动态环境描述 - 获取展示~~~
|
||
@route.get("/dynamic_des/", response=list[DataSchema])
|
||
@transaction.atomic
|
||
def get_dynamic_des(self, id: int):
|
||
project_obj = self.get_project_by_id(id)
|
||
data_list = self.get_res_from_info(project_obj, ProjectDynamicDescription)
|
||
if data_list:
|
||
return ChenResponse(status=200, code=20000, data=data_list)
|
||
return ChenResponse(status=200, code=20000, data=[])
|
||
|
||
# ~~~接口图新增或修改~~~
|
||
@route.post("/interface_image/")
|
||
@transaction.atomic
|
||
def post_interface_image(self, id: int, dataSchema: DataSchema):
|
||
project_obj = self.get_project_by_id(id)
|
||
image_qs = StuctSortData.objects.filter(project=project_obj)
|
||
if image_qs.exists():
|
||
image_qs.delete()
|
||
self.bulk_create_data_schemas(project_obj, [dataSchema])
|
||
|
||
# ~~~接口图-获取数据~~~
|
||
@route.get("/get_interface_image/", response=DataSchema)
|
||
@transaction.atomic
|
||
def get_interface_image(self, id: int):
|
||
project_obj = self.get_project_by_id(id)
|
||
image_qs = StuctSortData.objects.filter(project=project_obj)
|
||
if image_qs.exists():
|
||
# 如果存在则返回数据
|
||
image_obj = image_qs.first()
|
||
return ChenResponse(status=200, code=25001, data={
|
||
"type": image_obj.type,
|
||
"content": image_obj.content,
|
||
"fontnote": image_obj.fontnote,
|
||
})
|
||
return ChenResponse(status=200, code=25002, data=None)
|
||
|
||
# 动态返回是哪个模型
|
||
@classmethod
|
||
def get_model_from_category(cls, category: str):
|
||
mapDict = {
|
||
'静态软件项': StaticSoftItem,
|
||
'静态硬件项': StaticSoftHardware,
|
||
'动态软件项': DynamicSoftTable,
|
||
'动态硬件项': DynamicHardwareTable,
|
||
'测评数据': EvaluateData
|
||
}
|
||
return mapDict[category]
|
||
|
||
# ~~~静态软件项、静态硬件项、动态软件项、动态硬件项 - 获取~~~
|
||
@route.get("/get_static_dynamic_items/")
|
||
def get_static_dynamic_items(self, id: int, category: str):
|
||
project_obj = self.get_project_by_id(id)
|
||
item_qs = self.get_model_from_category(category).objects.filter(project=project_obj)
|
||
if item_qs.exists():
|
||
item_obj = item_qs.first()
|
||
return ChenResponse(status=200, code=25001, data={"table": item_obj.table, "fontnote": item_obj.fontnote})
|
||
return ChenResponse(status=200, code=25002, data=None)
|
||
|
||
# ~~~静态软件项、静态硬件项、动态软件项、动态硬件项 - 新增或修改~~~
|
||
@route.post("/post_static_dynamic_item/")
|
||
@transaction.atomic
|
||
def post_static_dynamic_item(self, data: StaticDynamicData):
|
||
project_obj = self.get_project_by_id(data.id)
|
||
model = self.get_model_from_category(data.category)
|
||
item_qs = model.objects.filter(project=project_obj)
|
||
if item_qs.exists():
|
||
# 如果存在则修改
|
||
item_qs.delete()
|
||
model.objects.create(project=project_obj, table=data.table, fontnote=data.fontnote)
|
||
|
||
# ~~~环境差异性分析 - 获取~~~
|
||
@route.get("/get_env_analysis/")
|
||
@transaction.atomic
|
||
def get_env_analysis(self, id: int):
|
||
project_obj = self.get_project_by_id(id)
|
||
qs = EnvAnalysis.objects.filter(project=project_obj)
|
||
if qs.exists():
|
||
obj = qs.first()
|
||
return ChenResponse(status=200, code=25001, data={"table": obj.table, "fontnote": obj.fontnote, "description": obj.description})
|
||
return ChenResponse(status=200, code=25002, data=None)
|
||
|
||
# ~~~环境差异性分析 - 新增和修改~~~
|
||
@route.post("/post_env_analysis/")
|
||
@transaction.atomic
|
||
def post_env_analysis(self, data: EnvAnalysisSchema):
|
||
project_obj = self.get_project_by_id(data.id)
|
||
qs = EnvAnalysis.objects.filter(project=project_obj)
|
||
if qs.exists():
|
||
qs.delete()
|
||
EnvAnalysis.objects.create(project=project_obj, table=data.table, fontnote=data.fontnote, description=data.description)
|