from common.db import delete_data, find_data


def delete_rid_cascade_scripts(rid):
    """
    级联删除与评价ID有关的所有数据
    1.删除评价报告
    2.删除评价结果
    3.删除综合评价填报
    4.删除评价记录
    5.删除风险数据
    Parameters:
        rid: 评价ID
    Returns:
        res: True 执行结果
    """

    delete_data('tfse', '评价', '报告数据', {"评价ID": rid})
    delete_data('tfse', '评价', '评价结果', {"评价ID": rid})
    delete_data('tfse', '评价', '综合评价填报', {"评价ID": rid})
    delete_data('tfse', '评价', '评价记录', {"评价ID": rid})
    delete_data('tfse', '评价', '风险数据', {"评价ID": rid})

    return True


def delete_cid_cascade_scripts(cid):
    """
    级联删除uid有关的所有数据
    Parameters:
        cid: 企业ID
    Returns:
        res: True 执行结果
    """
    delete_data('tfse', '用户', '用户信息', {"企业ID": cid})
    delete_data('tfse', '企业', '指标明细', {"企业ID": cid})
    delete_data('tfse', '企业', '综合评价分析', {"企业ID": cid})
    delete_data('tfse', '企业', '公司基本信息', {"企业ID": cid})
    delete_data('tfse', '企业', '财务要素分析', {"企业ID": cid})
    delete_data('tfse', '企业', '财务要素分析', {"企业ID": cid})

    rids = find_data('tfse', '评价', '评价记录', {"企业ID": cid})
    if rids:
        for rid in rids:
            delete_rid_cascade_scripts(rid['评价ID'])

    return True