import requests


def security_check(SECURITY_URL: str, auth_code: str, question: str, isRejection: bool, isRefusal: bool):
    '''
    SECURITY_URL: 拒识拒答url
    auth_code: token
    question: 被检测的字符串
    isRejection: 是否进行拒识检测
    isRefusal: 是否进行拒答检测
    threshold: 知识库检索最低阈值
    '''
    headers = {
        'content-type': "application/json",
        'authorization': auth_code,
    }
    security_json = {
        "query": question,
        "isRejection": isRejection,
        "isRefusal": isRefusal,
    }
    try:
        security_res = requests.post(SECURITY_URL, json=security_json, headers=headers)
        # {'code': 2, 'message': '拒识检测未通过!', 'result': False}
        security_res_json = security_res.json()
        if not security_res_json['result']:
            return {
                "result": False,
                "msg": security_res_json["message"]
            }
        elif security_res_json['result']:
            return {
                "result": True,
                "msg": security_res_json["message"]
            }
    except Exception as e:
        return {
            "result": False,
            "msg": f"error: {e}"
        }