Add xiaoice command

commands/ai.py

@@ -0,0 +1,46 @@
+import os
+import requests
+
+from command import CommandRegistry
+from commands import core
+from little_shit import get_source
+from apiclient import client as api
+
+__registry__ = cr = CommandRegistry()
+
+
+@cr.register('tuling123', 'chat', '聊天')
+def tuling123(args_text, ctx_msg, internal=False):
+    url = 'http://www.tuling123.com/openapi/api'
+    data = {
+        'key': os.environ.get('TURING123_API_KEY'),
+        'info': args_text,
+        'userid': get_source(ctx_msg)
+    }
+    resp = requests.post(url, data=data)
+    if resp.status_code == 200:
+        json = resp.json()
+        if internal:
+            return json
+        if int(json.get('code', 0)) == 100000:
+            reply = json.get('text', '')
+        else:
+            # Is not text type
+            reply = '腊鸡图灵机器人返回了一堆奇怪的东西，就不发出来了'
+    else:
+        if internal:
+            return None
+        reply = '腊鸡图灵机器人出问题了，先不管他，过会儿再玩他'
+    core.echo(reply, ctx_msg)
+
+
+@cr.register('xiaoice', '小冰')
+def xiaoice(args_text, ctx_msg, internal=False):
+    resp = api.wx_consult(account='xiaoice-ms', content=args_text)
+    if resp:
+        json = resp.json()
+        if json and json.get('reply'):
+            reply = json['reply']
+            core.echo(reply, ctx_msg, internal)
+            return reply
+    return None