44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
|
from django.db.models import QuerySet
|
|||
|
|
|
|||
|
|
from apps.project.models import Design, TestDemand
|
|||
|
|
from typing import Union
|
|||
|
|
|
|||
|
|
def DesignDrapAtoB(a: Design,
|
|||
|
|
b: Design,
|
|||
|
|
origin_qs: QuerySet[Design, Design],
|
|||
|
|
pos: Union[-1 | 1]) -> str:
|
|||
|
|
"""该函数传入拖拽design和释放到的design，然后更改排序，完成key的重新设置"""
|
|||
|
|
# 判断是移动到b前面还是后面
|
|||
|
|
list_qs = list(origin_qs)
|
|||
|
|
list_qs.remove(a)
|
|||
|
|
b_index = list_qs.index(b)
|
|||
|
|
if pos == -1:
|
|||
|
|
list_qs.insert(b_index, a)
|
|||
|
|
elif pos == 1:
|
|||
|
|
list_qs.insert(b_index + 1, a)
|
|||
|
|
# 重新完成排序后调整key
|
|||
|
|
prefix = "".join([a.dut.key, "-"])
|
|||
|
|
for index, obj in enumerate(list_qs):
|
|||
|
|
obj.key = "".join([prefix, str(index)])
|
|||
|
|
# 需要测试项调整key
|
|||
|
|
designConvertDemadnKey(obj)
|
|||
|
|
obj.save()
|
|||
|
|
return a.key
|
|||
|
|
|
|||
|
|
def designConvertDemadnKey(desgin_obj: Design):
|
|||
|
|
"""传入Design对象，集体修改demand和case的key"""
|
|||
|
|
for demand in desgin_obj.dtField.all():
|
|||
|
|
design_key = desgin_obj.key
|
|||
|
|
demand_last_key = demand.key.split("-")[-1]
|
|||
|
|
demand.key = "-".join([design_key, demand_last_key])
|
|||
|
|
demandConvertCaseKey(demand)
|
|||
|
|
demand.save()
|
|||
|
|
|
|||
|
|
def demandConvertCaseKey(demand_obj:TestDemand):
|
|||
|
|
"""传入Demand对象集体修改case的key"""
|
|||
|
|
for case in demand_obj.tcField.all():
|
|||
|
|
demand_key = demand_obj.key
|
|||
|
|
case_last_key = case.key.split("-")[-1]
|
|||
|
|
case.key = "-".join([demand_key, case_last_key])
|
|||
|
|
case.save()
|