Python拼接微信好友头像大图的实现方法


Posted in Python onAugust 01, 2018

基于 itchat 库来获取微信好友头像并执行拼接操作,对微信上文字化好友列表数据进行可视化展示。

Python拼接微信好友头像大图的实现方法

获取好友头像

def save_avatar(folder):
 """
 保存微信好友头像
 :param folder: 保存的文件夹
 """
 itchat.auto_login(hotReload=True)
 users = itchat.get_friends() or []
 print('%d friends found.' % len(users))
 if not os.path.exists(folder):
  os.makedirs(folder)
 index = 1
 for i, user in enumerate(users):
  nickname = user.RemarkName
  username = user.UserName
  file_path = os.path.join(folder, '%03d_%s.png' % (i, nickname))
  if not os.path.isfile(file_path): # 不重复下载
   avatar = itchat.get_head_img(username)
   with open(file_path, 'w') as f:
    f.write(avatar)
    print('Download %d: %s' % (index, file_path))
    index += 1

这里只需要传入一个保存头像的文件夹即可,运行 itchat.auto_login(hotReload=True) 后会弹出微信扫码界面让你授权微信登录,以便接下来的好友数据获取。

在图片下载时,我添加了一个防止重复下载的判断,以免多次运行时每次都要重新进行头像的下载。

取出待拼接头像

def get_image_files(folder, filters=None):
 """
 取出待拼接头像
 :param folder: 目标文件夹
 :param filters: 需要过滤的图片
 :return: 头像路径
 """
 filters = filters or []
 filenames = [os.path.join(folder, sub) for sub in os.listdir(folder)
     if sub.endswith('.png') and not filters.__contains__(sub)]
 return filenames

这里单独写个方法是为了把过滤的逻辑封装进来,以便于去掉指定的微信好友的头像(比如纯色的头像在拼接之后的大图看上去很明显,非强迫症可忽略)。

计算拼接的排列

def calculate_align_way(image_num, force_align=False):
 """
 计算图片排版对齐方式
 :param image_num: 图片数量
 :return: (rowls, columns)
 """
 actual_value = image_num ** 0.5
 suggest_value = int(actual_value)
 if actual_value == suggest_value or force_align:
  return suggest_value, suggest_value
 else:
  return suggest_value, suggest_value + 1

因为需要知道最终拼接图片的行列数,所有这里单独定义一个计算方法。算法就是直接对图片总数开根号,取出的结果如果正好是整数,就直接返回该结果。如果不是整数(大多数情况都如此),则根据参数 force_align 来决定是否强制进行正好全部铺满的显示。如果设为 True ,能强制铺满,但会有部分好友未显示完全;反之则是相对的情况。 后面发现拼接图片最后一行有很多黑色空位时,只需要更改该参数为True即可。

拼接

def join_images(image_files, rows, cols, width, height, save_file=None):
 """
 拼接操作
 :param image_files: 待拼接的图片
 :param rows: 行数
 :param cols: 列数
 :param width: 每张小头像的宽度
 :param height: 每张小头像的高度
 :param save_file: 拼接好图片的保存路径
 """
 canvas = np.ones((height * rows, width * cols, 3), np.uint8)
 for row in range(rows):
  for col in range(cols):
   index = row * cols + col
   if index >= len(image_files):
    break
   file_path = image_files[index]
   im = Image.open(file_path)
   im = im.resize((width, height))
   im_data = np.array(im)
   if len(im_data.shape) == 2:
    im_data = np.expand_dims(im_data, -1)
   x = col * width
   y = row * height
   canvas[y: y + height, x: x + width, :] = im_data
 image = Image.fromarray(canvas)
 image.show()
 if save_file:
  image.save(save_file)

拼接图片调用的是科学计算包 numpy 和图片库 PIL ,主要就是对 ndarray 进行操作。

最终将上面的步骤全部串联起来,执行如下主函数,便得到上面的拼接图片。

FOLDER = 'avatars'

if __name__ == '__main__':
 # 保存所有好友头像
 save_avatar(FOLDER)

 # 取到准备拼接的头像
 image_files = get_image_files(FOLDER)

 # 计算拼接的行列
 rows, columns = calculate_align_way(len(image_files), force_align=True)

 # 执行拼接操作
 join_images(image_files, rows, columns, 64, 64, 'result.png')

