python微信撤回监测代码

#coding=utf8
import itchat
import requests
import time
import os
import re
import threading

#全局变量,对于每个用户的机器人开关
User_bot_control_flag = {}
#全局变量,我的昵称
myNickName = ''

def bot_chat_init():
  # 获取好友列表
  friends = itchat.get_friends(update=True)[0:]
  #将标志位置为0
  for i in friends[1:]:
    User_bot_control_flag[i["UserName"]] = 0


@itchat.msg_register(itchat.content.TEXT)
def tuling_reply(msg):
  #  获取到发送消息者身份,如果身份匹配,就做对应的事
  # itchat.send_msg('已经收到了文本消息，消息内容为%s' % msg['Text'], toUserName=msg['FromUserName'])
  # 如果图灵Key出现问题，那么reply将会是None
  if msg['Text']=='service crond start':
    return u'你一看就是个程序员'
  if msg['Text'] == 'dididididi':
    return u'开车了'
  reply = get_response(msg['Text'])
  if not msg['FromUserName'] == myUserName:
    pass
    # 发送一条提示给文件助手
    # itchat.send_msg(u"[%s]收到好友@%s 的信息：%s\n" %
    #         (time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(msg['CreateTime'])),
    #         msg['User']['NickName'],
    #         msg['Text']), 'filehelper')
  # a or b的意思是，如果a有内容，那么返回a，否则返回b
  # 有内容一般就是指非空或者非None，你可以用`if a: print('True')`来测试
  return reply or u'[自动回复]您好，我现在有事不在，一会再和您联系。\n已经收到您的的信息：%s\n' % (msg['Text'])


def friend():
  # 初始化计数器，有男有女，当然，有些人是不填的
  # 获取好友列表
  friends = itchat.get_friends(update=True)[0:]
  male = female = other = 0

  # 遍历这个列表，列表里第一位是自己，所以从"自己"之后开始计算
  # 1表示男性，2女性
  for i in friends[1:]:
    print (i)      #打印出签名
    sex = i["Sex"]
    if sex == 1:
      male += 1
    elif sex == 2:
      female += 1
    else:
      other += 1
  # 总数算上，好计算比例啊～
  total = len(friends[1:])
  # 好了，打印结果
  print(u"共有好友：%d" % total)
  print (u"男性好友：%.2f%%" % (float(male) / total * 100))
  print (u"女性好友：%.2f%%" % (float(female) / total * 100))
  print (u"其他：%.2f%%" % (float(other) / total * 100))

def get_response(msg):
  # 这里我们就像在“3. 实现最简单的与图灵机器人的交互”中做的一样
  # 构造了要发送给服务器的数据
  apiUrl = 'http://www.tuling123.com/openapi/api'
  data = {
    'key'  : key,
    'info'  : msg,
    'userid' : 'wechat-robot',
  }
  try:
    r = requests.post(apiUrl, data=data).json()
    # 字典的get方法在字典没有'text'值的时候会返回None而不会抛出异常
    return r.get('text')+'----来自机器人小Z的智能回复----'
  # 为了防止服务器没有正常响应导致程序异常退出，这里用try-except捕获了异常
  # 如果服务器没能正常交互（返回非json或无法连接），那么就会进入下面的return
  except:
    # 将会返回一个None
    return


@itchat.msg_register(itchat.content.TEXT, isGroupChat=True) #msg['ActualNickName'] 群里发消息的人名 #msg['User']['NickName'] 群名称
def text_reply(msg):
  # print (msg['User'])    #一个宏大的结构体
  # print ("群聊名字"+msg['User']['NickName']) #群聊名称
  # print (msg['FromUserName'])
  #监控所有群的消息,后来做统计用,后面可以做关键词分析什么的
  file_object = open(myNickName+"群"+msg['User']['NickName'], 'a')
  write_data = ''.join(time.strftime("%Y-%m-%d %H:%M:%S" , time.localtime(msg['CreateTime'])))+" "+msg['ActualNickName']+": "+msg['Text']+"\n"
  file_object.write(write_data)
  file_object.close()
  #指定群聊可以智能群聊
  if msg['User']['NickName'] == '184':
    print (" 184 ok")
    itchat.send(get_response(msg['Text']),msg['FromUserName'])
  #监控群聊内容发送到文件助手,已经被自己屏蔽掉了
  # itchat.send_msg(u"[%s]收到%s群 %s 的信息：%s\n" %
  #         (time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(msg['CreateTime']))
  #         ,msg['User']['NickName'],msg['ActualNickName'],
  #         msg['Text']), 'filehelper')
  # 判断是否有人@自己
  if (msg.isAt):
   # 如果有人@自己，就发一个消息告诉对方我已经收到了信息
    itchat.send_msg("我已经收到了来自{0}的消息，实际内容为{1}".format(msg['ActualNickName'], msg['Text']),
      toUserName=msg['FromUserName'])

