2016-12-02 22:24:19 +08:00
|
|
|
import os
|
|
|
|
|
import requests
|
|
|
|
|
|
|
|
|
|
from command import CommandRegistry
|
|
|
|
|
from apiclient import client as api
|
|
|
|
|
|
|
|
|
|
__registry__ = cr = CommandRegistry()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@cr.register('echo', '重复', '跟我念')
|
|
|
|
|
def echo(args_text, ctx_msg):
|
|
|
|
|
msg_type = ctx_msg.get('type')
|
|
|
|
|
if msg_type == 'group_message':
|
2016-12-29 23:45:34 +08:00
|
|
|
api.send_group_message(content=args_text, ctx_msg=ctx_msg)
|
|
|
|
|
elif msg_type == 'discuss_message':
|
|
|
|
|
api.send_discuss_message(content=args_text, ctx_msg=ctx_msg)
|
|
|
|
|
elif msg_type == 'friend_message':
|
|
|
|
|
api.send_friend_message(content=args_text, ctx_msg=ctx_msg)
|
2016-12-02 22:24:19 +08:00
|
|
|
|
|
|
|
|
|
|
|
|
|
@cr.register('chat', '聊天')
|
|
|
|
|
def chat(args_text, ctx_msg):
|
|
|
|
|
url = 'http://www.tuling123.com/openapi/api'
|
|
|
|
|
data = {
|
|
|
|
|
'key': os.environ.get('TURING123_API_KEY'),
|
|
|
|
|
'info': args_text
|
|
|
|
|
}
|
2016-12-29 23:45:34 +08:00
|
|
|
if ctx_msg.get('sender_uid'):
|
|
|
|
|
data['userid'] = ctx_msg.get('sender_uid')
|
|
|
|
|
elif ctx_msg.get('sender_id'):
|
|
|
|
|
data['userid'] = ctx_msg.get('sender_id')[-32:]
|
2016-12-02 22:24:19 +08:00
|
|
|
resp = requests.post(url, data=data)
|
|
|
|
|
if resp.status_code == 200:
|
|
|
|
|
json = resp.json()
|
|
|
|
|
if int(json.get('code', 0)) == 100000:
|
|
|
|
|
reply = json.get('text', '')
|
|
|
|
|
else:
|
|
|
|
|
# Is not text type
|
|
|
|
|
reply = '腊鸡图灵机器人返回了一堆奇怪的东西，就不发出来了'
|
|
|
|
|
else:
|
|
|
|
|
reply = '腊鸡图灵机器人出问题了，先不管他，过会儿再玩他'
|
|
|
|
|
echo(reply, ctx_msg)
|
