前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >Python获取微信好友头像生成点阵图片

Python获取微信好友头像生成点阵图片

作者头像
周小董
发布2019-03-25 10:09:18
1.6K0
发布2019-03-25 10:09:18
举报
文章被收录于专栏:python前行者

首先是点阵字的概念:点阵字体是把每一个字符都分成 n * n 个点,然后用每个点的虚实来表示字符的轮廓。点阵字体也叫位图字体,其中每个字形都以一组二维像素信息表示。

汉字那么多,总不能每个字都去自己设计点阵吧?别担心,有现成的点阵字库可以直接使用:HZK16字库。HZK即汉字库的首字母缩写,HZK16字库是符合GB2312标准的16×16点阵字库,支持的汉字有6763个(但可惜不支持英文和数字),每个汉字模型需要16×16一共需要256个点来显示。

这样思路就出来了:我们自己输入汉字,根据字符串中汉字字符编码,去HZK16字库中获取点阵信息,拿到信息后根据16*16点阵每个点的数据,print 出不同字符。

  • 获取微信好友头像生成点阵图片代码
代码语言:javascript
复制
# -*- coding:utf-8 -*-
import binascii,os,itchat
from PIL import Image


def char2bit(textStr):
    KEYS = [0x80, 0x40, 0x20, 0x10, 0x08, 0x04, 0x02, 0x01]
    target = []
    global count
    count = 0
    for x in range(len(textStr)):
        text = textStr[x]
        rect_list = [] * 16
        for i in range(16):
            rect_list.append([] * 16)

        gb2312 = text.encode('gb2312')
        hex_str = binascii.b2a_hex(gb2312)#字符串转16进制
        result = str(hex_str, encoding='utf-8')
        area = eval('0x' + result[:2]) - 0xA0
        index = eval('0x' + result[2:]) - 0xA0
        offset = (94 * (area-1) + (index-1)) * 32

        font_rect = None
        with open("HZK16", "rb") as f:
            f.seek(offset)
            font_rect = f.read(32)

        for k in range(len(font_rect) // 2):
            row_list = rect_list[k]
            for j in range(2):
                for i in range(8):
                    asc = font_rect[k * 2 + j]
                    flag = asc & KEYS[i]
                    row_list.append(flag)

        output = []
        for row in rect_list:
            for i in row:
                if i:
                    output.append('1')
                    count+=1
                    #print('0', end=' ')
                else:
                    output.append('0')
                    #print('.', end=' ')
            #print()

        target.append(''.join(output))
    return target

def getHeadImgs():
    #通过二维码登录微信网页版
    itchat.auto_login()
    #获取微信好友信息列表
    friendList = itchat.get_friends(update=True)

    #这里会用到的微信好友信息如下: User= {'UserName': '@8238e922e8be7356b1750c306cb75768','PYQuanPin': 'TED','NickName': 'TED'}
    #获取用户个人昵称,用于之后文件夹命名、用户头像命名
    if friendList[0]['PYQuanPin']:
        user = friendList[0]['PYQuanPin']
    else:
        user = friendList[0]['NickName']

    #先读取用户本人头像,存储名为用户名称
    selfHead = "{}/{}.jpg".format(os.getcwd(),user)
    with open(selfHead,'wb') as f:
        head = itchat.get_head_img(friendList[0]['UserName'])
        f.write(head)

    #创建文件夹用于存储好友头像
    if not os.path.exists(user):
        os.mkdir(user)
    #工作路径转到新建文件夹中
    os.chdir(user)
    #获取新建文件夹路径
    userspace = os.getcwd()
    #开始读取好友头像写入新建文件夹中
    print("开始读取%d位好友头像..."%(len(friendList)-1))
    for i in range(1,len(friendList)):
        if i % 100 ==0:
            print("已读取%d位好友头像,请耐心等待~"%i)
        try:
            friendList[i]['head_img'] = itchat.get_head_img(userName=friendList[i]['UserName'])
            friendList[i]['head_img_name'] = "%s.jpg" % friendList[i]['UserName']
        except ConnectionError:
            print('Fail to get %s' % friendList[i]['UserName'])

        with open(friendList[i]['head_img_name'],'wb') as f:
            f.write(friendList[i]['head_img'])
    print("读取好友头像完毕!")

    #保存头像的文件夹路径和用户本人头像路径返回
    return user, selfHead

def head2char(workspace,user,self,outlist):
    folder = "{}\\{}".format(workspace,user)
    #将工作路径转移至头像文件夹
    os.chdir(folder)
    #获取文件夹内头像列表
    imgList = os.listdir(folder)
    #获取头像图片个数
    numImages = len(imgList)
    #设置头像裁剪后尺寸
    eachSize = 100
    #变量n用于循环遍历头像图片,即当所需图片大于头像总数时,循环使用头像图片
    n=0
    #变量count用于为最终生成的单字图片编号
    count = 0
    #img = Image.open(self)

    #初始化颜色列表,用于背景着色:FFFACD黄色 #F0FFFF白  #BFEFFF 蓝  #b7facd青色 #ffe7cc浅橙色  #fbccff浅紫色 #d1ffb8淡绿 #febec0淡红 #E0EEE0灰
    colorlist = ['#FFFACD','#F0FFFF','#BFEFFF','#b7facd','#ffe7cc','#fbccff','#d1ffb8','#febec0','#E0EEE0']
    #index用来改变不同字的背景颜色
    index = 0
    #每个item对应不同字的点阵信息
    for item in outlist:
        #将工作路径转到头像所在文件夹
        os.chdir(folder)
        #新建一个带有背景色的画布,16*16点阵,每个点处填充2*2张头像图片,故长为16*2*100
        #如果想要白色背景,将colorlist[index]改为'#FFFFFF'
        canvas = Image.new('RGB', (3200, 3200), colorlist[index])  # 新建一块画布
        #index变换,用于变换背景颜色
        index = (index+1)%9

        count += 1
        #每个16*16点阵中的点,用四张100*100的头像来填充
        for i in range(16*16):
            #点阵信息为1,即代表此处要显示头像来组字
            if item[i] == "1":
                #循环读取连续的四张头像图片
                x1 = n % len(imgList)
                x2 = (n+1) % len(imgList)
                x3 = (n+2) % len(imgList)
                x4 = (n+3) % len(imgList)
                #以下四组try,将读取到的四张头像填充到画板上对应的一个点位置
                #点阵处左上角图片1/4
                try:
                    img = Image.open(imgList[x1])  # 打开图片
                except IOError:
                    print("有1位朋友的头像读取失败,已使用本人头像替代")  # 有些人没设置头像,就会有异常
                    img = Image.open(self)
                finally:
                    img = img.resize((eachSize, eachSize), Image.ANTIALIAS)  # 缩小图片
                    canvas.paste(img, ((i % 16) * 2 * eachSize, (i // 16) * 2 * eachSize))  # 拼接图片
                # 点阵处右上角图片2/4
                try:
                    img = Image.open(imgList[x2])  # 打开图片
                except IOError:
                    print("有1位朋友的头像读取失败,已使用本人头像替代")  # 有些人没设置头像,就会有异常
                    img = Image.open(self)
                finally:
                    img = img.resize((eachSize, eachSize), Image.ANTIALIAS)  # 缩小图片
                    canvas.paste(img, (((i % 16) * 2 + 1) * eachSize, (i // 16) * 2 * eachSize))  # 拼接图片
                # 点阵处左下角图片3/4
                try:
                    img = Image.open(imgList[x3])  # 打开图片
                except IOError:
                    print("有1位朋友的头像读取失败,已使用本人头像替代")  # 有些人没设置头像,就会有异常
                    img = Image.open(self)
                finally:
                    img = img.resize((eachSize, eachSize), Image.ANTIALIAS)  # 缩小图片
                    canvas.paste(img, ((i % 16) * 2 * eachSize, ((i // 16) * 2 + 1 ) * eachSize))  # 拼接图片
                # 点阵处右下角图片4/4
                try:
                    img = Image.open(imgList[x4])  # 打开图片
                except IOError:
                    print("有1位朋友的头像读取失败,已使用本人头像替代")  # 有些人没设置头像,就会有异常
                    img = Image.open(self)
                finally:
                    img = img.resize((eachSize, eachSize), Image.ANTIALIAS)  # 缩小图片
                    canvas.paste(img, (((i % 16) * 2 + 1) * eachSize, ((i // 16) * 2 + 1) * eachSize))  # 拼接图片
                #调整n以读取后续图片
                n= (n+4) % len(imgList)

        os.chdir(workspace)
        # 创建文件夹用于存储输出结果
        if not os.path.exists('{}_output'.format(user)):
            os.mkdir('{}_output'.format(user))
        os.chdir('{}_output'.format(user))
        #quality代表图片质量,1-100
        canvas.save('result%d.jpg'% count, quality=100)


if __name__=="__main__":
    #将想转化的字赋给字符串
    inpt = "生日快乐,事事顺心!"
    #将字转化为汉字库的点阵数据
    outlist = char2bit(inpt)
    print(outlist)
    #获取当前文件夹路径
    workspace = os.getcwd()
    #获取用户本人名称和用户本人头像路径
    user,self = getHeadImgs()
    # #将头像图片按点阵拼接成单字图片
    head2char(workspace,user,self,outlist)
    print("Well done!")
  • 根据制定图片生成点阵图片代码
代码语言:javascript
复制
# -*- coding:utf-8 -*-
import binascii,os
from PIL import Image


def char2bit(textStr):
    KEYS = [0x80, 0x40, 0x20, 0x10, 0x08, 0x04, 0x02, 0x01]
    target = []
    global count
    count = 0
    for x in range(len(textStr)):
        text = textStr[x]
        rect_list = [] * 16
        for i in range(16):
            rect_list.append([] * 16)

        gb2312 = text.encode('gb2312')
        hex_str = binascii.b2a_hex(gb2312)
        result = str(hex_str, encoding='utf-8')
        area = eval('0x' + result[:2]) - 0xA0
        index = eval('0x' + result[2:]) - 0xA0
        offset = (94 * (area-1) + (index-1)) * 32

        font_rect = None
        with open("HZK16", "rb") as f:
            f.seek(offset)
            font_rect = f.read(32)

        for k in range(len(font_rect) // 2):
            row_list = rect_list[k]
            for j in range(2):
                for i in range(8):
                    asc = font_rect[k * 2 + j]
                    flag = asc & KEYS[i]
                    row_list.append(flag)

        output = []
        for row in rect_list:
            for i in row:
                if i:
                    output.append('1')
                    count+=1
                    #print('0', end=' ')
                else:
                    output.append('0')
                    #print('.', end=' ')
            #print()

        target.append(''.join(output))
    return target

def head2char(workspace,folder,self,outlist):
    #将工作路径转移至头像文件夹
    os.chdir(folder)
    #获取文件夹内头像列表
    imgList = os.listdir(folder)
    #获取头像图片个数
    numImages = len(imgList)
    #设置头像裁剪后尺寸
    eachSize = 100
    #变量n用于循环遍历头像图片,即当所需图片大于头像总数时,循环使用头像图片
    n=0
    #变量count用于为最终生成的单字图片编号
    count = 0
    #img = Image.open(self)

    #初始化颜色列表,用于背景着色:FFFACD黄色 #F0FFFF白  #BFEFFF 蓝  #b7facd青色 #ffe7cc浅橙色  #fbccff浅紫色 #d1ffb8淡绿 #febec0淡红 #E0EEE0灰
    colorlist = ['#FFFACD','#F0FFFF','#BFEFFF','#b7facd','#ffe7cc','#fbccff','#d1ffb8','#febec0','#E0EEE0']
    #index用来改变不同字的背景颜色
    index = 0
    #每个item对应不同字的点阵信息
    for item in outlist:
        #将工作路径转到头像所在文件夹
        os.chdir(folder)
        #新建一个带有背景色的画布,16*16点阵,每个点处填充2*2张头像图片,故长为16*2*100
        #如果想要白色背景,将colorlist[index]改为'#FFFFFF'
        canvas = Image.new('RGB', (3200, 3200), colorlist[index])  # 新建一块画布
        #index变换,用于变换背景颜色
        index = (index+1)%9

        count += 1
        #每个16*16点阵中的点,用四张100*100的头像来填充
        for i in range(16*16):
            #点阵信息为1,即代表此处要显示头像来组字
            if item[i] == "1":
                #循环读取连续的四张头像图片
                x1 = n % len(imgList)
                x2 = (n+1) % len(imgList)
                x3 = (n+2) % len(imgList)
                x4 = (n+3) % len(imgList)
                #以下四组try,将读取到的四张头像填充到画板上对应的一个点位置
                #点阵处左上角图片1/4
                try:
                    img = Image.open(imgList[x1])  # 打开图片
                except IOError:
                    print("有1张图片读取失败,已使用备用图像替代")
                    img = Image.open(self)
                finally:
                    img = img.resize((eachSize, eachSize), Image.ANTIALIAS)  # 缩小图片
                    canvas.paste(img, ((i % 16) * 2 * eachSize, (i // 16) * 2 * eachSize))  # 拼接图片
                # 点阵处右上角图片2/4
                try:
                    img = Image.open(imgList[x2])  # 打开图片
                except IOError:
                    print("有1张图片读取失败,已使用备用图像替代")
                    img = Image.open(self)
                finally:
                    img = img.resize((eachSize, eachSize), Image.ANTIALIAS)  # 缩小图片
                    canvas.paste(img, (((i % 16) * 2 + 1) * eachSize, (i // 16) * 2 * eachSize))  # 拼接图片
                # 点阵处左下角图片3/4
                try:
                    img = Image.open(imgList[x3])  # 打开图片
                except IOError:
                    print("有1张图片读取失败,已使用备用图像替代")
                    img = Image.open(self)
                finally:
                    img = img.resize((eachSize, eachSize), Image.ANTIALIAS)  # 缩小图片
                    canvas.paste(img, ((i % 16) * 2 * eachSize, ((i // 16) * 2 + 1 ) * eachSize))  # 拼接图片
                # 点阵处右下角图片4/4
                try:
                    img = Image.open(imgList[x4])  # 打开图片
                except IOError:
                    print("有1张图片读取失败,已使用备用图像替代")
                    img = Image.open(self)
                finally:
                    img = img.resize((eachSize, eachSize), Image.ANTIALIAS)  # 缩小图片
                    canvas.paste(img, (((i % 16) * 2 + 1) * eachSize, ((i // 16) * 2 + 1) * eachSize))  # 拼接图片
                #调整n以读取后续图片
                n= (n+4) % len(imgList)

        os.chdir(workspace)
        # 创建文件夹用于存储输出结果
        if not os.path.exists('{}_输出'.format(user)):
            os.mkdir('{}_输出'.format(user))
        os.chdir('{}_输出'.format(user))
        #存储将拼接后的图片,quality为图片质量,1-100,100最高
        canvas.save('result%d.jpg'% count, quality=100)


if __name__=="__main__":
    #将想转化的字赋给字符串
    inpt = "二零一九新年快乐!"
    #将字转化为汉字库的点阵数据
    outlist = char2bit(inpt)
    #获取当前文件夹路径
    workspace = os.getcwd()
    #用于拼接的图片所在文件夹名称
    user = "周小董"
    #获取图片文件夹所在路径
    folder = "{}\\{}".format(workspace,user)
    #若读取图片失败,用于替代的备用图片路径
    self=workspace+"\\"+"周小董.jpg"
    #运行后将生成 user_输出 文件夹
    head2char(workspace,folder,self,outlist)
    print("Well done!")

参考:https://my.oschina.net/crossin/blog/3011967 git源码:https://github.com/pengfexue2/pic2char

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2019年02月21日,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档