164 lines
7.4 KiB
Python
164 lines
7.4 KiB
Python
from typing import Tuple, Any
|
||
from copy import deepcopy
|
||
from django.shortcuts import get_object_or_404
|
||
from apps.project.models import Project
|
||
from ninja.errors import HttpError
|
||
|
||
# 1. case被移动到某测试项下面
|
||
def case_move_to_test(project_id: int, case_key: str, demand_key: str) -> Tuple[str, Any]:
|
||
"""移动case到某个测试项下面，传入project_id，case的key，测试项的key，renturn -> 元组(旧case的key,新case的key)"""
|
||
same_root_flag = False
|
||
if '-'.join(case_key.split('-')[:-1]) == demand_key:
|
||
same_root_flag = True
|
||
# 判断是否移动到自己所属的demand的，直接返回不操作
|
||
if same_root_flag:
|
||
raise HttpError(500, message='无法移动到自己所属测试项里面')
|
||
project_qs = get_object_or_404(Project, id=project_id)
|
||
case = project_qs.pcField.filter(key=case_key).first()
|
||
demand_origin = case.test # 未变化之前case对应的测试项
|
||
demand = project_qs.ptField.filter(key=demand_key).first() # 新的demand对象
|
||
case.ident = demand.ident
|
||
case.test = demand
|
||
case.round = demand.round
|
||
case.dut = demand.dut
|
||
case.design = demand.design
|
||
# 查询被拖拽到的demand有多少用例，用于设置key
|
||
case.key = "".join([demand.key, '-', str(demand.tcField.count())])
|
||
case.save()
|
||
# 因为移动要删除之前demand的用例，所以需要重新设置key
|
||
index = 0
|
||
for c in demand_origin.tcField.all():
|
||
c.key = "".join([demand_origin.key, '-', str(index)])
|
||
c.save()
|
||
index += 1
|
||
return case_key, case.key
|
||
|
||
# 2.case复制到测试项中
|
||
def case_copy_to_test(project_id: int, case_key: str, demand_key: str) -> Tuple[str, Any]:
|
||
# 初始化内容
|
||
project_qs = get_object_or_404(Project, id=project_id)
|
||
case = project_qs.pcField.filter(key=case_key).first()
|
||
demand = project_qs.ptField.filter(key=demand_key).first()
|
||
origin_case = deepcopy(case)
|
||
case.id = None
|
||
case.ident = demand.ident
|
||
case.test = demand
|
||
case.round = demand.round
|
||
case.dut = demand.dut
|
||
case.design = demand.design
|
||
case.key = "".join([demand.key, '-', str(demand.tcField.count())])
|
||
case.save()
|
||
# 用例步骤也要复制一份过去
|
||
for case_step in origin_case.step.all():
|
||
case_step.id = None
|
||
case_step.case = case
|
||
case_step.save()
|
||
return case_key, case.key
|
||
|
||
# 3.（需要拆分）拖拽用例到用例的函数,TODO:必须拆分
|
||
def case_to_case_copy_or_move(project_id: int, drag_key: str, drop_key: str, move: bool, position: int):
|
||
"""待优化和上面一样：TODO:封装多个函数理清思路"""
|
||
project_qs = get_object_or_404(Project, id=project_id)
|
||
drag_case = project_qs.pcField.filter(key=drag_key).first()
|
||
drag_demand = drag_case.test
|
||
drop_case = project_qs.pcField.filter(key=drop_key).first()
|
||
drop_demand = drop_case.test
|
||
# 判断是否移动到同级
|
||
same_root_flag = False
|
||
if drag_key.split('-')[:-2] == drop_key.split('-')[:-2]:
|
||
same_root_flag = True
|
||
if move:
|
||
# 1.移动到同级demand - 只需要改变2个key
|
||
if same_root_flag:
|
||
case_list = list(drag_demand.tcField.all()) # 这是已经排序的
|
||
case_list.pop(case_list.index(drag_case))
|
||
# 移动出去后，查看现在drop_case索引
|
||
drop_case_index = case_list.index(drop_case)
|
||
if position == 0 or position == 1: # 1和0就是往下放，首先drop的位置是不变的
|
||
case_list.insert(drop_case_index + 1, drag_case)
|
||
elif position == -1:
|
||
case_list.insert(drop_case_index, drag_case)
|
||
# 已经排序好了就开始修改key
|
||
ca_index = 0
|
||
for ca in case_list:
|
||
ca.key = "".join([drag_demand.key, '-', str(ca_index)])
|
||
ca.save()
|
||
ca_index += 1
|
||
# 2.移动到非同级demand
|
||
else:
|
||
drag_case.ident = drop_demand.ident
|
||
drag_case.test = drop_demand
|
||
drag_case.round = drop_demand.round
|
||
drag_case.dut = drop_demand.dut
|
||
drag_case.design = drop_demand.design
|
||
drag_case.save() # 到这里drag_case已经放入了drop_demand
|
||
# 查询所有的drop_demand的所有用例
|
||
drop_case_list = list(drop_demand.tcField.all())
|
||
drop_case_list.pop(drop_case_list.index(drag_case))
|
||
drop_case_index = drop_case_list.index(drop_case)
|
||
if position == 0 or position == 1: # 1和0就是往下放，首先drop的位置是不变的
|
||
drop_case_list.insert(drop_case_index + 1, drag_case)
|
||
elif position == -1:
|
||
drop_case_list.insert(drop_case_index, drag_case)
|
||
ca_ind = 0
|
||
for ca in drop_case_list:
|
||
ca.key = "".join([drop_demand.key, '-', str(ca_ind)])
|
||
ca.save()
|
||
ca_ind += 1
|
||
# 因为移动要删除之前demand的用例，所以需要重新设置key
|
||
index = 0
|
||
for c in drag_demand.tcField.all():
|
||
c.key = "".join([drag_demand.key, '-', str(index)])
|
||
c.save()
|
||
index += 1
|
||
# 如果是复制
|
||
else:
|
||
origin_drag_case = deepcopy(drag_case) # 深度复制被拖拽的case
|
||
# 同一个测试项内用例的复制，还要看顺序
|
||
if same_root_flag:
|
||
drag_case.id = None
|
||
drag_case.save()
|
||
# 开始插入到被拖拽case上下
|
||
case_sx_list = list(drag_demand.tcField.all())
|
||
case_sx_list.pop(case_sx_list.index(drag_case))
|
||
drop_case_in = case_sx_list.index(drop_case)
|
||
if position == 0 or position == 1: # 1和0就是往下放，首先drop的位置是不变的
|
||
case_sx_list.insert(drop_case_in + 1, drag_case)
|
||
elif position == -1:
|
||
case_sx_list.insert(drop_case_in, drag_case)
|
||
ind = 0
|
||
for cas in case_sx_list:
|
||
cas.key = "".join([drag_demand.key, '-', str(ind)])
|
||
cas.save()
|
||
ind += 1
|
||
# 根据origin复制步骤
|
||
for c_step in origin_drag_case.step.all():
|
||
c_step.id = None
|
||
c_step.case = drag_case # 外键设置为新用例
|
||
c_step.save()
|
||
else:
|
||
drag_case.id = None
|
||
drag_case.ident = drop_demand.ident
|
||
drag_case.test = drop_demand
|
||
drag_case.round = drop_demand.round
|
||
drag_case.dut = drop_demand.dut
|
||
drag_case.design = drop_demand.design
|
||
drag_case.save()
|
||
cases_list = list(drop_demand.tcField.all())
|
||
cases_list.pop(cases_list.index(drag_case))
|
||
drop_case_idx = cases_list.index(drop_case)
|
||
if position == 0 or position == 1: # 1和0就是往下放，首先drop的位置是不变的
|
||
cases_list.insert(drop_case_idx + 1, drag_case)
|
||
elif position == -1:
|
||
cases_list.insert(drop_case_idx, drag_case)
|
||
cas_idx = 0
|
||
for cas in cases_list:
|
||
cas.key = "".join([drop_demand.key, '-', str(cas_idx)])
|
||
cas.save()
|
||
cas_idx += 1
|
||
# 复制用例步骤
|
||
for cs_step in origin_drag_case.step.all():
|
||
cs_step.id = None
|
||
cs_step.case = drag_case # 外键设置为新用例
|
||
cs_step.save()
|