使用itchat库,实现微信的登录和好友信息的获取,将获取的好友头像进行拼接生成照片墙,拼接时加限制条件生成心形照片墙,
获取好友个性签名生成词云。
涉及:
1. itchat的简单使用
2. 图像拼接
3. 按照指定规则进行图像拼接,生成心形图像
4. wordcloud生成词云
详解:
相关库的安装
pip install itchat 微信接口模块
pip install wordcloud 词云模块
pip install jieba jieba分词模块
登录微信,获取好友头像及个性签名
def get_wechat(self):try:# 加上hotReload=True,就会保留登录的状态,至少在后面的几次登录过程中不会再次扫描二维码,# 该参数生成一个静态文件itchat.pkl用于存储登录状态# itchat.auto_login(hotReload=True)# 扫码登录itchat.auto_login()except:print("请检查网络,再重试")# 获取微信好友friends = itchat.get_friends(update=True)# 创建photo_image文件夹来保存微信头像图片if not os.path.exists('photo_image'):# 若文件夹不存在,则创建目录os.mkdir('photo_image')num=0for friend in friends:# 根据用户名获取对应的微信头像img=itchat.get_head_img(userName=friend['UserName'])# 图片路径名img_name=''.join(['photo_image/img',str(num),'.jpg'])# 保存图片with open(img_name,'wb') as f:f.write(img)num+=1# 获取生成词云需要的所有好友的个性签名signature=friend['Signature'].strip()# 剔除个性签名为空以及含span标签的,将所有签名通过空格拼成字符串if len(signature)>0 and not '<span class=' in signature:self.signature_txt+=signature+' '
拼接头像,生成照片墙
def merge_image(self):# 获取指定路径下的文件列表all_image=os.listdir('photo_image')# 设定每个头像的大小each_size=int(math.sqrt(float(1024*1024)/len(all_image)))# 照片墙的行数lines=int(1024/each_size)print(lines)# 创建Image对象,初始化大小image=Image.new('RGBA',(1024,1024))x,y=0,0for i in range(len(all_image)):try:img = Image.open(''.join(['photo_image/img', str(i), '.jpg']))# 重新设置图像大小img = img.resize((each_size, each_size), Image.ANTIALIAS)# 根据x,y坐标位置拼接图像image.paste(img, (x * each_size, y * each_size))# 更新下一张图像位置x += 1except:passfinally:# 一行一行拼接if x == lines:x = 0y += 1# 保存生成的照片墙# RGBA意思是红色,绿色,蓝色,Alpha的色彩空间,Alpha指透明度。而JPG不支持透明度,所以要么丢弃Alpha,要么保存为.png文件image.save('photo_image/photo_wall.png')itchat.send_msg('好友头像照片墙', 'filehelper')# 发送图像到微信文件传输助手,打开手机微信可查看itchat.send_image('photo_image/photo_wall.png','filehelper')
心形函数
在直角坐标系中,爱心线的方程的python 表达为:
x** 2+ y** 2 + a * x= a * sqrt(x** 2+y** 2)
x** 2+ y** 2 - a * x= a * sqrt(x** 2+y** 2)
# 计算心形,判断图像的坐标是否在心形函数内@classmethoddef get_heart_shape(cls,x,y):y1 = 0.618 * np.abs(x) - 0.7 * np.sqrt(262144 - x ** 2)y2 = 0.618 * np.abs(x) + 0.7 * np.sqrt(262144 - x ** 2)if y<y1 or y>y2:return Falseelse:return True
拼接心形照片墙
# 拼接为心形照片墙def merge_love_image(self):# 获取指定路径下的文件列表all_image=os.listdir('photo_image')# 设定每个头像的大小each_size=int(math.sqrt(float(1024*1024)/len(all_image)))# 一行的图像个数,若为偶数个,则+1转换为奇数个,修改每个图像的大小,使最后的心形对称好看num=int(1024/each_size)if num%2==0:num+=1each_size=int(1024/num)# 照片墙的行数lines=int(1024/each_size)print(lines)# 创建Image对象,初始化大小,其大小不直接设定为(1024*1024),因为照片拼接出的尺寸不是正好等于1024,故用实际拼接的尺寸image=Image.new('RGBA',(lines*each_size,lines*each_size))x,y=0,0for i in range(len(all_image)):try:# 由于图像坐标从(0,0)开始,而心形函数对应左上角坐标为(-512,512),故此在判断坐标时,稍作转换is_heart_part=self.get_heart_shape(-512+x * each_size,512-y * each_size)if not is_heart_part:passelse:img = Image.open(''.join(['photo_image/img', str(i), '.jpg']))# 重新设置图像大小img = img.resize((each_size, each_size), Image.ANTIALIAS)# 根据x,y坐标位置拼接图像image.paste(img, (x * each_size, y * each_size))# 更新下一张图像位置x += 1except:passfinally:# 一行一行拼接if x == lines:x = 0y += 1
个性签名生成词云
# 生成词云def get_wordcloud(self):try:image = np.array(Image.open('./namei.jpeg'))# jieba分词signature_txt_list = jieba.cut(self.signature_txt, cut_all=False)jieba_txt = ' '.join(signature_txt_list)# 生成词云wordcloud = WordCloud(font_path=r'C:\Windows\Fonts\msyh.ttc', # 调用系统自带字体(微软雅黑)background_color='white', # 背景色max_words=500, # 最大显示单词数max_font_size=60, # 频率最大单词字体大小mask=image # 自定义显示的效果图).generate(jieba_txt)plt.imshow(wordcloud, interpolation='bilinear')plt.axis('off')plt.show()# 保存为图片wordcloud.to_file('./signature_wordcloud.png')itchat.send_msg('好友个性签名词云', 'filehelper')# tips: 发送的图片名称必须为英文,若为中文,会发送失败itchat.send_image('signature_wordcloud.png', 'filehelper')except:print('词云生成失败,请重试')
运行结果:
运行程序,弹出二维码,扫描登录
等待图像下载拼接,结果将发送到微信的文件传输助手
直接拼接的照片墙
心形图
心形照片墙
个性签名词云
背景图
词云