93 lines
3.3 KiB
Python
93 lines
3.3 KiB
Python
|
from django.shortcuts import get_object_or_404
|
|||
|
|
|
|||
|
|
# 1.1)这个函数参数1：request、参数2：schema或schema.dict()、参数3：ORM模型
|
|||
|
|
def create(request, data, model):
|
|||
|
|
if not isinstance(data, dict):
|
|||
|
|
data = data.dict()
|
|||
|
|
query_set = model.objects.create(**data)
|
|||
|
|
return query_set
|
|||
|
|
|
|||
|
|
# 1.2)新增快捷函数，直接填入data
|
|||
|
|
def createWithOutRequestParam(data, model):
|
|||
|
|
if not isinstance(data, dict):
|
|||
|
|
data = data.dict()
|
|||
|
|
query_set = model.objects.create(**data)
|
|||
|
|
return query_set
|
|||
|
|
|
|||
|
|
# 2.1)更新便捷函数，特别注意参数0为request后面有用，参数1：id，参数2：schema，参数3：ORM模型
|
|||
|
|
def update(request, id, data, model):
|
|||
|
|
dict_data = data.dict()
|
|||
|
|
instance = get_object_or_404(model, id=id)
|
|||
|
|
for attr, value in dict_data.items():
|
|||
|
|
setattr(instance, attr, value)
|
|||
|
|
instance.save()
|
|||
|
|
return instance
|
|||
|
|
|
|||
|
|
# 2.2)更新便捷函数-无request参数 -> data参数为schema对象
|
|||
|
|
def updateWithoutRequestParam(id, data, model):
|
|||
|
|
dict_data = data.model_dump(exclude_none=True)
|
|||
|
|
instance = get_object_or_404(model, id=id)
|
|||
|
|
for attr, value in dict_data.items():
|
|||
|
|
if attr != 'id': # 不对id更新
|
|||
|
|
setattr(instance, attr, value)
|
|||
|
|
instance.save()
|
|||
|
|
return instance
|
|||
|
|
|
|||
|
|
# 多个id删除便捷函数，参数1:ids数组，参数2：ORM模型
|
|||
|
|
def multi_delete(ids, model):
|
|||
|
|
for item in ids:
|
|||
|
|
# 删除多对多关系 - project
|
|||
|
|
instance = get_object_or_404(model, pk=item)
|
|||
|
|
instance.delete()
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
# project删除，对应problem问题单多对多关系也要删除
|
|||
|
|
def multi_delete_project(ids, model):
|
|||
|
|
idents = []
|
|||
|
|
for item in ids:
|
|||
|
|
instance = get_object_or_404(model, pk=item)
|
|||
|
|
# (注意：project所属problem全部删除，且关联关系删除)
|
|||
|
|
for problem in instance.projField.all():
|
|||
|
|
problem.case.clear()
|
|||
|
|
idents.append(instance.ident)
|
|||
|
|
instance.delete()
|
|||
|
|
return idents
|
|||
|
|
|
|||
|
|
# testDemand多个id删除便捷函数，参数1:ids数组，参数2：ORM模型
|
|||
|
|
def multi_delete_testDemand(ids, model):
|
|||
|
|
for item in ids:
|
|||
|
|
instance = get_object_or_404(model, pk=item)
|
|||
|
|
# （多对多删除）case下面的problem关联删除
|
|||
|
|
for case in instance.tcField.all():
|
|||
|
|
case.caseField.clear()
|
|||
|
|
instance.delete()
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
# dut的删除，需要多对多case-problem
|
|||
|
|
def multi_delete_dut(ids, model):
|
|||
|
|
for item in ids:
|
|||
|
|
instance = get_object_or_404(model, pk=item)
|
|||
|
|
# （多对多删除）case下面的problem关联删除
|
|||
|
|
for case in instance.ducField.all():
|
|||
|
|
case.caseField.clear()
|
|||
|
|
instance.delete()
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
# design的多个id删除便捷函数，参数1:ids数组，参数2：ORM模型
|
|||
|
|
def multi_delete_design(ids, model):
|
|||
|
|
for item in ids:
|
|||
|
|
instance = get_object_or_404(model, pk=item)
|
|||
|
|
# （多对多删除）case下面problem关联删除
|
|||
|
|
for case in instance.dcField.all():
|
|||
|
|
case.caseField.clear()
|
|||
|
|
instance.delete()
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
# 由于case有多对多关系，所以单独提出来删除
|
|||
|
|
def multi_delete_case(ids, model):
|
|||
|
|
for item in ids:
|
|||
|
|
instance = get_object_or_404(model, pk=item)
|
|||
|
|
instance.caseField.clear()
|
|||
|
|
instance.delete()
|
|||
|
|
pass
|