init: build minimum viable version

bot/baidu/baidu_unit_bot.py

@@ -0,0 +1,26 @@
+# encoding:utf-8
+
+import json
+import requests
+from bot.bot import Bot
+
+
+class BaiduUnitBot(Bot):
+    def reply(self, query):
+        token = self.get_token()
+        url = 'https://aip.baidubce.com/rpc/2.0/unit/service/v3/chat?access_token=' + token
+        post_data = "{\"version\":\"3.0\",\"service_id\":\"S73177\",\"session_id\":\"\",\"log_id\":\"7758521\",\"skill_ids\":[\"1221886\"],\"request\":{\"terminal_id\":\"88888\",\"query\":\"" + query + "\", \"hyper_params\": {\"chat_custom_bot_profile\": 1}}}"
+        print(post_data)
+        headers = {'content-type': 'application/x-www-form-urlencoded'}
+        response = requests.post(url, data=post_data.encode(), headers=headers)
+        if response:
+            return response.json()['result']['context']['SYS_PRESUMED_HIST'][1]
+
+    def get_token(self):
+        access_key = '${YOUR_ACCESS_KEY}'
+        secret_key = '${YOUR_SECRET_KEY}'
+        host = 'https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=' + access_key + '&client_secret=' + secret_key
+        response = requests.get(host)
+        if response:
+            print(response.json())
+            return response.json()['access_token']

bot/bot.py

@@ -0,0 +1,13 @@
+"""
+Auto-replay chat robot abstract class
+"""
+
+
+class Bot(object):
+    def reply(self, query):
+        """
+        bot auto-reply content
+        :param req: received message
+        :return: reply content
+        """
+        raise NotImplementedError

bot/bot_factory.py

@@ -0,0 +1,16 @@
+"""
+channel factory
+"""
+
+from bot.baidu.baidu_unit_bot import BaiduUnitBot
+
+
+def create_bot(bot_type):
+    """
+    create a channel instance
+    :param channel_type: channel type code
+    :return: channel instance
+    """
+    if bot_type == 'baidu':
+        return BaiduUnitBot()
+    raise RuntimeError