43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
import json
|
||
|
||
import requests
|
||
|
||
|
||
def text_detection(text):
|
||
"""
|
||
百度检验文字是否违规
|
||
:param text:
|
||
:return:
|
||
"""
|
||
url = "https://aip.baidubce.com/rest/2.0/solution/v1/text_censor/v2/user_defined?access_token=" + get_baidu_access_token()
|
||
payload = 'text=' + text
|
||
headers = {
|
||
'Content-Type': 'application/x-www-form-urlencoded',
|
||
'Accept': 'application/json'
|
||
}
|
||
|
||
response = requests.request("POST", url, headers=headers, data=payload)
|
||
content = str(response.text)
|
||
data = json.loads(content)
|
||
print(data)
|
||
|
||
# 安全地获取 conclusion 字段，如果不存在则返回默认值
|
||
conclusion = data.get('conclusion')
|
||
return conclusion
|
||
|
||
|
||
def get_baidu_access_token():
|
||
"""
|
||
使用 AK，SK 生成鉴权签名（Access Token），百度信息获取
|
||
:return: access_token，或是None(如果错误)
|
||
"""
|
||
API_KEY = "K7T5muBpNEdx1ebfc2YTag4W"
|
||
SECRET_KEY = "U3jkLq3BljN1q9YKnbFou2BY2167HZl6"
|
||
|
||
url = "https://aip.baidubce.com/oauth/2.0/token"
|
||
params = {"grant_type": "client_credentials", "client_id": API_KEY, "client_secret": SECRET_KEY}
|
||
return str(requests.post(url, params=params).json().get("access_token"))
|
||
|
||
text = "我草泥马"
|
||
result = text_detection(text)
|
||
print(result) |