#####################
# WeChat Bot Server #
#####################
## APIs
- /qrcode
------------------------
@brief login qrcode
@method get
@return raw image
------------------------
- /runtime
-----------------------------
@brief basic info
@method get
@return application/json
{
'ret': 0,
'runtime': '',
'total_size': '',
'db_size': '',
'msg_count': '',
'image_count': '',
'voice_count': '',
'video_count': '',
}
-----------------------------
- /group_list
------------------------
@brief list groups
@method get
------------------------
- /group_member_list/<g_id>
-----------------------------
@brief list group member
@method get
@param g_id String
-----------------------------
- /group_chat_log/<g_name>
-------------------------------
@brief list group chat log
@method get
@param g_name String
-------------------------------
- /upload
-------------------------------
@brief upload a file
@method get/post
@return application/json
{
'ret': 0,
'msg': '',
}
-------------------------------
- /send_msg/<to>/<msg>
-------------------------------------------
@brief send message to user or gourp
@method get
@param to: String, user id or group id
@param msg: String, words
@return application/json
{
'ret': 0,
}
-------------------------------------------
- /send_img/<to>/<img>
-------------------------------------------
@brief send image to user or gourp
@method get
@param to: String, user id or group id
@param img: String, image file name
@return application/json
{
'ret': 0,
}
-------------------------------------------
- /send_emot/<to>/<emot>
-------------------------------------------
@brief send emotion to user or gourp
@method get
@param to: String, user id or group id
@param emot: String, emotion file name
@return application/json
{
'ret': 0,
}
-------------------------------------------
- /send_file/<to>/<file>
-------------------------------------------
@brief send emotion to user or gourp
@method get
@param to: String, user id or group id
@param file: String, file name
@return application/json
{
'ret': 0,
}
-------------------------------------------
- /mass_send_msg
-------------------------------------------
@brief send text to mass users or gourps
@method post
@param application/json
{
'to_list': [
'group_id',
...
],
'msg': '',
}
@return application/json
{
'ret': 0,
'unsend_list': [],
}
-------------------------------------------
- /mass_send_img
-------------------------------------------
@brief send image to mass users or gourps
@method post
@param application/json
{
'to_list': [
'group_id',
...
],
'msg': '',
}
@return application/json
{
'ret': 0,
'unsend_list': [],
}
-------------------------------------------
- /mass_send_emot
-------------------------------------------
@brief send emoticon to mass users or gourps
@method post
@param application/json
{
'to_list': [
'group_id',
...
],
'msg': '',
}
@return application/json
{
'ret': 0,
'unsend_list': [],
}
-------------------------------------------
- /mass_send_file
-------------------------------------------
@brief send file to mass users or gourps
@method post
@param application/json
{
'to_list': [
'group_id',
...
],
'msg': '',
}
@return application/json
{
'ret': 0,
'unsend_list': [],
}
-------------------------------------------
' + str(wechat) + '\n\n-----\nLogs:\n-----\n\n' + ''.join(log_file.readlines()[-100:]) + '' sg.send_mail(subject, mail_content, 'text/html') log_file.close() if wechat.exit_code == 0: echo(Constant.MAIN_RESTART) else: # kill process os.system(Constant.LOG_MSG_KILL_PROCESS % os.getpid()) ================================================ FILE: wxbot_project_py2.7/wx_handler/__init__.py ================================================ #!/usr/bin/env python # coding: utf-8 from wechat_msg_processor import WeChatMsgProcessor from bot import Bot from sendgrid_mail import SGMail ================================================ FILE: wxbot_project_py2.7/wx_handler/bot.py ================================================ #!/usr/bin/env python # coding: utf-8 #=================================================== from wechat.utils import * from config import Constant #--------------------------------------------------- import random, time, json #=================================================== class Bot(object): def __init__(self): self.emoticons = Constant.EMOTICON self.gifs = [] self.last_time = time.time() def time_schedule(self): r = '' now = time.time() if int(now - self.last_time) > 3600: self.last_time = now url_latest = Constant.BOT_ZHIHU_URL_LATEST url_daily = Constant.BOT_ZHIHU_URL_DAILY data = get(url_latest) j = json.loads(data) story = j['stories'][random.randint(0, len(j['stories'])-1)] r = story['title'] + '\n' + url_daily + str(story['id']) return r.encode('utf-8') def reply(self, text): APIKEY = Constant.BOT_TULING_API_KEY api_url = Constant.BOT_TULING_API_URL % (APIKEY, text, '12345678') r = json.loads(get(api_url)) if r.get('code') == 100000 and r.get('text') != Constant.BOT_TULING_BOT_REPLY: p = random.randint(1, 10) if p > 3: return r['text'] elif p > 1: # send emoji if random.randint(1, 10) > 5: n = random.randint(0, len(self.emoticons)-1) m = random.randint(1, 3) reply = self.emoticons[n].encode('utf-8') * m return reply return '' ================================================ FILE: wxbot_project_py2.7/wx_handler/sendgrid_mail.py ================================================ #!/usr/bin/env python # coding: utf-8 #=================================================== import sendgrid from sendgrid.helpers.mail import * #=================================================== class SGMail(object): def __init__(self, apikey, from_email, to_email): self.sg = sendgrid.SendGridAPIClient(apikey=apikey) self.from_email = Email(from_email) self.to_email = Email(to_email) def send_mail(self, subject, plain_content, type='text/plain'): content = Content(type, plain_content) mail = Mail(self.from_email, subject, self.to_email, content) response = self.sg.client.mail.send.post(request_body=mail.get()) return response.status_code == 202 ================================================ FILE: wxbot_project_py2.7/wx_handler/wechat_msg_processor.py ================================================ #!/usr/bin/env python # coding: utf-8 #=================================================== from wechat.utils import * from config import ConfigManager from config import Constant from config import Log #--------------------------------------------------- import os import time import json import re #=================================================== class WeChatMsgProcessor(object): """ Process fetched data """ def __init__(self, db): self.db = db self.wechat = None # recieve `WeChat` class instance # for call some wechat apis # read config cm = ConfigManager() [self.upload_dir, self.data_dir, self.log_dir] = cm.setup_database() def clean_db(self): """ @brief clean database, delete table & create table """ self.db.delete_table(Constant.TABLE_GROUP_LIST()) self.db.delete_table(Constant.TABLE_GROUP_USER_LIST()) self.db.create_table(Constant.TABLE_GROUP_MSG_LOG, Constant.TABLE_GROUP_MSG_LOG_COL) self.db.create_table(Constant.TABLE_GROUP_LIST(), Constant.TABLE_GROUP_LIST_COL) self.db.create_table(Constant.TABLE_GROUP_USER_LIST(), Constant.TABLE_GROUP_USER_LIST_COL) self.db.create_table(Constant.TABLE_RECORD_ENTER_GROUP, Constant.TABLE_RECORD_ENTER_GROUP_COL) self.db.create_table(Constant.TABLE_RECORD_RENAME_GROUP, Constant.TABLE_RECORD_RENAME_GROUP_COL) def handle_wxsync(self, msg): """ @brief Recieve webwxsync message, saved into json @param msg Dict: webwxsync msg """ fn = time.strftime(Constant.LOG_MSG_FILE, time.localtime()) save_json(fn, msg, self.log_dir, 'a+') def handle_group_list(self, group_list): """ @brief handle group list & saved in DB @param group_list Array """ fn = Constant.LOG_MSG_GROUP_LIST_FILE save_json(fn, group_list, self.data_dir) cols = [( g['NickName'], g['UserName'], g['OwnerUin'], g['MemberCount'], g['HeadImgUrl'] ) for g in group_list] self.db.insertmany(Constant.TABLE_GROUP_LIST(), cols) def handle_group_member_list(self, group_id, member_list): """ @brief handle group member list & saved in DB @param member_list Array """ fn = group_id + '.json' save_json(fn, member_list, self.data_dir) cols = [( group_id, m['UserName'], m['NickName'], m['DisplayName'], m['AttrStatus'] ) for m in member_list] self.db.insertmany(Constant.TABLE_GROUP_USER_LIST(), cols) def handle_group_list_change(self, new_group): """ @brief handle adding a new group & saved in DB @param new_group Dict """ self.handle_group_list([new_group]) def handle_group_member_change(self, group_id, member_list): """ @brief handle group member changes & saved in DB @param group_id Dict @param member_list Dict """ self.db.delete(Constant.TABLE_GROUP_USER_LIST(), "RoomID", group_id) self.handle_group_member_list(group_id, member_list) def handle_group_msg(self, msg): """ @brief Recieve group messages @param msg Dict: packaged msg """ # rename media files for k in ['image', 'video', 'voice']: if msg[k]: t = time.localtime(float(msg['timestamp'])) time_str = time.strftime("%Y%m%d%H%M%S", t) # format: 时间_消息ID_群名 file_name = '/%s_%s_%s.' % (time_str, msg['msg_id'], msg['group_name']) new_name = re.sub(r'\/\w+\_\d+\.', file_name, msg[k]) Log.debug('rename file to %s' % new_name) os.rename(msg[k], new_name) msg[k] = new_name if msg['msg_type'] == 10000: # record member enter in group m = re.search(r'邀请(.+)加入了群聊', msg['sys_notif']) if m: name = m.group(1) col_enter_group = ( msg['msg_id'], msg['group_name'], msg['from_user_name'], msg['to_user_name'], name, msg['time'], ) self.db.insert(Constant.TABLE_RECORD_ENTER_GROUP, col_enter_group) # record rename group n = re.search(r'(.+)修改群名为“(.+)”', msg['sys_notif']) if n: people = n.group(1) to_name = n.group(2) col_rename_group = ( msg['msg_id'], msg['group_name'], to_name, people, msg['time'], ) self.db.insert(Constant.TABLE_RECORD_RENAME_GROUP, col_rename_group) # upadte group in GroupList for g in self.wechat.GroupList: if g['UserName'] == msg['from_user_name']: g['NickName'] = to_name break # normal group message col = ( msg['msg_id'], msg['group_owner_uin'], msg['group_name'], msg['group_count'], msg['from_user_name'], msg['to_user_name'], msg['user_attrstatus'], msg['user_display_name'], msg['user_nickname'], msg['msg_type'], msg['emoticon'], msg['text'], msg['image'], msg['video'], msg['voice'], msg['link'], msg['namecard'], msg['location'], msg['recall_msg_id'], msg['sys_notif'], msg['time'], msg['timestamp'] ) self.db.insert(Constant.TABLE_GROUP_MSG_LOG, col) text = msg['text'] if text and text[0] == '@': n = trans_coding(text).find(u'\u2005') name = trans_coding(text)[1:n].encode('utf-8') if name in [self.wechat.User['NickName'], self.wechat.User['RemarkName']]: self.handle_command(trans_coding(text)[n+1:].encode('utf-8'), msg) def handle_user_msg(self, msg): """ @brief Recieve personal messages @param msg Dict """ wechat = self.wechat text = trans_coding(msg['text']).encode('utf-8') uid = msg['raw_msg']['FromUserName'] if text == 'test_revoke': # 撤回消息测试 dic = wechat.webwxsendmsg('这条消息将被撤回', uid) wechat.revoke_msg(dic['MsgID'], uid, dic['LocalID']) elif text == 'reply': wechat.send_text(uid, '自动回复') def handle_command(self, cmd, msg): """ @brief handle msg of `@yourself cmd` @param cmd String @param msg Dict """ wechat = self.wechat g_id = '' for g in wechat.GroupList: if g['NickName'] == msg['group_name']: g_id = g['UserName'] cmd = cmd.strip() if cmd == 'runtime': wechat.send_text(g_id, wechat.get_run_time()) elif cmd == 'test_sendimg': wechat.send_img(g_id, 'test/emotion/7.gif') elif cmd == 'test_sendfile': wechat.send_file(g_id, 'test/Data/upload/shake.wav') elif cmd == 'test_bot': # reply bot # --------- if wechat.bot: r = wechat.bot.reply(cmd) if r: wechat.send_text(g_id, r) else: pass elif cmd == 'test_emot': img_name = [ '0.jpg', '1.jpeg', '2.gif', '3.jpg', '4.jpeg', '5.gif', '6.gif', '7.gif', '8.jpg', '9.jpg' ] name = img_name[int(time.time()) % 10] emot_path = os.path.join('test/emotion/', name) wechat.send_emot(g_id, emot_path) else: pass def check_schedule_task(self): # update group member list at 00:00 am every morning t = time.localtime() if t.tm_hour == 0 and t.tm_min <= 1: # update group member Log.debug('update group member list everyday') self.db.delete_table(Constant.TABLE_GROUP_LIST()) self.db.delete_table(Constant.TABLE_GROUP_USER_LIST()) self.db.create_table(Constant.TABLE_GROUP_LIST(), Constant.TABLE_GROUP_LIST_COL) self.db.create_table(Constant.TABLE_GROUP_USER_LIST(), Constant.TABLE_GROUP_USER_LIST_COL) self.wechat.fetch_group_contacts()