# def sendmsgToPsh():
#   while (True):
#     pass
#     # print ("123456")
#
# threads = []
# t1 = threading.Thread(target=sendmsgToPsh())


# 说明：可以撤回的有文本文字、语音、视频、图片、位置、名片、分享、附件

# {msg_id:(msg_from,msg_to,msg_time,msg_time_rec,msg_type,msg_content,msg_share_url)}
msg_dict = {}

# 文件存储临时目录
rev_tmp_dir = "/home/seen/PycharmProjects/Code"
if not os.path.exists(rev_tmp_dir): os.mkdir(rev_tmp_dir)

# 表情有一个问题 | 接受信息和接受note的msg_id不一致 巧合解决方案
face_bug = None


# # 将接收到的消息存放在字典中，当接收到新消息时对字典中超时的消息进行清理 | 不接受不具有撤回功能的信息
# # [TEXT, PICTURE, MAP, CARD, SHARING, RECORDING, ATTACHMENT, VIDEO, FRIENDS, NOTE]
# @itchat.msg_register([itchat.content.TEXT, itchat.content.PICTURE, itchat.content.MAP, itchat.content.CARD, itchat.content.SHARING,
#            itchat.content.RECORDING,itchat.content. ATTACHMENT, itchat.content.VIDEO],isGroupChat=True)
# def handler_receive_msg(msg):
#   #回复特定用户消息
#   # if msg['User']['NickName']=='YYYYY' or msg['User']['NickName']=='彭芊芊':
#   #   print ("yhj ok")
#   #   itchat.send_msg(get_response(msg['Text']), toUserName=msg['FromUserName'])
#   # 先获取对方说来的话
#   # 下面一行是获取发送消息者昵称
#   send_user_name = itchat.search_friends(userName=msg['FromUserName'])['NickName']
#   file_object = open(myNickName + "&" + msg['User']['NickName'], 'a')
#   write_data = ''.join(time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(msg['CreateTime']))) + " " + \
#         send_user_name + ": " + msg['Text'] + "\n"
#   file_object.write(write_data)
#   file_object.close()
#
#   #控制指令检测模块
#   if msg['Text'] == 'service robot start':
#     User_bot_control_flag[msg['FromUserName']]=1   #检测到开启指令后开启机器人
#     itchat.send_msg("Robot small Z started...waiting for your service", toUserName=msg['FromUserName'])
#   if msg['Text'] == 'service robot stop':
#     User_bot_control_flag[msg['FromUserName']]=0   #检测到开启指令后关闭机器人
#     itchat.send_msg("Robot small Z stoped...get 'service robot start' restarted", toUserName=msg['FromUserName'])
#   #在开关开启的情况下回复对方对话
#   if not msg['FromUserName'] == myUserName:
#     if User_bot_control_flag[msg['FromUserName']]:
#       # 存储单人对话模块
#       # 下面一行是获取发送消息者昵称
#       reply = get_response(msg['Text'])
#       file_object = open(myNickName + "&" + msg['User']['NickName'], 'a')
#       write_data = ''.join(time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(msg['CreateTime']))) + " " + \
#             myNickName + ": " + reply + "\n"
#       file_object.write(write_data)
#       file_object.close()
#       itchat.send_msg(reply, toUserName=msg['FromUserName'])
#
#   global face_bug
#   # 获取的是本地时间戳并格式化本地时间戳 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']
#   # 消息时间
#   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
#       }
#     }
#   )

