76 lines
1.9 KiB
Python
76 lines
1.9 KiB
Python
#!/usr/bin/env python3.4
|
|
# encoding: utf-8
|
|
"""
|
|
Created on 19-3-19
|
|
@title: '微信好友头像拼接'
|
|
@author: Xusl
|
|
"""
|
|
import itchat
|
|
import math
|
|
import PIL.Image as Image
|
|
import os
|
|
|
|
import logging.config
|
|
|
|
from config import logger_path
|
|
|
|
logging.config.fileConfig(logger_path)
|
|
logger = logging.getLogger("root")
|
|
|
|
|
|
def photo_joint():
|
|
func_name = "好友头像拼接"
|
|
logger.info('start %s ' % func_name)
|
|
itchat.auto_login(hotReload=True)
|
|
itchat.dump_login_status()
|
|
friends = itchat.get_friends(update=True)[0:]
|
|
user = friends[0]["UserName"]
|
|
|
|
num = 0
|
|
pwd_path = os.path.abspath(os.path.dirname(os.getcwd()))
|
|
desc_photos = os.path.join(pwd_path, 'res/photos')
|
|
desc_full = os.path.join(pwd_path, 'res')
|
|
|
|
for i in friends:
|
|
img = itchat.get_head_img(userName=i["UserName"])
|
|
file_image = open(desc_photos + "/" + str(num) + ".jpg", 'wb')
|
|
file_image.write(img)
|
|
file_image.close()
|
|
num += 1
|
|
|
|
ls = os.listdir(desc_photos)
|
|
each_size = int(math.sqrt(float(640 * 640) / len(ls))) # 算出每张图片的大小多少合适
|
|
lines = int(640 / each_size)
|
|
image = Image.new('RGBA', (640, 640)) # 创建640*640px的大图
|
|
x = 0
|
|
y = 0
|
|
|
|
for i in range(0, len(ls) + 1):
|
|
try:
|
|
img = Image.open(desc_photos + "/" + str(i) + ".jpg")
|
|
except IOError:
|
|
print("Error")
|
|
else:
|
|
img = img.resize((each_size, each_size), Image.ANTIALIAS)
|
|
image.paste(img, (x * each_size, y * each_size)) # 粘贴位置
|
|
x += 1
|
|
if x == lines: # 换行
|
|
x = 0
|
|
y += 1
|
|
|
|
image.save(desc_full + "/好友头像拼接图.jpg")
|
|
itchat.send_image(desc_full + "/好友头像拼接图.jpg", 'filehelper') # 拼接完成发送给文件传输助手
|
|
logger.info('end %s ' % func_name)
|
|
|
|
|
|
def deal():
|
|
"""
|
|
处理入口
|
|
:return:
|
|
"""
|
|
photo_joint()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
deal()
|