23 |
23 | 
90 | 
792 |
20 | #####################
21 | # WeChat Bot Server #
22 | #####################
23 |
24 |
25 | ## APIs
26 |
27 | - /qrcode
28 | ------------------------
29 | @brief login qrcode
30 | @method get
31 | @return raw image
32 | ------------------------
33 |
34 | - /runtime
35 | -----------------------------
36 | @brief basic info
37 | @method get
38 | @return application/json
39 | {
40 | 'ret': 0,
41 | 'runtime': '',
42 | 'total_size': '',
43 | 'db_size': '',
44 | 'msg_count': '',
45 | 'image_count': '',
46 | 'voice_count': '',
47 | 'video_count': '',
48 | }
49 | -----------------------------
50 |
51 | - /group_list
52 | ------------------------
53 | @brief list groups
54 | @method get
55 | ------------------------
56 |
57 | - /group_member_list/<g_id>
58 | -----------------------------
59 | @brief list group member
60 | @method get
61 | @param g_id String
62 | -----------------------------
63 |
64 | - /group_chat_log/<g_name>
65 | -------------------------------
66 | @brief list group chat log
67 | @method get
68 | @param g_name String
69 | -------------------------------
70 |
71 | - /upload
72 | -------------------------------
73 | @brief upload a file
74 | @method get/post
75 | @return application/json
76 | {
77 | 'ret': 0,
78 | 'msg': '',
79 | }
80 | -------------------------------
81 |
82 | - /send_msg/<to>/<msg>
83 | -------------------------------------------
84 | @brief send message to user or gourp
85 | @method get
86 | @param to: String, user id or group id
87 | @param msg: String, words
88 | @return application/json
89 | {
90 | 'ret': 0,
91 | }
92 | -------------------------------------------
93 |
94 | - /send_img/<to>/<img>
95 | -------------------------------------------
96 | @brief send image to user or gourp
97 | @method get
98 | @param to: String, user id or group id
99 | @param img: String, image file name
100 | @return application/json
101 | {
102 | 'ret': 0,
103 | }
104 | -------------------------------------------
105 |
106 | - /send_emot/<to>/<emot>
107 | -------------------------------------------
108 | @brief send emotion to user or gourp
109 | @method get
110 | @param to: String, user id or group id
111 | @param emot: String, emotion file name
112 | @return application/json
113 | {
114 | 'ret': 0,
115 | }
116 | -------------------------------------------
117 |
118 | - /send_file/<to>/<file>
119 | -------------------------------------------
120 | @brief send emotion to user or gourp
121 | @method get
122 | @param to: String, user id or group id
123 | @param file: String, file name
124 | @return application/json
125 | {
126 | 'ret': 0,
127 | }
128 | -------------------------------------------
129 |
130 | - /mass_send_msg
131 | -------------------------------------------
132 | @brief send text to mass users or gourps
133 | @method post
134 | @param application/json
135 | {
136 | 'to_list': [
137 | 'group_id',
138 | ...
139 | ],
140 | 'msg': '',
141 | }
142 | @return application/json
143 | {
144 | 'ret': 0,
145 | 'unsend_list': [],
146 | }
147 | -------------------------------------------
148 |
149 | - /mass_send_img
150 | -------------------------------------------
151 | @brief send image to mass users or gourps
152 | @method post
153 | @param application/json
154 | {
155 | 'to_list': [
156 | 'group_id',
157 | ...
158 | ],
159 | 'msg': '',
160 | }
161 | @return application/json
162 | {
163 | 'ret': 0,
164 | 'unsend_list': [],
165 | }
166 | -------------------------------------------
167 |
168 | - /mass_send_emot
169 | -------------------------------------------
170 | @brief send emoticon to mass users or gourps
171 | @method post
172 | @param application/json
173 | {
174 | 'to_list': [
175 | 'group_id',
176 | ...
177 | ],
178 | 'msg': '',
179 | }
180 | @return application/json
181 | {
182 | 'ret': 0,
183 | 'unsend_list': [],
184 | }
185 | -------------------------------------------
186 |
187 | - /mass_send_file
188 | -------------------------------------------
189 | @brief send file to mass users or gourps
190 | @method post
191 | @param application/json
192 | {
193 | 'to_list': [
194 | 'group_id',
195 | ...
196 | ],
197 | 'msg': '',
198 | }
199 | @return application/json
200 | {
201 | 'ret': 0,
202 | 'unsend_list': [],
203 | }
204 | -------------------------------------------
205 |
206 | ' + str(wechat) + '\n\n-----\nLogs:\n-----\n\n' + ''.join(log_file.readlines()[-100:]) + '' 274 | sg.send_mail(subject, mail_content, 'text/html') 275 | log_file.close() 276 | 277 | if wechat.exit_code == 0: 278 | echo(Constant.MAIN_RESTART) 279 | else: 280 | # kill process 281 | os.system(Constant.LOG_MSG_KILL_PROCESS % os.getpid()) 282 | -------------------------------------------------------------------------------- /wxbot_project_py2.7/wx_handler/__init__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # coding: utf-8 3 | 4 | from wechat_msg_processor import WeChatMsgProcessor 5 | from bot import Bot 6 | from sendgrid_mail import SGMail -------------------------------------------------------------------------------- /wxbot_project_py2.7/wx_handler/bot.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # coding: utf-8 3 | 4 | #=================================================== 5 | from wechat.utils import * 6 | from config import Constant 7 | #--------------------------------------------------- 8 | import random, time, json 9 | #=================================================== 10 | 11 | 12 | class Bot(object): 13 | 14 | def __init__(self): 15 | self.emoticons = Constant.EMOTICON 16 | self.gifs = [] 17 | self.last_time = time.time() 18 | 19 | def time_schedule(self): 20 | r = '' 21 | now = time.time() 22 | if int(now - self.last_time) > 3600: 23 | self.last_time = now 24 | url_latest = Constant.BOT_ZHIHU_URL_LATEST 25 | url_daily = Constant.BOT_ZHIHU_URL_DAILY 26 | data = get(url_latest) 27 | j = json.loads(data) 28 | story = j['stories'][random.randint(0, len(j['stories'])-1)] 29 | r = story['title'] + '\n' + url_daily + str(story['id']) 30 | return r.encode('utf-8') 31 | 32 | def reply(self, text): 33 | APIKEY = Constant.BOT_TULING_API_KEY 34 | api_url = Constant.BOT_TULING_API_URL % (APIKEY, text, '12345678') 35 | r = json.loads(get(api_url)) 36 | if r.get('code') == 100000 and r.get('text') != Constant.BOT_TULING_BOT_REPLY: 37 | p = random.randint(1, 10) 38 | if p > 3: 39 | return r['text'] 40 | elif p > 1: 41 | # send emoji 42 | if random.randint(1, 10) > 5: 43 | n = random.randint(0, len(self.emoticons)-1) 44 | m = random.randint(1, 3) 45 | reply = self.emoticons[n].encode('utf-8') * m 46 | return reply 47 | return '' 48 | -------------------------------------------------------------------------------- /wxbot_project_py2.7/wx_handler/sendgrid_mail.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # coding: utf-8 3 | 4 | #=================================================== 5 | import sendgrid 6 | from sendgrid.helpers.mail import * 7 | #=================================================== 8 | 9 | class SGMail(object): 10 | 11 | def __init__(self, apikey, from_email, to_email): 12 | self.sg = sendgrid.SendGridAPIClient(apikey=apikey) 13 | self.from_email = Email(from_email) 14 | self.to_email = Email(to_email) 15 | 16 | def send_mail(self, subject, plain_content, type='text/plain'): 17 | content = Content(type, plain_content) 18 | mail = Mail(self.from_email, subject, self.to_email, content) 19 | response = self.sg.client.mail.send.post(request_body=mail.get()) 20 | return response.status_code == 202 21 | -------------------------------------------------------------------------------- /wxbot_project_py2.7/wx_handler/wechat_msg_processor.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # coding: utf-8 3 | 4 | #=================================================== 5 | from wechat.utils import * 6 | from config import ConfigManager 7 | from config import Constant 8 | from config import Log 9 | #--------------------------------------------------- 10 | import os 11 | import time 12 | import json 13 | import re 14 | #=================================================== 15 | 16 | 17 | class WeChatMsgProcessor(object): 18 | """ 19 | Process fetched data 20 | """ 21 | 22 | def __init__(self, db): 23 | self.db = db 24 | self.wechat = None # recieve `WeChat` class instance 25 | # for call some wechat apis 26 | 27 | # read config 28 | cm = ConfigManager() 29 | [self.upload_dir, self.data_dir, self.log_dir] = cm.setup_database() 30 | 31 | def clean_db(self): 32 | """ 33 | @brief clean database, delete table & create table 34 | """ 35 | self.db.delete_table(Constant.TABLE_GROUP_LIST()) 36 | self.db.delete_table(Constant.TABLE_GROUP_USER_LIST()) 37 | self.db.create_table(Constant.TABLE_GROUP_MSG_LOG, Constant.TABLE_GROUP_MSG_LOG_COL) 38 | self.db.create_table(Constant.TABLE_GROUP_LIST(), Constant.TABLE_GROUP_LIST_COL) 39 | self.db.create_table(Constant.TABLE_GROUP_USER_LIST(), Constant.TABLE_GROUP_USER_LIST_COL) 40 | self.db.create_table(Constant.TABLE_RECORD_ENTER_GROUP, Constant.TABLE_RECORD_ENTER_GROUP_COL) 41 | self.db.create_table(Constant.TABLE_RECORD_RENAME_GROUP, Constant.TABLE_RECORD_RENAME_GROUP_COL) 42 | 43 | def handle_wxsync(self, msg): 44 | """ 45 | @brief Recieve webwxsync message, saved into json 46 | @param msg Dict: webwxsync msg 47 | """ 48 | fn = time.strftime(Constant.LOG_MSG_FILE, time.localtime()) 49 | save_json(fn, msg, self.log_dir, 'a+') 50 | 51 | def handle_group_list(self, group_list): 52 | """ 53 | @brief handle group list & saved in DB 54 | @param group_list Array 55 | """ 56 | fn = Constant.LOG_MSG_GROUP_LIST_FILE 57 | save_json(fn, group_list, self.data_dir) 58 | cols = [( 59 | g['NickName'], 60 | g['UserName'], 61 | g['OwnerUin'], 62 | g['MemberCount'], 63 | g['HeadImgUrl'] 64 | ) for g in group_list] 65 | self.db.insertmany(Constant.TABLE_GROUP_LIST(), cols) 66 | 67 | def handle_group_member_list(self, group_id, member_list): 68 | """ 69 | @brief handle group member list & saved in DB 70 | @param member_list Array 71 | """ 72 | fn = group_id + '.json' 73 | save_json(fn, member_list, self.data_dir) 74 | cols = [( 75 | group_id, 76 | m['UserName'], 77 | m['NickName'], 78 | m['DisplayName'], 79 | m['AttrStatus'] 80 | ) for m in member_list] 81 | self.db.insertmany(Constant.TABLE_GROUP_USER_LIST(), cols) 82 | 83 | def handle_group_list_change(self, new_group): 84 | """ 85 | @brief handle adding a new group & saved in DB 86 | @param new_group Dict 87 | """ 88 | self.handle_group_list([new_group]) 89 | 90 | def handle_group_member_change(self, group_id, member_list): 91 | """ 92 | @brief handle group member changes & saved in DB 93 | @param group_id Dict 94 | @param member_list Dict 95 | """ 96 | self.db.delete(Constant.TABLE_GROUP_USER_LIST(), "RoomID", group_id) 97 | self.handle_group_member_list(group_id, member_list) 98 | 99 | def handle_group_msg(self, msg): 100 | """ 101 | @brief Recieve group messages 102 | @param msg Dict: packaged msg 103 | """ 104 | # rename media files 105 | for k in ['image', 'video', 'voice']: 106 | if msg[k]: 107 | t = time.localtime(float(msg['timestamp'])) 108 | time_str = time.strftime("%Y%m%d%H%M%S", t) 109 | # format: 时间_消息ID_群名 110 | file_name = '/%s_%s_%s.' % (time_str, msg['msg_id'], msg['group_name']) 111 | new_name = re.sub(r'\/\w+\_\d+\.', file_name, msg[k]) 112 | Log.debug('rename file to %s' % new_name) 113 | os.rename(msg[k], new_name) 114 | msg[k] = new_name 115 | 116 | if msg['msg_type'] == 10000: 117 | # record member enter in group 118 | m = re.search(r'邀请(.+)加入了群聊', msg['sys_notif']) 119 | if m: 120 | name = m.group(1) 121 | col_enter_group = ( 122 | msg['msg_id'], 123 | msg['group_name'], 124 | msg['from_user_name'], 125 | msg['to_user_name'], 126 | name, 127 | msg['time'], 128 | ) 129 | self.db.insert(Constant.TABLE_RECORD_ENTER_GROUP, col_enter_group) 130 | 131 | # record rename group 132 | n = re.search(r'(.+)修改群名为“(.+)”', msg['sys_notif']) 133 | if n: 134 | people = n.group(1) 135 | to_name = n.group(2) 136 | col_rename_group = ( 137 | msg['msg_id'], 138 | msg['group_name'], 139 | to_name, 140 | people, 141 | msg['time'], 142 | ) 143 | self.db.insert(Constant.TABLE_RECORD_RENAME_GROUP, col_rename_group) 144 | 145 | # upadte group in GroupList 146 | for g in self.wechat.GroupList: 147 | if g['UserName'] == msg['from_user_name']: 148 | g['NickName'] = to_name 149 | break 150 | 151 | # normal group message 152 | col = ( 153 | msg['msg_id'], 154 | msg['group_owner_uin'], 155 | msg['group_name'], 156 | msg['group_count'], 157 | msg['from_user_name'], 158 | msg['to_user_name'], 159 | msg['user_attrstatus'], 160 | msg['user_display_name'], 161 | msg['user_nickname'], 162 | msg['msg_type'], 163 | msg['emoticon'], 164 | msg['text'], 165 | msg['image'], 166 | msg['video'], 167 | msg['voice'], 168 | msg['link'], 169 | msg['namecard'], 170 | msg['location'], 171 | msg['recall_msg_id'], 172 | msg['sys_notif'], 173 | msg['time'], 174 | msg['timestamp'] 175 | ) 176 | self.db.insert(Constant.TABLE_GROUP_MSG_LOG, col) 177 | 178 | text = msg['text'] 179 | if text and text[0] == '@': 180 | n = trans_coding(text).find(u'\u2005') 181 | name = trans_coding(text)[1:n].encode('utf-8') 182 | if name in [self.wechat.User['NickName'], self.wechat.User['RemarkName']]: 183 | self.handle_command(trans_coding(text)[n+1:].encode('utf-8'), msg) 184 | 185 | def handle_user_msg(self, msg): 186 | """ 187 | @brief Recieve personal messages 188 | @param msg Dict 189 | """ 190 | wechat = self.wechat 191 | 192 | text = trans_coding(msg['text']).encode('utf-8') 193 | uid = msg['raw_msg']['FromUserName'] 194 | 195 | if text == 'test_revoke': # 撤回消息测试 196 | dic = wechat.webwxsendmsg('这条消息将被撤回', uid) 197 | wechat.revoke_msg(dic['MsgID'], uid, dic['LocalID']) 198 | elif text == 'reply': 199 | wechat.send_text(uid, '自动回复') 200 | 201 | 202 | def handle_command(self, cmd, msg): 203 | """ 204 | @brief handle msg of `@yourself cmd` 205 | @param cmd String 206 | @param msg Dict 207 | """ 208 | wechat = self.wechat 209 | g_id = '' 210 | for g in wechat.GroupList: 211 | if g['NickName'] == msg['group_name']: 212 | g_id = g['UserName'] 213 | 214 | cmd = cmd.strip() 215 | if cmd == 'runtime': 216 | wechat.send_text(g_id, wechat.get_run_time()) 217 | elif cmd == 'test_sendimg': 218 | wechat.send_img(g_id, 'test/emotion/7.gif') 219 | elif cmd == 'test_sendfile': 220 | wechat.send_file(g_id, 'test/Data/upload/shake.wav') 221 | elif cmd == 'test_bot': 222 | # reply bot 223 | # --------- 224 | if wechat.bot: 225 | r = wechat.bot.reply(cmd) 226 | if r: 227 | wechat.send_text(g_id, r) 228 | else: 229 | pass 230 | elif cmd == 'test_emot': 231 | img_name = [ 232 | '0.jpg', '1.jpeg', '2.gif', '3.jpg', '4.jpeg', 233 | '5.gif', '6.gif', '7.gif', '8.jpg', '9.jpg' 234 | ] 235 | name = img_name[int(time.time()) % 10] 236 | emot_path = os.path.join('test/emotion/', name) 237 | wechat.send_emot(g_id, emot_path) 238 | else: 239 | pass 240 | 241 | def check_schedule_task(self): 242 | # update group member list at 00:00 am every morning 243 | t = time.localtime() 244 | if t.tm_hour == 0 and t.tm_min <= 1: 245 | # update group member 246 | Log.debug('update group member list everyday') 247 | self.db.delete_table(Constant.TABLE_GROUP_LIST()) 248 | self.db.delete_table(Constant.TABLE_GROUP_USER_LIST()) 249 | self.db.create_table(Constant.TABLE_GROUP_LIST(), Constant.TABLE_GROUP_LIST_COL) 250 | self.db.create_table(Constant.TABLE_GROUP_USER_LIST(), Constant.TABLE_GROUP_USER_LIST_COL) 251 | self.wechat.fetch_group_contacts() 252 | 253 | --------------------------------------------------------------------------------