itchat+pillow实现微信好友头像爬取和拼接

itchat+pillow实现微信好友头像爬取和拼接

from numpy import *
import itchat
import urllib
import requests
import os
from PIL import Image
from os import listdir
import math

itchat.auto_login(hotReload=True)

friends = itchat.get_friends(update=True)[0:]

user = 'images'
if os.path.exists(user) == False:
os.mkdir(user)

num = 1
for i in friends:
img = itchat.get_head_img(userName=i["UserName"])
fileImage = open(user + "/" + str(num) + ".jpg",'wb')
fileImage.write(img)
fileImage.close()
num += 1

pics = listdir(user)

numPic = len(pics)
print(numPic)

eachsize = int(math.sqrt(float(640 * 640) / numPic))

print(eachsize)

numline = int(640 / eachsize)

toImage = Image.new('RGB', (640, 640))

print(numline)

x = 0
y = 0

for i in pics:
try:

    img = Image.open(user + "/" + i)
except IOError:
    print("Error: 没有找到文件或读取文件失败"+i)
else:
    #缩小图片
    img = img.resize((eachsize, eachsize), Image.ANTIALIAS)
    #拼接图片
    toImage.paste(img, (x * eachsize, y * eachsize))
    x += 1
    if x == numline:
        x = 0
        y += 1

toImage.save(user + ".jpg")
itchat.send_image(user + ".jpg", 'filehelper')