首先需要用到三个库:
- wxpy库(用于获取好友头像然后下载)、
- pillow库(用于拼接头像)、
- pyinstaller库(用于打包python程序成exe文件)
这里我说一下创建虚拟环境的步骤:
我先要提醒一点,下面的操作你应该先找到pycharmprojects目录下的Scripts文件夹(里面得有pip文件)
1.安装pipenv和pyinstaller包,用于后续创建虚拟环境和打包程序
pip install pipenv
pip install pyinstaller
2.选择一个合适的目录作为python虚拟环境
pipenv install #创建虚拟环境
pipenv shell #进入虚拟环境
3.安装程序引用的库(wxpy,math,os,pillow)
pipenv install wxpy math os
4.打包运行程序
程序代码如下:
from wxpy import *
import math
from PIL import Image
import os
# 创建头像存放文件夹
def creat_filepath():
avatar_dir = os.getcwd() + "\\wechat\\"
if not os.path.exists(avatar_dir):
os.mkdir(avatar_dir)
return avatar_dir
# 保存好友头像
def save_avatar(avatar_dir):
# 初始化机器人,扫码登陆
bot = Bot()
friends = bot.friends(update=True)
num = 0
for friend in friends:
friend.get_avatar(avatar_dir + '\\' + str(num) + ".jpg")
print('好友昵称:%s' % friend.nick_name)
num = num + 1
# 拼接头像
def joint_avatar(path):
# 获取文件夹内头像个数
length = len(os.listdir(path))
# 设置画布大小
image_size = 2560
# 设置每个头像大小
each_size = math.ceil(2560 / math.floor(math.sqrt(length)))
# 计算所需各行列的头像数量
x_lines = math.ceil(math.sqrt(length))
y_lines = math.ceil(math.sqrt(length))
image = Image.new('RGB', (each_size * x_lines, each_size * y_lines))
x = 0
y = 0
for (root, dirs, files) in os.walk(path):
for pic_name in files:
# 增加头像读取不出来的异常处理
try:
with Image.open(path + pic_name) as img:
img = img.resize((each_size, each_size))
image.paste(img, (x * each_size, y * each_size))
x += 1
if x == x_lines:
x = 0
y += 1
except IOError:
print("头像读取失败")
img = image.save(os.getcwd() + "/wechat.png")
print('微信好友头像拼接完成!')
if __name__ == '__main__':
avatar_dir = creat_filepath()
save_avatar(avatar_dir)
joint_avatar(avatar_dir)