Python-微信聊天机器人

功能实现

1. 好友性别分析
2. 微信消息自动回复
3. 好友头像自动下载

import itchat
import requests
import json
import os
import matplotlib.pyplot as plt
import random

def analysisFriends():
    # 获取好友信息
    friends = itchat.get_friends()
    male = 0
    female = 0
    others = 0
    for i in friends:
        sex = i['Sex']
        # print(sex)
        if sex == 1:
            male += 1
        elif sex == 2:
            female += 1
        else:
            others += 1
    total = len(friends[1:])
    print('男性好友：%.2f%%' % (float(male) / total * 100))
    print('女性好友：%.2f%%' % (float(female) / total * 100))
    print('其他：%.2f%%' % (float(others) / total * 100))
    # print(friends)
    man_rate = float(male) / total * 100
    woman_rate = float(female) / total * 100
    others_rate = float(others) / total * 100
    rate = [man_rate, woman_rate, others_rate]
    labels = ['男', '女', '其他']
    # explode = [0.1, 0, 0]
    plt.rcParams['font.sans-serif'] = ['SimHei']
    plt.rcParams['axes.unicode_minus'] = False
    plt.axes(aspect='equal')
    plt.pie(
        x=rate,
        # explode=explode,
        labels=labels,
        autopct='%.1f%%',
        pctdistance=0.8,
        labeldistance=1.2,
        startangle=180,
        radius=1.2,
        wedgeprops={'linewidth': 1.5, 'edgecolor': 'pink'},
        textprops={'fontsize': 15, 'color': 'black'},
    )
    # friends_dic = json.loads(friends)
    # a = plt.title(friends['NickName'][0:0] + '的好友比例分布图')
    # print(type(a))
    plt.show()

def getHeadImg():
    friends = itchat.get_friends()[0:]
    for j in friends:
        img = itchat.get_head_img(userName=j['UserName'])
        # path = 'C:\\Users\仁义\Desktop\好友头像\\' + j['NickName'] + '.jpg'
        dirname = '好友头像'
        if not os.path.exists(dirname):
            os.mkdir(dirname)
        path = dirname + '/' + j['RemarkName'] + '.jpg'
        try:
            with open(path, 'wb') as f:
                f.write(img)
        except Exception as e:
            print(repr(e))
        print(j['RemarkName'] + '的头像已下载完成。')

def get_response(message):
    apiUil = 'http://openapi.tuling123.com/openapi/api/v2'
    keys = ['key1',
            'key2',
            'key3',
            'key4',
            'key5',
            ]
    key = random.choice(keys)
    data = {
        "reqType": '',
        "perception": {
            "inputText": {
                "text": message
            },
            "inputImage": {
                "url": "imageUrl"
            },
            "selfInfo": {
                "location": {
                    "city": "哈尔滨",
                    "province": "黑龙江省",
                    "street": "电碳路"
                }
            }
        },
        "userInfo": {
            "apiKey": 'key',
            "userId": "robot"
        }
    }
    # print(data)
    # 将dic格式的data编码为UTF-8
    data = json.dumps(data)
    # print(data)
    # requests的请求方式为post,并将请求结果转换为文本
    re = requests.post(url=apiUil, data=data, ).text
    # 将请求结果str格式转换为dict格式
    re_dic = json.loads(re)
    # 提取机器人回复内容
    robotResponse = re_dic['results'][0]['values']['text']
    return robotResponse

@itchat.msg_register(itchat.content.TEXT, isFriendChat=True)
def robot_reply(message):
    defaultReply = '收到'
    reply = get_response(message['Text'])
    return reply or defaultReply


if __name__ == '__main__':
    # 登录微信
    itchat.login()
    print('##########     微信助手V1.0-仁义     ##########')
    # itchat.auto_login(hotReload=True, )
    a = input('是否分析好友性别比例(y/n):')
    if a == 'y':
        analysisFriends()
    b = input('是否开始机器人自动回复(y/n):')
    if b == 'y':
        # get_response('你好')
        # robot_reply()
        print('开启机器人自动回复......')
        itchat.run()
    c = input('是否下载好友头像(y/n):')

    if c == 'y':
        getHeadImg()