python微信头像大集合

无意间见到朋友有人发了一张微信好友大集合的拼图,让我十分的眼气哈哈

这里用到了itchat这个python库,图片拼接用到了PIL库里面的Image模块。由于有些好友的名字有些太奇葩比如那些emjoy表情,一朵花,一个太阳。。。。用昵称和备注名拼接成图片名称会报错,我就改成了随机数。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
#_*_ coding:utf-8 _*_

import itchat
import os
import random
import string
from math import sqrt
from PIL import Image

#imgPath是存放好友头像图的文件夹的路径
headImgPath = './img/'

isExists = os.path.exists(headImgPath)
if not isExists:
os.makedirs(headImgPath)

itchat.auto_login()
for friend in itchat.get_friends(update=True)[0:]:
#可以用此句print查看好友的微信名、备注名
#print(friend['NickName']+"("+friend['RemarkName']+")")
img = itchat.get_head_img(userName=friend["UserName"])
#path = headImgPath+friend['NickName']+"("+friend['RemarkName']+").jpg"
salt = ''.join(random.sample(string.ascii_letters + string.digits, 8))
path = headImgPath+salt+".jpg"
try:
with open(path,'wb') as f:
f.write(img)
except Exception as e:
print(repr(e))

pathList = []
# os.listdir() 方法用于返回指定的文件夹包含的文件或文件夹的名字的列表。这个列表以字母顺序。 它不包括 '.' 和'..' 即使它在文件夹中。
for item in os.listdir(headImgPath):
imgPath = os.path.join(headImgPath, item)
pathList.append(imgPath)

total = len(pathList)#total是好友头像图片总数
line = int(sqrt(total))#line是拼接图片的行数(即每一行包含的图片数量)
NewImage = Image.new('RGB', (128*line,128*line))
x = y = 0
for item in pathList:
try:
img = Image.open(item)
# Image.ANTIALIAS:平滑滤波。对所有可以影响输出像素的输入像素进行高质量的重采样滤波,以计算输出像素值。
img = img.resize((128,128),Image.ANTIALIAS)
NewImage.paste(img, (x * 128 , y * 128))
x += 1
except IOError:
print("第%d行,%d列文件读取失败!IOError:%s" % (y,x,item))
x -= 1
if x == line:
x = 0
y += 1
if (x+line*y) == line*line:
break
NewImage.save("final.jpg")