import itchat

from itchat.content import TEXT

import re

import os

import numpy as np

from PIL import Image

from wordcloud.wordcloud import WordCloud

if itchat.load_login_status("itchat.pkl"):       # itchat.load_login_status() 用于读取设置

    data = itchat.get_friends(update=True)

    wordList = []

    for i in data:

        if i["Signature"]:

            signature = re.sub(r'<span.*</span>','',i["Signature"]).strip().replace('\n','')

            wordList.append(signature)

    text = " ".join(wordList)

    # 获取当前文件的执行路径

    src_dir = os.getcwd()

    # 生成词云形状的图片地址

    imagePath = src_dir + "\\ciyun.png"

    # 生成词云字体地址（防止中文乱码）

    font = src_dir + "\\SimHei.ttf"

    # 生成的词云图地址

    resultPath = src_dir + "\\好友个性签名词云图.png"

    print("开始生成微信好友个性签名词云...")

    # 开始生成图片

    bg = np.array(Image.open(imagePath))

    wc = WordCloud(

        mode='RGBA',  # 设置透明底色

        mask=bg,         # 造型遮盖

        background_color="white",  # 背景颜色

        max_font_size=150,  # 字体最大值

        min_font_size=10,  # 字体最小值

        max_words=5000,  # 词云显示的最大词数

        random_state=100,  # 设置有多少种随机生成状态，即有多少种配色方案

        font_path=font,  # 设置字体

    ).generate(text)

    wc.to_file(resultPath)

    print("词云图片已生成" + resultPath)

else:

    itchat.auto_login()

    itchat.dump_login_status()