python監控wechat

import osimport reimport shutilimport timeimport itchatfrom itchat.content import *# 說明：能夠撤回的有文本文字、語音、視頻、圖片、位置、名片、分享、附件# {msg_id:(msg_from,msg_to,msg_time,msg_time_rec,msg_type,msg_content,msg_share_url)}msg_dict = {}# 文件存儲臨時目錄rev_tmp_dir = os.path.join(os.getcwd(),"rollback_itchat/")# rec_tmp_dir = os.path.join(os.getcwd(), 'tmp/')# print(rec_tmp_dir)if not os.path.exists(rev_tmp_dir): os.makedirs(rev_tmp_dir)# 表情有一個問題 | 接受信息和接受note的msg_id不一致 巧合解決方案face_bug = None# 將接收到的消息存放在字典中，當接收到新消息時對字典中超時的消息進行清理 | 不接受不具備撤回功能的信息# [TEXT, PICTURE, MAP, CARD, SHARING, RECORDING, ATTACHMENT, VIDEO, FRIENDS, NOTE]@itchat.msg_register([TEXT, PICTURE, MAP, CARD, SHARING, RECORDING, ATTACHMENT, VIDEO])def handler_receive_msg(msg):    global face_bug    # 獲取的是本地時間戳並格式化本地時間戳 e: 2019-02-16 13:43:20    msg_time_rec = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())    # 消息ID    msg_id = msg['MsgId']    # 消息時間    msg_time = msg['CreateTime']    # 消息發送人暱稱 | 這裏也能夠使用RemarkName備註　可是本身或者沒有備註的人爲None    msg_from = (itchat.search_friends(userName=msg['FromUserName']))["NickName"]    # 消息內容    msg_content = None    # 分享的連接    msg_share_url = None    if msg['Type'] == 'Text'\            or msg['Type'] == 'Friends':        msg_content = msg['Text']    elif msg['Type'] == 'Recording' \            or msg['Type'] == 'Attachment' \            or msg['Type'] == 'Video' \            or msg['Type'] == 'Picture':        msg_content = r"" + msg['FileName']        # 保存文件        msg['Text'](rev_tmp_dir + msg['FileName'])    elif msg['Type'] == 'Card':        msg_content = msg['RecommendInfo']['NickName'] + r" 的名片"    elif msg['Type'] == 'Map':        x, y, location = re.search(            "<location x=\"(.*?)\" y=\"(.*?)\".*label=\"(.*?)\".*", msg['OriContent']).group(1, 2, 3)        if location is None:            msg_content = r"緯度->" + x.__str__() + " 經度->" + y.__str__()        else:            msg_content = r"" + location    elif msg['Type'] == 'Sharing':        msg_content = msg['Text']        msg_share_url = msg['Url']    face_bug = msg_content    # 更新字典    msg_dict.update(        {            msg_id: {                "msg_from": msg_from, "msg_time": msg_time, "msg_time_rec": msg_time_rec,                "msg_type": msg["Type"],                "msg_content": msg_content, "msg_share_url": msg_share_url            }        }    )    print(msg_from+"給你發送了"+msg_content)#若須要監聽公衆號，羣聊的消息，能夠修改裝飾器參數爲：#@itchat.msg_register(NOTE, isFriendChat=True, isGroupChat=True, isMpChat=True)# 收到note通知類消息，判斷是否是撤回並進行相應操做@itchat.msg_register([NOTE])def send_msg_helper(msg):    global face_bug    if re.search(r"\<\!\[CDATA\[.*撤回了一條消息\]\]\>", msg['Content']) is not None:        # 獲取消息的id        old_msg_id = re.search("\<msgid\>(.*?)\<\/msgid\>", msg['Content']).group(1)        old_msg = msg_dict.get(old_msg_id, {})        if len(old_msg_id) < 11:            itchat.send_file(rev_tmp_dir + face_bug, toUserName='filehelper')            os.remove(rev_tmp_dir + face_bug)        else:            msg_body = "告訴你一個祕密~" + "\n" \                       + old_msg.get('msg_time_rec')+"\n" \                       + old_msg.get('msg_from') + " 撤回了 " + old_msg.get("msg_type") + " 消息" + "\n" \                       + "撤回了什麼 ⇣" + "\n" \                       + r"" + old_msg.get('msg_content')            print(msg_body)            # 若是是分享存在連接            if old_msg['msg_type'] == "Sharing": msg_body += "\n就是這個連接➣ " + old_msg.get('msg_share_url')            # 將撤回消息發送到文件助手            itchat.send(msg_body, toUserName='filehelper')            # 有文件的話也要將文件發送回去            if old_msg["msg_type"] == "Picture" \                    or old_msg["msg_type"] == "Recording" \                    or old_msg["msg_type"] == "Video" \                    or old_msg["msg_type"] == "Attachment":                file = '@fil@%s' % (rev_tmp_dir + old_msg['msg_content'])                itchat.send(msg=file, toUserName='filehelper')                os.remove(rev_tmp_dir + old_msg['msg_content'])            # 刪除字典舊消息            msg_dict.pop(old_msg_id)if __name__ == '__main__':    #itchat.auto_login(hotReload=True,enableCmdQR=2)    itchat.auto_login(hotReload=True)    itchat.run()