Python批量生成特定尺寸图片及图画任意文字

Posted 2023-01-28 上帝De助手

因为工作需要生成各种大小的图片，所以写了个小脚本，顺便支持了下图画文字内容。具体代码如下：

from PIL import Image, ImageDraw, ImageFont
'''
    Auth: Xiaowu Chen
    Note: Please install [pillow] library before run this script.
'''

root = r'/root/path'

def draw_image(new_img, text, show_image=False):
    text = str(text)
    draw = ImageDraw.Draw(new_img)
    img_size = new_img.size
    draw.line((0, 0) + img_size, fill=128)
    draw.line((0, img_size[1], img_size[0], 0), fill=128)

    font_size = 40
    fnt = ImageFont.truetype('arial.ttf', font_size)
    fnt_size = fnt.getsize(text)
    while fnt_size[0] > img_size[0] or fnt_size[0] > img_size[0]:
        font_size -= 5
        fnt = ImageFont.truetype('arial.ttf', font_size)
        fnt_size = fnt.getsize(text)

    x = (img_size[0] - fnt_size[0]) / 2
    y = (img_size[1] - fnt_size[1]) / 2
    draw.text((x, y), text, font=fnt, fill=(255, 0, 0))

    if show_image:
        new_img.show()
    del draw


def new_image(width, height, text='default', color=(100, 100, 100, 255), show_image=False):
    new_img = Image.new('RGBA', (int(width), int(height)), color)
    draw_image(new_img, text, show_image)
    new_img.save('%s/%s_%s_%s.png' % (root, width, height, text))
    del new_img


def new_image_with_file(fn):
    with open(fn, encoding='utf-8') as f:
        for l in f:
            l = l.strip()
            if l:
                ls = l.split(',')
                if '#' == l[0] or len(ls) < 2:
                    continue

                new_image(*ls)


if '__main__' == __name__:
    new_image(400, 300, 'hello world any size', show_image=True)
    # new_image_with_file('image_data.txt')

执行后的效果如下：

 如果你需要批量的话，批量数据文件的格式如下：

#width,height,text
200,200,hello world
300,255,hello python

想学习更多Python知识，请扫描如下二维码！