From ca70e6217d03c217bd06354006037a42f6b34718 Mon Sep 17 00:00:00 2001 From: xianhu Date: Sat, 8 Apr 2017 21:56:15 +0800 Subject: [PATCH] update python_wechat.py --- python_wechat.py | 77 ++++++++++++++++++++++++------------------------ 1 file changed, 39 insertions(+), 38 deletions(-) diff --git a/python_wechat.py b/python_wechat.py index b1254d1..d65033c 100644 --- a/python_wechat.py +++ b/python_wechat.py @@ -16,19 +16,23 @@ import datetime from itchat.content import * # 初始化 -newInstance = itchat.new_instance() -newInstance.auto_login(hotReload=False, enableCmdQR=2) -newInstance.global_keys = ["人工智能", "机器学习", "算法", "数据挖掘"] -newInstance.to_user_name = "filehelper" +my = itchat.new_instance() +my.auto_login(hotReload=False, enableCmdQR=2) +my.global_keys = ["人工智能", "机器学习", "算法", "数据挖掘"] +my.to_user_name = "filehelper" -# 获取自己的属性字典：UserName, NickName, RemarkName, Sex(1 or 2) -newInstance.owner = newInstance.loginInfo["User"] +# my还包括的以下属性，注意用点.查看： +# (1) alive 是否还活着，isLogging 是否已登陆 +# (2) loginInfo 登陆信息，其中的User属性为自己的信息User字典类，包括UserName, NickName, RemarkName, Sex(1 or 2)等 +# (3) memberList 通讯录列表，每一项为一个User字典类 +# (4) chatroomList 群聊列表，每一项为一个Chatroom字典类，包括UserName, NickName, RemarkName, MemberCount等 +# (5) mpList 订阅号列表，每一项为一个MassivePlatform字典类，包括UserName, NickName等 -# 获取通讯录: {UserName: UserInstance} -newInstance.friends = {user["UserName"]: user for user in newInstance.get_friends(update=True)} +# 获取并更新通讯录: {UserName: UserInstance} +my.friends = {user["UserName"]: user for user in my.get_friends(update=True)} # 消息存储队列 -newInstance.msg_store = {} +my.msg_store = {} # 消息提取函数 @@ -49,14 +53,16 @@ def get_msg_list(msg): msg_file = msg["FileName"] # 消息中所带文件的名称 msg_url = msg["Url"] # 消息中带有的链接地址 + wind_name = msg["User"]["RemarkName"] if msg["User"].get("RemarkName") else ( + msg["User"]["NickName"] if msg["User"].get("NickName") else to_user_name + ) + if from_user_name.startswith("@@"): # 群消息 - wind_name = msg["User"]["RemarkName"] if msg["User"]["RemarkName"] else msg["User"]["NickName"] - nick_name = msg["ActualNickName"] if (msg["ActualUserName"] not in newInstance.friends) or \ - (not newInstance.friends[msg["ActualUserName"]]["RemarkName"]) else newInstance.friends[msg["ActualUserName"]]["RemarkName"] + nick_name = msg["ActualNickName"] if (msg["ActualUserName"] not in my.friends) or \ + (not my.friends[msg["ActualUserName"]]["RemarkName"]) else my.friends[msg["ActualUserName"]]["RemarkName"] else: # 个人消息 - wind_name = msg["User"]["RemarkName"] if msg["User"]["RemarkName"] else msg["User"]["NickName"] nick_name = wind_name we_type = msg["Type"] # 消息类型 @@ -66,7 +72,7 @@ def get_msg_list(msg): # 消息注册，主要处理群消息 -@newInstance.msg_register([TEXT, MAP, CARD, NOTE, SHARING, PICTURE, RECORDING, ATTACHMENT, VIDEO], isFriendChat=True, isGroupChat=True) +@my.msg_register([TEXT, MAP, CARD, NOTE, SHARING, PICTURE, RECORDING, ATTACHMENT, VIDEO], isFriendChat=True, isGroupChat=True) def text_reply(msg): """ 消息自动接收, 接受全部的消息 @@ -74,61 +80,56 @@ def text_reply(msg): # 消息提取 msg_id, from_user_name, to_user_name, msg_type, msg_content, msg_time, msg_file, msg_url, wind_name, nick_name, we_type, we_text = get_msg_list(msg) - # 消息过滤，过滤自己发送的消息 - if from_user_name == newInstance.owner["UserName"]: - logging.warning("message from myself, skip") - return - # 消息过滤, 只监测文字、注解、分享、图片、语音、视频、附件等 if we_type not in ["Text", "Note", "Sharing", "Picture", "Recording", "Video", "Attachment"]: logging.warning("message ignored") return - if we_type in ["Picture", "Recording", "Video", "Attachment"]: - re_length = re.search("[\"\s]length=\"(?P[\d]+?)\"", msg_content, flags=re.IGNORECASE) - if (not msg_content) or (re_length and (int(re_length.group("length")) < 5000000)): - we_text(".Cache/" + msg_file) - logging.warning("downloading %s to .Cache/", msg_file) + # 处理来自自己的消息 + if from_user_name == my.loginInfo["User"]["UserName"]: + return - # 消息存储 - newInstance.msg_store[msg_id] = msg - - # 删除过期消息 - ids_list = [_id for _id in newInstance.msg_store if time.time() - newInstance.msg_store[_id]["CreateTime"] > 120] - for _id in ids_list: + # 消息存储，删除过期消息 + my.msg_store[msg_id] = msg + for _id in [_id for _id in my.msg_store if time.time() - my.msg_store[_id]["CreateTime"] > 120]: logging.warning("delete message, message_id = %s", _id) - newInstance.msg_store.pop(_id) + my.msg_store.pop(_id) # 处理群消息 if from_user_name.startswith("@@"): # 红包消息处理 if we_type == "Note" and we_text.find("收到红包，请在手机上查看") >= 0: - newInstance.send("【%s】中有红包，快抢！\nFrom: %s\nContent: %s\nTime: %s" % (wind_name, nick_name, msg_content, msg_time), toUserName=newInstance.to_user_name) + my.send("【%s】中有红包，快抢！" % wind_name, toUserName=my.to_user_name) # 提到自己消息处理 if msg["IsAt"]: - newInstance.send("【%s】中有@你的消息:\nFrom: %s\nContent: %s\nTime: %s" % (wind_name, nick_name, msg_content, msg_time), toUserName=newInstance.to_user_name) + my.send("【%s】中有@你的消息:\nFrom: %s\nTime: %s\n%s" % (wind_name, nick_name, msg_time, msg_content), toUserName=my.to_user_name) - for key in newInstance.global_keys: + for key in my.global_keys: if msg_content.find(key) >= 0: - newInstance.send("【%s】中有关键字【%s】:\nFrom: %s\nContent: %s\nTime: %s" % (wind_name, key, nick_name, msg_content, msg_time), toUserName=newInstance.to_user_name) + my.send("【%s】中有关键字【%s】:\nFrom: %s\nTime: %s\n%s" % (wind_name, key, nick_name, msg_time, msg_content), toUserName=my.to_user_name) break # 撤回消息处理 if we_type == "Note" and we_text.find("撤回了一条消息") >= 0: - old_msg = newInstance.msg_store.get(msg_content[msg_content.find("")+7: msg_content.find("")]) + old_msg = my.msg_store.get(msg_content[msg_content.find("")+7: msg_content.find("")]) if not old_msg: return msg_id, from_user_name, to_user_name, msg_type, msg_content, msg_time, msg_file, msg_url, wind_name, nick_name, we_type, we_text = get_msg_list(old_msg) if we_type in ["Picture", "Recording", "Video", "Attachment"]: + re_length = re.search("[\"\s]length=\"(?P[\d]+?)\"", msg_content, flags=re.IGNORECASE) + if (not msg_content) or (re_length and (int(re_length.group("length")) < 5000000)): + we_text(".Cache/" + msg_file) + logging.warning("downloading %s to .Cache/", msg_file) + # 更改内容 msg_content = msg_file elif we_type == "Sharing": msg_content = we_text + ": " + msg_url - newInstance.send("【%s】中有消息被撤回:\nFrom: %s\nType: %s\nContent: %s\nTime: %s" % (wind_name, nick_name, we_type, msg_content, msg_time), toUserName=newInstance.to_user_name) + my.send("【%s】中有消息被撤回:\nFrom: %s\nType: %s\nTime: %s\n%s" % (wind_name, nick_name, we_type, msg_time, msg_content), toUserName=my.to_user_name) return # 运行程序 -newInstance.run() +my.run()