欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Python之微信-微信好友头像合成

程序员文章站 2022-09-28 09:57:45
仔细看下图,你的头像就藏在里面哦!!! 有没有犯密集恐惧症?这并不震撼,如果你有 5000 位好友的话,做出来的图看着会更刺激些。 看完了图,你可能想知道这个图咋做出来的,不会是我闲着无聊把把好友头像一个个保存下来再用 PS 拼的吧? 自然不是了,Python 做的,是不是觉得没有 Python 干 ......

仔细看下图,你的头像就藏在里面哦!!!

Python之微信-微信好友头像合成

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

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

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

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

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

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

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

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

核心是利用以下两个库:

  1. pip --default-timeout=100 install -u wxpy
  2. pip --default-timeout=100 install -u pillow==4.2.1

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

完整代码如下:

from wxpy import *
import math
from pil import image
import os


# 创建头像存放文件夹
def create_file_path():
    avatar_dir = os.path.join(os.getcwd(), 'wechat')
    if not os.path.exists(avatar_dir):
        os.mkdir(avatar_dir)
    return avatar_dir


# 保存好友头像
def save_wx_avater(avatar_dir):
    # 初始化机器人,扫码登陆
    bot = bot(cache_path=true)
    friends = bot.friends(update=true)
    num = 0
    for friend in friends:

        friend.get_avatar(os.path.join(avatar_dir, f'{str(num)}.jpg'))
        print('好友昵称:%s' % friend.nick_name)
        num = num + 1


# 拼接头像
def joint_wx_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(os.path.join(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 exception as e:
                print(f"头像读取失败,错误:{e}")

    img = image.save(os.path.join(os.getcwd(), 'wechat.png'))
    print('微信好友头像拼接完成!')


if __name__ == '__main__':
    avatar_dir = create_file_path()
    save_wx_avater(avatar_dir)
    joint_wx_avatar(avatar_dir)

可以直接在运行程序文件,也可以用 pyinstaller 文件打包后运行,我这里就不介绍 pipenv+pyinstaller 打包的方法了,有兴趣的自行百度。

Python之微信-微信好友头像合成

以上就是用 python 制作微信好友图片墙的装逼指南。