Github源码

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python实现带声音的摩斯码翻译实现方法
May 20 Python
Python的Flask框架应用程序实现使用QQ账号登录的方法
Jun 07 Python
win与linux系统中python requests 安装
Dec 04 Python
非递归的输出1-N的全排列实例(推荐)
Apr 11 Python
python3连接MySQL数据库实例详解
May 24 Python
对dataframe进行列相加,行相加的实例
Jun 08 Python
python 机器学习之支持向量机非线性回归SVR模型
Jun 26 Python
解决Python3 控制台输出InsecureRequestWarning问题
Jul 15 Python
pyhton中__pycache__文件夹的产生与作用详解
Nov 24 Python
Python爬虫解析网页的4种方式实例及原理解析
Dec 30 Python
Python实现转换图片背景颜色代码
Apr 30 Python
python单向链表实例详解
May 25 Python
Python Flask前后端Ajax交互的方法示例
Jul 31 #Python
Python常见内置高效率函数用法示例
Jul 31 #Python
Python使用jsonpath-rw模块处理Json对象操作示例
Jul 31 #Python
flask-socketio实现WebSocket的方法
Jul 31 #Python
深入flask之异步非堵塞实现代码示例
Jul 31 #Python
Django框架使用富文本编辑器Uedit的方法分析
Jul 31 #Python
Windows下python3.6.4安装教程
Jul 31 #Python
You might like
php对二维数组进行排序的简单实例
2013/12/19 PHP
30个php操作redis常用方法代码例子
2014/07/05 PHP
php使用Jpgraph绘制3D饼状图的方法
2015/06/10 PHP
php正则匹配文章中的远程图片地址并下载图片至本地
2015/09/29 PHP
使用PHPStorm+XDebug搭建单步调试环境
2017/11/19 PHP
php删除一个路径下的所有文件夹和文件的方法
2018/02/07 PHP
PHP数组对象与Json转换操作实例分析
2019/10/22 PHP
js创建元素(节点)示例
2014/01/02 Javascript
jQuery表单事件实例代码分享
2016/08/18 Javascript
基于SpringMVC+Bootstrap+DataTables实现表格服务端分页、模糊查询
2016/10/30 Javascript
BootStrap 实现各种样式的进度条效果
2016/12/07 Javascript
jquery插件bootstrapValidator表单验证详解
2016/12/15 Javascript
详解Node.js 命令行程序开发教程
2017/06/07 Javascript
浅谈vue的踩坑路
2017/08/31 Javascript
JavaScript中.min.js和.js文件的区别讲解
2019/02/13 Javascript
微信小程序 slot踩坑的解决
2019/04/01 Javascript
JavaScript语句错误throw、try及catch实例解析
2020/08/18 Javascript
python中查找excel某一列的重复数据 剔除之后打印
2013/02/10 Python
python脚本设置超时机制系统时间的方法
2016/02/21 Python
Python搭建FTP服务器的方法示例
2018/01/19 Python
python的scikit-learn将特征转成one-hot特征的方法
2018/07/10 Python
Python 带有参数的装饰器实例代码详解
2018/12/06 Python
Python实现串口通信(pyserial)过程解析
2019/09/25 Python
如何实现更换Jupyter Notebook内核Python版本
2020/05/18 Python
Python 实现进度条的六种方式
2021/01/06 Python
瑞典度假品牌:OAS
2019/05/28 全球购物
C++面试题目
2013/06/25 面试题
广告学专业推荐信范文
2013/11/23 职场文书
春季运动会广播稿大全
2014/02/19 职场文书
国培远程培训感言
2014/03/08 职场文书
爱与责任演讲稿
2014/05/20 职场文书
幼儿园开学家长寄语(2015秋季)
2015/05/27 职场文书
销售人员管理制度
2015/08/06 职场文书
使用SQL实现车流量的计算的示例代码
2022/02/28 SQL Server
CSS font-variation 可变字体的魅力(实例详解)
2022/03/03 HTML / CSS
table不让td文字溢出操作方法
2022/12/24 HTML / CSS