微信爬取自己的朋友头像

from wxpy import *
import PIL.Image as Image
import os
import sys

#完成好友头像的获取及下载

def get_image():
    curr_dir=os.getcwd()
    if not os.path.exists(curr_dir+"Image/"):
        os.mkdir(curr_dir+"Image/")
    #登陆微信并获取好友信息
    bot=Bot()
    friends=bot.friends(update=True)
    #存储微信好友头像并存储在Image目录中
    n=0
    for friend in friends:
        friend.get_avatar(curr_dir+"Image/"+str(n)+".jpg")
        n+=1
#用于制作生成照片墙
def to_Photo_wall():
    #定义生成微信好友头像的尺寸
    image=Image.new("RGB",(650,650))
    #定义初始图片的位置
    x=0
    y=0
    #获取下载头像的文件
    ls=os.listdir(os.getcwd()+"Image/")
    #遍历文件夹的图片
    for file_names in ls:
        try:
            img=Image.open(os.getcwd()+"Image/"+file_names)
        except IOError:
            continue
        else:
            #重新设置图片的大小
            img=img.resize((50,50),Image.ANTIALIAS)
            image.paste(img,(x*50,y*50))
            x+=1
            if x==13:
                x=0;
                y+=1
    img=image.save(os.getcwd()+"friends.jpg")
if __name__=='__main__':
    get_image()
    to_Photo_wall()