python获取微信好友信息,制作头像照片墙集合!

image

模块:

  • itchat
  • PIL
  • os
  • math

登录微信,并获取好友头像,创建文件夹并保存:

itchat.auto_login(hotReload=True)
friends = itchat.get_friends(update=True)
if not os.path.exists('img'):
    os.mkdir('img')
# os.makedirs('img', exist_ok=True)
num = 0
# enumerate(friends)
for friend in friends:
    img = itchat.get_head_img(userName=friend['UserName'])
    print(friend)
    print(num)
    # urlretrieve(img, 'img'+'\\'+str(num))
    with open('img'+'/'+str(num)+'.png', 'wb') as f:
        f.write(img)
    num += 1

获取好友头像个数:

images = os.listdir('img')
# len(images) 为好友头像个数

自定义总面积除以好友头像个数,得到单张头像的面积,然后开根号得到一张图片的长于宽,长=宽:

# 每个头像的大小,长与宽
each_size = int(math.sqrt(640*640/len(images)))
# 每行可以容纳的照片个数
lines = int(640/each_size)
# 创建Image对象,初始化大小
image = Image.new('RGBA', (640, 640))

缩放头像比例为each_size然后进行粘贴,粘贴的方式为根据坐标:

x = 0
y = 0
for i in range(0, len(images)):
    # 打开头像

    img = Image.open('img'+'/'+str(i)+'.png')
    # 重新设置头像大小
    img = img.resize((each_size, each_size), Image.ANTIALIAS)
    # print(image.size)
    # 根据x,y坐标位置拼接图片
    image.paste(img, (x*each_size, y*each_size))
    # Image.NEAREST
    # 低质量
    # Image.BILINEAR
    # 双线性
    # Image.BICUBIC
    # 三次样条插值
    # Image.ANTIALIAS
    # 高质量
    x += 1
    # 一行一行的拼接
    if x == lines:
        # 如果一行满了 ,设置x=0
        x = 0
        # 进入下一行继续,y+1
        y += 1

最后粘贴完保存图片即可,并发送图片至文件助手:

image.save('img'+'/'+'all'+'.png')
file = '@fil@%s' % ('img'+'/'+'all'+'.png')
itchat.send(msg=file, toUserName='filehelper')

 

你可能感兴趣的:(python)