# 将接收到的消息存放在字典中，当接收到新消息时对字典中超时的消息进行清理 | 不接受不具有撤回功能的信息
# [TEXT, PICTURE, MAP, CARD, SHARING, RECORDING, ATTACHMENT, VIDEO, FRIENDS, NOTE]
@itchat.msg_register([itchat.content.TEXT, itchat.content.PICTURE, itchat.content.MAP, itchat.content.CARD, itchat.content.SHARING,
           itchat.content.RECORDING,itchat.content. ATTACHMENT, itchat.content.VIDEO])
def handler_receive_msg(msg):
  #回复特定用户消息
  # if msg['User']['NickName']=='YYYYY' or msg['User']['NickName']=='彭芊芊':
  #   print ("yhj ok")
  #   itchat.send_msg(get_response(msg['Text']), toUserName=msg['FromUserName'])
  # 先获取对方说来的话
  # 下面一行是获取发送消息者昵称
  send_user_name = itchat.search_friends(userName=msg['FromUserName'])['NickName']
  file_object = open(myNickName + "&" + msg['User']['NickName'], 'a')
  write_data = ''.join(time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(msg['CreateTime']))) + " " + \
         send_user_name + ": " + msg['Text'] + "\n"
  file_object.write(write_data)
  file_object.close()

  #控制指令检测模块
  if msg['Text'] == 'service robot start':
    User_bot_control_flag[msg['FromUserName']]=1   #检测到开启指令后开启机器人
    itchat.send_msg("Robot small Z started...waiting for your service", toUserName=msg['FromUserName'])
  if msg['Text'] == 'service robot stop':
    User_bot_control_flag[msg['FromUserName']]=0   #检测到开启指令后关闭机器人
    itchat.send_msg("Robot small Z stoped...get 'service robot start' restarted", toUserName=msg['FromUserName'])
  #在开关开启的情况下回复对方对话
  if not msg['FromUserName'] == myUserName:
    if User_bot_control_flag[msg['FromUserName']]:
      # 存储单人对话模块
      # 下面一行是获取发送消息者昵称
      reply = get_response(msg['Text'])
      file_object = open(myNickName + "&" + msg['User']['NickName'], 'a')
      write_data = ''.join(time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(msg['CreateTime']))) + " " + \
             myNickName + ": " + reply + "\n"
      file_object.write(write_data)
      file_object.close()
      itchat.send_msg(reply, toUserName=msg['FromUserName'])

  global face_bug
  # 获取的是本地时间戳并格式化本地时间戳 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']
  # 消息时间
  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
      }
    }
  )


# # 收到note通知类消息，判断是不是撤回并进行相应操作,针对于群
# @itchat.msg_register([itchat.content.NOTE],isGroupChat=True)
# 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_from') + " 撤回了 " + old_msg.get("msg_type") + " 消息" + "\n" \
#            + old_msg.get('msg_time_rec') + "\n" \
#            + "撤回了什么 ⇣" + "\n" \
#            + r"" + old_msg.get('msg_content')
#       # 如果是分享存在链接
#       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)

# 收到note通知类消息，判断是不是撤回并进行相应操作
@itchat.msg_register([itchat.content.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_from') + " 撤回了 " + old_msg.get("msg_type") + " 消息" + "\n" \
            + old_msg.get('msg_time_rec') + "\n" \
            + "撤回了什么 ⇣" + "\n" \
            + r"" + old_msg.get('msg_content')
      # 如果是分享存在链接
      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)




key = '02dd1dd1b5594e179aa4aca9a6a690a6'
if __name__ == '__main__':
  itchat.auto_login(hotReload=True)
  # 获取自己的UserName
  myNickName = itchat.get_friends(update=True)[0]["NickName"]
  myUserName = itchat.get_friends(update=True)[0]["UserName"]
  #做函数功能的实验
  # print (itchat.search_friends(name='彭芊芊')[0]['UserName'])    #我居然会用了这种办法我是真的猛
  # print(type(itchat.search_friends(name='彭芊芊')))
  #itchat.send("init messages to dindsong,A message from bangbangtang,distant areas...", toUserName='@509f2668d9380a6aeb1951585256827dc1d475c2de885b62fae401401d522f9b')
  friend()     #获取朋友信息

  bot_chat_init()  #初始化开关模块
  itchat.run()