炫技!一键拼出你的微信好友图片墙

每个人的好友少者几百,多者三五千。学Python这么久,有没有想过用Python一键拼出你的好友图,做一些有意思的项目,是不是很有成就感,你就会发现学Python真心有用的,我们先来看一下震撼的效果图。 

炫技!一键拼出你的微信好友图片墙

有没有犯密集恐惧症?这并不震撼,如果你有 5000 位好友的话,做出来的图看着会更刺激些。

看完了图,你可能想知道这个图咋做出来的,不会是我闲着无聊把把好友头像一个个保存下来再用 PS 拼的吧?

自然不是了,Python 做的,是不是觉得没有 Python 干不了的事儿。其实,这种图很早就有人玩过了,不过下面还是来说说怎么做出来,这样你可以做一个自己的微信好友图片墙,拿去朋友圈装装逼。

有两种方法,一种简单的,不用接触 Python 代码,一种稍微复杂点,需要写代码。

先说简单的方法,只需要两步:运行程序然后扫微信二维码就行了。剩下的交给程序自己蹦Q,泡杯茶在电脑前等待几分钟左右就可以得到图片,具体的等待时间视微信好友数量而不同,我近 2000 好友,用时 10 分钟左右。

一个简单的操作示意图:

炫技!一键拼出你的微信好友图片墙

几分钟后就可以得到上面的图片了。

其实到这儿就完了,是不是很简单。

你要感兴趣怎么实现的,可以往下看用 Python 代码怎么实现的,代码不长,60 行就可以搞定。

核心是利用三个个库:

  • wxpy 库,用于获取好友头像然后下载
  • Pillow 库,用于拼接头像
  • Pyinstaller 库,用来打包 Python 程序成 exe 文件

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

完整代码如下:

# -*- coding: utf-8 -*- 


 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 字样表示程序打包成功: