Python写春联，给您拜年了

Posted 2022-02-13 香饽々

        “爆竹声中一岁除，春风送暖入屠苏。千门万户曈曈日，总把新桃换旧符。”

         又到农历新年，家家户户张灯结彩，扫除，贴春联，出处洋溢着喜庆和欢乐。又是Python大显身手的时刻，今天有请它来写春联。

        开始之前，先做一点准备工作：

• 下载自己喜欢的字体（本文所用字体为临时展示使用，版权归原作者） 
• 下载自己喜欢的背景，在此我选择的是传统的烫金红纸（临时展示使用，版权归原作者） 
• 安装freetype和numpy等第三方库【FreeType库是一个开源的、高质量且可移植的字体引擎；numpy为开源的数值计算扩展，可用来存储和处理大型矩阵】

        准备就绪，接下来，就看Python的了！

        完整代码如下：

#写春联

import freetype
import numpy as np
from PIL import Image

font = r'...'    #所需Truetype字体，请自行补齐存放文件路径
BG = r'...'    #所需背景图片，最好640*640像素，请自行补齐存放文件路径

def text2image(word, font, size, color)): 
    '''
    将字库相应汉字转换为图像（默认size=128，color=(0,0,0)）
    '''
      
    face = freetype.Face(font)
    face.set_char_size(size*size)    
    face.load_char(word)
    btm_obj = face.glyph.bitmap
    w, h = btm_obj.width, btm_obj.rows
    pixels = np.array(btm_obj.buffer, dtype=np.uint8).reshape(h, w)
    
    dx = int(face.glyph.metrics.horiBearingX/64)
    if dx > 0:
        patch = np.zeros((pixels.shape[0], dx), dtype=np.uint8)
        pixels = np.hstack((patch, pixels))
    
    r = np.ones(pixels.shape) * color[0] * 255
    g = np.ones(pixels.shape) * color[1] * 255
    b = np.ones(pixels.shape) * color[2] * 255
    im = np.dstack((r, g, b, pixels)).astype(np.uint8)
    
    return Image.fromarray(im)

def write_couplets(text, horv, quality):
    '''   
    text        - 春联字符串
    horv        - H-横排，V-竖排
    quality     - 单字分辨率，H-640像素，L-320像素
    '''
    
    size, tsize = (320, 128) if quality == 'L' else (640, 180)
    ow, oh = (size, size*len(text)) if horv == 'V' else (size*len(text), size)
    im_out = Image.new('RGBA', (ow, oh), '#f0f0f0')
    im_bg = Image.open(BG)
    if size < 640:
        im_bg = im_bg.resize((size, size))
    
    for i, w in enumerate(text):
        im_w = text2image(w, font, size=tsize, color=(0,0,0))
        w, h = im_w.size
        dw, dh = (size - w)//2, (size - h)//2
        
        if horv == 'V':
            im_out.paste(im_bg, (0, i*size))
            im_out.paste(im_w, (dw, i*size+dh), mask=im_w)
        else:
            im_out.paste(im_bg, (i*size, 0))
            im_out.paste(im_w, (i*size+dw, dh), mask=im_w)
    
    im_out.save('%s.png'%text)    #输出png图片，默认位置为我的文档   

left = '一帆风顺年年好'
right = '万事如意步步高'
central = '虎年大吉'
if __name__ == '__main__':
    write_couplets(left, horv='V', quality='H')
    write_couplets(right, horv='V', quality='H')
    write_couplets(central, horv='H', quality='H')

        代码部分借鉴大神天元浪子，在此顶礼膜拜！