10分钟教你用python下载和拼接微信好友头像图片

短短的路走走停停

发布于 2019-05-14 19:17:39

6540

发布于 2019-05-14 19:17:39

文章被收录于专栏：程序猿声程序猿声

相信微信大家是用得再多也不过了。那么，对于python+微信，又能玩出什么新的花样呢？下面小编就给大家带来一个好玩的东西。用python下载所有的微信好友的头像，然后拼接成一张大图。这样，大家就可以转载到朋友圈装XY了。哈哈。

还是先给大家看看效果图如何。毕竟无图言X。

效果图：

PS：图侵删。这不是小编的微信哈……

好了废话不多说，看看我们是怎么一步一步做出来的吧。

下载微信好友头像

第一步当然是下载微信好友的头像，然后才能进行拼接。这里我们用到了itchat这个python的库。关于这个库，更多好玩的东西我们在后期会推出的，大家敬请关注and期待就好了。先看代码：

 1itchat.auto_login()
 2for friend in itchat.get_friends(update=True)[0:]:
 3    #可以用此句print查看好友的微信名、备注名
 4    print(friend['NickName']+"("+friend['RemarkName']+")")
 5    img = itchat.get_head_img(userName=friend["UserName"])
 6    path = headImgPath+friend['NickName']+"("+friend['RemarkName']+").jpg"
 7    try:
 8        with open(path,'wb') as f:
 9            f.write(img)
10    except Exception as e:
11        print(repr(e))

代码灰常简单。首先我们调用itchat.auto_login()进行微信登陆。

扫码登陆成功以后，我们再获取好友，然后获取头像，再配置一下文件保存路径，最后把头像写入文件里面。是不是简单到难以置信？

拼接好友头像

拼接我们用了PIL库里面的Image模块。代码看下面：

 1pathList = []
 2# os.listdir() 方法用于返回指定的文件夹包含的文件或文件夹的名字的列表。这个列表以字母顺序。 它不包括 '.' 和'..' 即使它在文件夹中。
 3for item in os.listdir(headImgPath):
 4    imgPath = os.path.join(headImgPath, item)
 5    pathList.append(imgPath)
 6
 7total = len(pathList)#total是好友头像图片总数
 8line = int(sqrt(total))#line是拼接图片的行数（即每一行包含的图片数量）
 9NewImage = Image.new('RGB', (128*line,128*line))
10x = y = 0
11for item in pathList:
12    try:
13        img = Image.open(item)
14        # Image.ANTIALIAS：平滑滤波。对所有可以影响输出像素的输入像素进行高质量的重采样滤波，以计算输出像素值。
15        img = img.resize((128,128),Image.ANTIALIAS)
16        NewImage.paste(img, (x * 128 , y * 128))
17        x += 1
18    except IOError:
19        print("第%d行,%d列文件读取失败！IOError:%s" % (y,x,item))
20        x -= 1
21    if x == line:
22        x = 0
23        y += 1
24    if (x+line*y) == line*line:
25        break
26NewImage.save("final.jpg")

我们先是用os.listdir获取了所有头像的路径保存到pathList里面。接着创建一个大图片，将大图片划分为n个小方格，再往小方格里面塞好友头像。需要注意的是，在塞好友头像前，需要对头像进行缩略变小才能塞进去。最后，生成最终文件。在当前目录下即可找到final.jpg。即为我们的成品。大家就可以拿着它去朋友圈装逼了~