# -*-encoding:utf-8-*-import osimport reimport shutilimport timeimport itchatfrom itchat.content import *# 說(shuō)明：可以撤回的有文本文字、語(yǔ)音、視頻、圖片、位置、名片、分享、附件# {msg_id:(msg_from,msg_to,msg_time,msg_time_rec,msg_type,msg_content,msg_share_url)}msg_dict = {}# 文件存儲(chǔ)臨時(shí)目錄rev_tmp_dir = "/RevDir/"if not os.path.exists(rev_tmp_dir): os.mkdir(rev_tmp_dir)# 表情有一個(gè)問(wèn)題 | 接受信息和接受note的msg_id不一致 巧合解決方案face_bug = None# 將接收到的消息存放在字典中，當(dāng)接收到新消息時(shí)對(duì)字典中超時(shí)的消息進(jìn)行清理 | 不接受不具有撤回功能的信息# [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  # 獲取的是本地時(shí)間戳并格式化本地時(shí)間戳 e: 2017-04-21 21:30:08  msg_time_rec = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())  # 消息ID  msg_id = msg['MsgId']  # 消息時(shí)間  msg_time = msg['CreateTime']  # 消息發(fā)送人昵稱(chēng) | 這里也可以使用RemarkName備注　但是自己或者沒(méi)有備注的人為None  msg_from = (itchat.search_friends(userName=msg['FromUserName']))["NickName"]  # 消息內(nèi)容  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__() + " 經(jīng)度->" + 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      }    }  )# 收到note通知類(lèi)消息，判斷是不是撤回并進(jìn)行相應(yīng)操作@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 = "愛(ài)你蕾蕾 ?" + "/n" /            + old_msg.get('msg_from') + " 撤回了 " + old_msg.get("msg_type") + " 消息" + "/n" /            + old_msg.get('msg_time_rec') + "/n" /            + "內(nèi)容 ?" + "/n" /            + r"" + old_msg.get('msg_content')      # 如果是分享存在鏈接      if old_msg['msg_type'] == "Sharing": msg_body += "/n就是這個(gè)鏈接? " + old_msg.get('msg_share_url')      # 將撤回消息發(fā)送到文件助手      itchat.send(msg_body, toUserName='filehelper')      # 有文件的話(huà)也要將文件發(fā)送回去      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=1)  itchat.run()