评论接入百度云 文本审核 API

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

"""
@author: yinzhuoqun
@site: http://zhuoqun.info/
@email: yin@zhuoqun.info
@time: 2019/11/20 1:21 AM
"""

from aip import AipImageCensor
from utils.json_to_object import json_to_object

BAIDU_APP_ID = '填自己申请的'
BAIDU_API_KEY = '填自己申请的'
BAIDU_SECRET_KEY = '填自己申请的'

labels_text = {
    1: "暴恐违禁",
    2: "色情内容",
    3: "政治敏感",
    4: "恶意推广",
    5: "低俗辱骂",
    6: "低质灌水"
}

spam_text = {
    0: "不违禁",
    1: "违禁",
    2: "人工复审"
}


class CheckText:
    def __init__(self, text):
        self.text = text
        self.app_id = BAIDU_APP_ID
        self.api_key = BAIDU_API_KEY
        self.secret_key = BAIDU_SECRET_KEY
        self.client = AipImageCensor(self.app_id, self.api_key, self.secret_key)

    def request_api(self):
        # print(self.text)
        return self.client.antiSpam(self.text)

    def response(self):
        res = self.request_api()
        if "error_msg" in res.keys():
            return {"status": 203, "msg": res["error_msg"]}
        else:
            check_status = res["result"]["spam"]
            if check_status == 0:
                return {"status": 201, "msg": "审核通过"}
            elif check_status == 1:
                labels = res["result"]["reject"]
                labels_id = [(review["label"]) for review in labels]
                reason = ",".join([labels_text[label_id] for label_id in labels_id]).strip(",")
                # print(reason)
                return {"status": 403, "msg": "审核不通过", "reason": reason}
            elif check_status == 2:
                labels = res["result"]["review"]
                labels_id = [(review["label"]) for review in labels]
                reason = ",".join([labels_text[label_id] for label_id in labels_id]).strip(",")
                # print(reason)
                return {"status": 403, "msg": "人工复审", "reason": reason}
            else:
                return {"status": 200, "msg": "人工复审", "reason": "百度异常"}

    def response_format(self):
        return json_to_object(self.response())


if __name__ == "__main__":
    text = "今晚弹药枪支吗傻x"
    # text = "傻x"
    c = CheckText(text)
    print(c.response_format())

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

"""
@author: yinzhuoqun
@site: http://zhuoqun.info/
@email: yin@zhuoqun.info
@time: 2019/11/20 12:02 AM
"""

from argparse import Namespace
import json


def json_to_object(data):
    """
    json 或 字典 转成 object
    :param data:
    :return:
    """
    if isinstance(data, dict):
        return json.loads(json.dumps(data), object_hook=lambda d: Namespace(**d))
    else:
        return json.loads(data, object_hook=lambda d: Namespace(**d))

>>> from django.utils.html import strip_tags
>>> 
>>> html = "<a>xieboke.net</a>"
>>> strip_tags(html)
'xieboke.net'
>>>