首页技术文章正文

Python 一键制作微信好友图片墙的方法【黑马python培训】

更新时间:2019年07月26日 10时49分18秒 来源:黑马程序员论坛

用 Python 代码怎么实现的,代码不长,60 行就可以搞定。

核心是利用三个个库:

wxpy 库,用于获取好友头像然后下载
Pillow 库,用于拼接头像
Pyinstaller 库,用来打包 Python 程序成 exe 文件
程序通过三个函数实现,第一个 creat_filepath 函数生成图片下载文件路径,第二个 save_avatar 函数循环获取微信好友头像然后保存到本地,第三个 joint_avatar 函数就是把头像拼接成一张大图。

完整代码如下:

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)
可以直接在运行程序文件,也可以用 Pyinstaller 文件打包后运行。这里额外说一下 pyinstaller 打包的方法和闭坑指南。

不要直接在系统中用 pyinstaller 打包,否则打包出来的 exe 文件会很大。建议在虚拟环境中打包,打包出来的 exe 文件会小很多, 10MB 左右。

虚拟环境创建很简单,简单说一下步骤:

1 安装 pipenv 和 pyinstaller 包,用于后续创建虚拟环境和打包程序:
pip install pipenv
pip install pyinstaller # 已安装就不用安装了

2 选择一个合适的目录作为 Python 虚拟环境,运行:
pipenv install # 创建虚拟环境
pipenv shell # 创建好后,进入虚拟环境

3 安装程序引用的库,上面程序引用了四个库:wxpy、math、os 和 PIL,一行代码就可以完成安装。

pipenv install wxpy math os

4 这里要额外注意 PIL 的安装,现在不用 PIL 库,而是用 Pillow 库取代,所以安装 Pillow 库就行。但不要安装最新的 6.0.0 版本,否则可能会遇到各种错误,
例如:PIL 无法识别下载的 jpg 头像文件。

OSError: cannot identify image file <ImageFieldFile: images

正确的安装方法是安装低版本,经尝试安装 4.2.1 版本没有问题,安装命令:

pipenv install Pillow==4.2.1

5 然后打包程序就可以了:

pyinstaller -F C:\Users\sony\Desktop\wechat_avatar.py
# 程序路径要改成你电脑上的路径
# -F 表示生成单个 exe 文件,方便运行
运行如下:

运行命令,1 分钟左右若显示 successfully 字样表示程序打包成功:

接着在程序目录下找到 wechat_avatar.exe 文件,然后按照第一种方法那样运行就行了。
---------------------
和我们在线交谈!