utils.py 文件源码

def add_fail_count_to_tasklist(tasklist):
    tid_list=[i.tid for i in  tasklist]
    buff={}
    for i in tasklist:
        buff[i.tid]=i
    if len(tid_list)==0:
        pass
        return True
    for i in tasklist:
        i.count=0
    if len(tid_list)==1:
        tid_list*=2
    tid_tuple=tuple(tid_list)
   # _query= Runlog.objects.filter(tid__in=tid_list).query
    #_query.group_by=["tid"]
    #_buff=QuerySet(query=_query,model=Runlog)
    #_result=_buff.annotate(count=Count("rid"))
    cursor=connection.cursor()
    cursor.execute("SELECT `task_runlog`.`rid`, `task_runlog`.`tid`,COUNT(`task_runlog`.`rid`) AS `count` FROM `task_runlog` WHERE `task_runlog`.`tid` IN %s GROUP BY tid;",(tid_tuple,))
    _result=cursor.fetchall()
    for _rid,_tid,_count  in _result:
        buff[_tid].count=int(_count)
    return True