100 lines
3.9 KiB
Python
100 lines
3.9 KiB
Python
|
import json
|
|||
|
|
import requests
|
|||
|
|
import time
|
|||
|
|
from core.authorize_tb import Authorize_Tb
|
|||
|
|
from utils import config_util as cfg
|
|||
|
|
from utils import util
|
|||
|
|
|
|||
|
|
def get_sentiment(cont):
|
|||
|
|
emotion = Emotion()
|
|||
|
|
answer = emotion.get_sentiment(cont)
|
|||
|
|
return answer
|
|||
|
|
|
|||
|
|
class Emotion:
|
|||
|
|
|
|||
|
|
def __init__(self):
|
|||
|
|
self.app_id = cfg.baidu_emotion_app_id
|
|||
|
|
self.authorize_tb = Authorize_Tb()
|
|||
|
|
|
|||
|
|
def get_sentiment(self, cont):
|
|||
|
|
token = self.__check_token()
|
|||
|
|
if token is None or token == 'expired':
|
|||
|
|
token_info = self.__get_token()
|
|||
|
|
if token_info is not None and token_info['access_token'] is not None:
|
|||
|
|
#转换过期时间
|
|||
|
|
updated_in_seconds = time.time()
|
|||
|
|
expires_timedelta = token_info['expires_in']
|
|||
|
|
expiry_timestamp_in_seconds = updated_in_seconds + expires_timedelta
|
|||
|
|
expiry_timestamp_in_milliseconds = expiry_timestamp_in_seconds * 1000
|
|||
|
|
if token == 'expired':
|
|||
|
|
self.authorize_tb.update_by_userid(self.app_id, token_info['access_token'], expiry_timestamp_in_milliseconds)
|
|||
|
|
else:
|
|||
|
|
self.authorize_tb.add(self.app_id, token_info['access_token'], expiry_timestamp_in_milliseconds)
|
|||
|
|
token = token_info['access_token']
|
|||
|
|
else:
|
|||
|
|
token = None
|
|||
|
|
|
|||
|
|
if token is not None:
|
|||
|
|
try:
|
|||
|
|
url=f"https://aip.baidubce.com/rpc/2.0/nlp/v1/sentiment_classify?access_token={token}"
|
|||
|
|
req = json.dumps({"text": cont})
|
|||
|
|
headers = {
|
|||
|
|
'Content-Type': 'application/json',
|
|||
|
|
'Accept': 'application/json'
|
|||
|
|
}
|
|||
|
|
r = requests.post(url, headers=headers, data=req)
|
|||
|
|
if r.status_code != 200:
|
|||
|
|
util.log(1, f"百度情感分析对接有误: {r.text}")
|
|||
|
|
return 0
|
|||
|
|
info = json.loads(r.text)
|
|||
|
|
if not self.has_field(info,'error_code'):
|
|||
|
|
return info['items'][0]['sentiment']
|
|||
|
|
else:
|
|||
|
|
util.log(1, f"百度情感分析对接有误： {info['error_msg']}")
|
|||
|
|
return 0
|
|||
|
|
except Exception as e:
|
|||
|
|
util.log(1, f"百度情感分析对接有误： {str(e)}")
|
|||
|
|
return 0
|
|||
|
|
else:
|
|||
|
|
return 0
|
|||
|
|
|
|||
|
|
def __check_token(self):
|
|||
|
|
self.authorize_tb.init_tb()
|
|||
|
|
info = self.authorize_tb.find_by_userid(self.app_id)
|
|||
|
|
if info is not None:
|
|||
|
|
if info[1] >= int(time.time())*1000:
|
|||
|
|
return info[0]
|
|||
|
|
else:
|
|||
|
|
return 'expired'
|
|||
|
|
else:
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
def __get_token(self):
|
|||
|
|
try:
|
|||
|
|
url=f"https://aip.baidubce.com/oauth/2.0/token?client_id={cfg.baidu_emotion_api_key}&client_secret={cfg.baidu_emotion_secret_key}&grant_type=client_credentials"
|
|||
|
|
headers = {'Content-Type':'application/json;charset=UTF-8'}
|
|||
|
|
r = requests.post(url, headers=headers)
|
|||
|
|
if r.status_code != 200:
|
|||
|
|
info = json.loads(r.text)
|
|||
|
|
if info["error"] == "invalid_client":
|
|||
|
|
util.log(1, f"请检查baidu_emotion_api_key")
|
|||
|
|
else:
|
|||
|
|
util.log(1, f"请检查baidu_emotion_secret_key")
|
|||
|
|
return None
|
|||
|
|
info = json.loads(r.text)
|
|||
|
|
if not self.has_field(info,'error_code'):
|
|||
|
|
return info
|
|||
|
|
else:
|
|||
|
|
util.log(1, f"百度情感分析对接有误： {info['error_msg']}")
|
|||
|
|
util.log(1, f"请检查baidu_emotion_api_key和baidu_emotion_secret_key")
|
|||
|
|
return None
|
|||
|
|
except Exception as e:
|
|||
|
|
util.log(1, f"百度情感分析有1误： {str(e)}")
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
|
|||
|
|
def has_field(self, array, field):
|
|||
|
|
return any(field in item for item in array)
|
|||
|
|
|
|||
|
|
|