49 lines
2.0 KiB
Python
49 lines
2.0 KiB
Python
from copy import deepcopy
|
||
from django.shortcuts import get_object_or_404
|
||
from apps.project.models import Project
|
||
|
||
def demand_copy_to_design(project_id: int, demand_key: str, design_id: int, depth: bool = False):
|
||
"""注意传入项目id，测试项是key，设计需求是id"""
|
||
project_qs = get_object_or_404(Project, id=project_id)
|
||
design = project_qs.psField.filter(id=design_id).first()
|
||
demand = project_qs.ptField.filter(key=demand_key).first()
|
||
# 1.这里先要根据老的demand取出其步骤，并深度复制
|
||
origin_demand = deepcopy(demand)
|
||
demand.id = None
|
||
demand.title = '(复制)' + demand.title
|
||
demand.name = '(复制)' + demand.name
|
||
demand.dut = design.dut
|
||
demand.round = design.round
|
||
demand.design = design
|
||
demand.ident += '(复制)'
|
||
# 先查询目标design下面有多少个demand，然后key+1
|
||
demand.key = "".join([design.key, '-', str(design.dtField.count())])
|
||
demand.save()
|
||
for content_obj in origin_demand.testQField.all():
|
||
content_origin = deepcopy(content_obj)
|
||
content_obj.id = None
|
||
content_obj.testDemand = demand
|
||
content_obj.save()
|
||
for step_obj in content_origin.testStepField.all():
|
||
step_obj.testDemandContent = content_obj
|
||
step_obj.id = None
|
||
step_obj.save()
|
||
# 2.如果depth=True，则复制用例过去
|
||
if depth:
|
||
for case in origin_demand.tcField.all():
|
||
origin_case = deepcopy(case)
|
||
case.id = None
|
||
case.round = demand.round
|
||
case.dut = demand.dut
|
||
case.design = demand.design
|
||
case.test = demand
|
||
# 处理新case的key
|
||
case_key_lastkey = case.key.split("-")[-1]
|
||
case.key = "".join([demand.key, '-', case_key_lastkey])
|
||
case.save()
|
||
for case_step in origin_case.step.all():
|
||
case_step.id = None
|
||
case_step.case = case
|
||
case_step.save()
|
||
return demand.key
|