#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() |