import wxpy
import re
 
# 實(shí)例化微信對(duì)象
bot = wxpy.Bot(cache_path=True)
 
# 啟用puid 并指定puid所需映射數(shù)據(jù)保存的路徑
bot.enable_puid(path='wxpy_puid.pkl')
 
# 用于存放每個(gè)用戶最近發(fā)送的消息
msgs = dict()
 
# 信息的類型
msg_types = {
 'Text': '文本',
 'Map': '位置',
 'Picture': '圖片',
 'Video': '視頻',
 'Attachment': '文件',
 'Sharing': '分享',
 'Card': '名片',
 'Recording': '語(yǔ)音',
}
 
 
@bot.register()
def handle_receive_msg(msg):
 """
 監(jiān)聽消息
 :param msg: 接收到的消息
 :return:
 """
 # 原信息數(shù)據(jù)
 raw = msg.raw
 
 # 如果消息的狀態(tài)碼是4 即撤回消息
 if raw.get('Status') == 4:
  # 如果是群消息
  if isinstance(msg.chat, wxpy.api.chats.group.Group):
   # 獲取群成員的puid
   puid = msg.member.puid
   # 獲取發(fā)送者的昵稱
   name = msg.member.nick_name
  # 如果是好友消息
  elif isinstance(msg.chat, wxpy.api.chats.friend.Friend):
   # 獲取好友的puid
   puid = msg.chat.puid
   # 獲取好友的昵稱
   name = msg.chat.nick_name
  else:
   puid = None
   name = None
  if puid:
   # 被撤回消息的msgid的匹配規(guī)則
   msg_id_regex = re.compile('<msgid>(\d+)</msgid>')
   # 獲取被撤回消息的msgid
   old_msg_id = msg_id_regex.findall(raw.get('Content'))[0]
   # 獲取該發(fā)送者的最后5次的消息記錄
   chat_msgs = msgs.get(puid)
   # 遍歷消息記錄
   for chat_msg in chat_msgs[::-1]:
    # 跳過(guò)不是被撤回的信息
    if str(chat_msg.id) != old_msg_id:
     continue
    chat = chat_msg.chat
    # 如果被撤回的信息是文本信息
    if chat_msg.type == "Text":
     # 如果消息長(zhǎng)度過(guò)長(zhǎng) 則不予處理
     if len(chat_msg.text) >= 150:
      warning = "【您撤回的消息過(guò)長(zhǎng)，有炸群嫌疑，不予處理！??！】"
      bot.file_helper.send('%s撤回了一條文本消息--【%s】'.decode('utf-8') % (name, warning))
      break
     # 將此消息轉(zhuǎn)發(fā)出來(lái)
     chat_msg.forward(chat, prefix='%s撤回了一條文本消息，消息內(nèi)容為：'.decode('utf-8') % name)
    # 如果被撤回的是位置信息
    elif chat_msg.type == "Map":
     # 位置信息的匹配規(guī)則
     map_regex = re.compile(r'label="(.+?)"')
     # 獲取位置信息中的位置
     map = map_regex.findall(chat_msg.raw.get("OriContent"))[0]
     # 將位置信息發(fā)出來(lái)
     msg.reply('%s撤回了一條位置消息，位置信息為：【%s】'.decode('utf-8') % (name, map))
    else:
     # 獲取信息的類型
     msg_type = msg_types.get(chat_msg.type).decode('utf-8')
     # 將信息轉(zhuǎn)發(fā)出來(lái)
     chat_msg.forward(chat, prefix='%s撤回了一條%s消息， 消息內(nèi)容為：'.decode('utf-8') % (name, msg_type))
    break
 else:
  # 如果是群消息
  if isinstance(msg.chat, wxpy.api.chats.group.Group):
   # 獲取群成員的puid
   puid = msg.member.puid
  # 如果是好友消息
  elif isinstance(msg.chat, wxpy.api.chats.friend.Friend):
   # 獲取好友的puid
   puid = msg.chat.puid
  else:
   puid = None
  if puid:
   # 記錄消息
   msgs.setdefault(puid, []).append(msg)
   # 截取消息 保留最大5條記錄
   msgs[puid] = msg[puid][-5:]
 
# 使機(jī)器人后臺(tái)運(yùn)行，并進(jìn)入交互模式
wxpy.embed()

# 指定一個(gè)好友
friend = bot.friends().search('您對(duì)朋友的備注')
# 指定一個(gè)群
group = bot.groups().search('群名')
 
# 實(shí)例化圖靈機(jī)器人 免費(fèi)申請(qǐng)圖靈機(jī)器人: http://www.tuling123.com/
tu_ling = wxpy.Tuling(api_key="自行注冊(cè)")
# chats指定對(duì)哪些用戶起作用， 如果chats=None(默認(rèn)值）對(duì)所有用戶起作用
@bot.register(chats=[friend, group]) 
def reply_msg(msg):
 """
 自動(dòng)回復(fù)消息
 :param msg: 接收到的信息數(shù)據(jù)
 :return: 回復(fù)文本
 """
 # do_reply會(huì)自動(dòng)回復(fù)消息并返回消息文本
 tu_ling.do_reply(msg)