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()
|