简单介绍下本篇文章幕后小背景吧——
某周末，炒鸡热
吃完中午饭，和某人在家静静地呆着刷剧，听着窗外似有若无的知了虫鸣
突然，脑中灵光闪现，想起了最近正在鼓捣的PIL
于是转头冲某人邪魅一笑：想不想 炸一波 朋！友！圈！？
某人：蛤？
··· ···
说干就干！小本本小本本抱过来！小板凳小板凳坐起来！
确认语言环境：Pyhton3
####-第一步-
pip导入需要用到的四个库
import itchat
import os
import math
import PIL.Image as Image
####-第二步-
指定好友头像缓存目录，因为头像文件数量比较多，所以在当前路径下添加目录 ‘/img/’ ；若无此目录，捕获 FileNotFoundError 异常，创建目录，继续调用函数。
# 好友头像缓存路径
IMG_DIR = os.getcwd() + '/img/'
# 下载好友头像
def download_imgs(firends):
num = 1
try:
for firend in firends:
img_name = str(num) + '.jpg'
img = itchat.get_head_img(userName=firend['UserName'])
with open(IMG_DIR + img_name, 'wb') as file:
file.write(img)
print(u'已下载：%d' % num)
num += 1
return num
except FileNotFoundError:
os.mkdir(IMG_DIR)
return download_imgs(firends)
####-第三步-
process_imgs() 比较简单，主要是读取头像缓存目录，加工处理获取:
| 头像列表 | 头像总计数 | 列数 | 行数 |
|---|---|---|---|
| img_list | img_num | max_column | max_row |
列数与行数为总计数平方根向下取整，所以余出来的好友将会无情的被抛除于矩阵之外。
# 加工头像列表信息
def process_imgs():
# 头像列表
img_list = []
for root, dirs, files in os.walk(IMG_DIR):
for file in files:
if "jpg" in file and os.path.getsize(os.path.join(root, file)) > 0:
img_list.append(os.path.join(root, file))
img_num = len(img_list)
max_column = int(math.sqrt(img_num))
max_row = int(math.sqrt(img_num))
return img_list, img_num, max_column, max_row
####-第四步-
万事具备！就差这一哆嗦！
准备好合成图像保存路径 MERGE_PATH
设置好单个好友头像等比例压缩参数 HEAD_IMG_WIDTH 和 HEAD_IMG_HEIGHT
# 合成矩阵图保存路径(例如：A/B/C)
MERGE_PNG = '/heiheipiupiupiu.png'
MERGE_PATH = 'A/B/C' + MERGE_PNG
# 好友头像压缩(数字越大，合成图像质量越高)
HEAD_IMG_WIDTH = 80
HEAD_IMG_HEIGHT = 80
# 合成好友头像矩阵图
def merge_imgs():
img_list, img_num, max_line, max_row = process_imgs()
print(max_line, max_row, img_num)
num = 0
imgs_count = max_line * max_row
# 创建空白画板
white_board = Image.new('RGBA', (HEAD_IMG_WIDTH * max_line, HEAD_IMG_HEIGHT * max_row))
# 逐行列绘制好友头像
for i in range(0, max_row):
for j in range(0, max_line):
head_img = Image.open(img_list[num])
# 等比例压缩头像
temp_head_img = head_img.resize((HEAD_IMG_WIDTH, HEAD_IMG_HEIGHT))
# 绘画位置
place = (int(j % max_row * HEAD_IMG_WIDTH), int(i % max_row * HEAD_IMG_HEIGHT))
# 绘制粘贴
white_board.paste(temp_head_img, place)
num = num + 1
if num >= len(img_list):
break
if num >= imgs_count:
break
print(white_board.size)
white_board.save(MERGE_PATH)
####-第五步-
做事手脚干净些，清除好友头像缓存，不给机器内存添麻烦
# 清除好友头像缓存
def clean_imgs(path):
for img in os.listdir(path):
i = os.path.join(path, img)
if os.path.isfile(i):
os.remove(i)
####-第六步-
照例加上主函数，获取好友信息列表，使线程优雅地运转起来，泡上一杯清茶···
下载，呦呦呦
好嘞！
合成！走你～
清～除～
if __name__ == '__main__':
itchat.auto_login(hotReload=True)
#好友列表
firends = itchat.get_friends(update=True)
print('[···开始下载好友头像···]')
num = download_imgs(firends)
count = num - 1
print(u'[···头像下载完成，总计下载数量为：%d···]' % count)
print('[···开始合成好友头像···]')
merge_imgs()
print('[···合成好友头像完成···]')
print('[···开始清除好友头像缓存···]')
clean_imgs(IMG_DIR)
print('[···清除缓存完成···]')
小可爱，快返回你的合成图像保存路径下看看吧！拿出那张图片，你就可以去盆友圈**分(zhuang)享(bi)**了···
像这样：(保护好友隐私，另加深度模糊处理，真实程序暂不支持模糊特效噢)
有更多精彩或想提问题可扫下方二维码，关注后台私M。随缘回复，无量天尊
