获取微信好友头像并拼成大图(Python, PIL, Itchat)

如题的一个小程序,做与学结合,只有动手,才能更快的掌握新的语言

最终会生成一个这样的图片(保护隐私,已做模糊处理):


用到的Python包:

  1. Pillow Python的图像处理库,PIL的一个Fork版本
  2. Itchat 一个封装好的微信接口库

源代码

一 总体流程

  1. 使用Itchat登录微信并获取所有好友的头像图片地址保存到images/文件夹下
  2. 假设图片数量为n*n,使用Pillow创建一个n行,n列的正方形底图
  3. 使用Pillow缩放每张头像图片到200px*200px的大小,那么第二步的底图大小为width=n*200px,height=n*200px
  4. 将所有缩放后的头像图片按照顺序粘贴到第二步的底图上,完成

二 代码

未安装itchat和Pillow的需要先安装这两个包,我用的是Anaconda(一个python的包控制软件,可以在安装python包的时候把所有依赖补全),用pip的同学可能在安装pillow的时候需要解决一些依赖问题。

conda install itchat
conda install pillow

首先登陆微信并获取所有好友的头像图片地址并下载到images/文件夹下

#导入itchat包
import itchat

#下载所有好友头像并存储
def download_images(frined_list):
    image_dir = "./images/"
    #计数器,保存每一个头像图片名为 num.jpg
    num = 1
    for friend in frined_list:
        #赋值当前需要保存的图片的名称
        image_name = str(num)+'.jpg'
        num+=1
        #使用itchat自带函数get_head_img获取好友头像图片的二进制流
        #friend["UserName"]为当前好友的唯一标识符
        img = itchat.get_head_img(userName=friend["UserName"])
        #将图片二进制流img变量写入到images/文件夹下对应jpg文件
        with open(image_dir+image_name, 'wb') as file:
            file.write(img)

if __name__ == '__main__':
    #自动登陆,会出现二维码,扫码确认后登陆微信
    itchat.auto_login()
    #get_friends获取所有好友信息函数,返回list存储到friends变量
    friends = itchat.get_friends(update=True)
    #把好友信息传入下载图片函数
    download_images(friends)

现在在当前目录下会有一个images/文件夹,里面保存了所有微信好友的头像图片。因为需要一张一张下载,所以这个过程可能时间会比较长需要几分钟。

接着,把所有头像图片合并:

from PIL import Image
import os
import math

#合并所有好友头像
#path为存储头像图像的文件夹相对于当前路径的相对路径,这里应该为'images/'
def merge_images(path):
    print("Merging head images......")

    #设置每个图片需要缩放到的大小
    photo_width = 200
    photo_height = 200

    #保存所有本地图片的绝对地址
    photo_list = []
    #头像图片文件夹的绝对路径
    dirName = os.getcwd()+path

    #os.walk用来遍历某一个文件夹下的所有文件夹和文件,递归便利,os是python自带库
    #具体参数用法参考python手册
    for root, dirs, files in os.walk(dirName):
        for file in files:
            #遍历所有文件,如果文件名包含jpg则获取该文件绝对路径添加到photo_list
            #os.path.join(root, file)拼接为这个文件的绝对路径
            if "jpg" in file and os.path.getsize(os.path.join(root, file)) > 0:
                photo_list.append(os.path.join(root, file))

    pic_num = len(photo_list)
    #合并图片的列数
    line_max = int(math.sqrt(pic_num))
    #合并图片的行数
    row_max = int(math.sqrt(pic_num))
    print(line_max, row_max, pic_num)

    #如果好友太多行数大于20行则限制为20行
    if line_max > 20:
        line_max = 20
        row_max = 20
    
    num = 0
    #需要合并的图片总数
    pic_max=line_max*row_max

    #新建底图,长款为行数*200px,列数*200px
    toImage = Image.new('RGBA',(photo_width*line_max, photo_height*row_max))

    #循环粘贴每一个头像图片
    for i in range(0,row_max):
        for j in range(0,line_max):
            #读取对应的头像图片
            pic_fole_head = Image.open(photo_list[num])
            #把图片伸缩到设置的大小(200px*200px)
            tmppic = pic_fole_head.resize((photo_width,photo_height))
            #计算图片粘贴的位置
            loc = (int(j%row_max*photo_width),int(i%row_max*photo_height))
            #把头像图片粘贴到底图对应位置
            toImage.paste(tmppic,loc)
            num = num+1
            if num >= len(photo_list):
                break
        if num >= pic_max:
            break
    print(toImage.size)
    #保存图片
    toImage.save('merged.png')
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 195,585评论 5 462
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 82,283评论 2 373
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 142,760评论 0 324
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 52,461评论 1 266
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 61,280评论 4 357
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 46,268评论 1 273
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 36,656评论 3 385
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 35,322评论 0 253
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 39,629评论 1 293
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 34,691评论 2 312
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 36,445评论 1 326
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,299评论 3 313
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 37,694评论 3 299
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 28,982评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,244评论 1 251
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 41,642评论 2 342
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 40,829评论 2 335

推荐阅读更多精彩内容