每天一个 Python 小程序--0000

# -*- coding: utf-8 -*-
# 第0000题：将你的QQ头像（或者微博头像）右上角加上红色的数字，类似于微信未读信息数量那种提示。

# using PIL in http://www.lfd.uci.edu/~gohlke/pythonlibs/#pillow
from PIL import Image
from PIL import ImageFont
from PIL import ImageDraw


def write_number(image_file_path, number=1):
    img = Image.open(image_file_path)
    font_size = img.size[0] if img.size[0] < img.size[1] else img.size[1]
    font_size = font_size / 4
    number_txt = str(number) + ‘ ‘ if number < 100 else ‘99+‘
    font = ImageFont.truetype("arial.ttf", size=font_size)
    if font.getsize(number_txt)[0] > img.size[0] or font.getsize(number_txt)[1] > img.size[1]:
        return img
    position = img.size[0] - font.getsize(number_txt)[0]
    ImageDraw.Draw(img).text((position, 0), number_txt, (255, 0, 0), font)
    return img

write_number(‘0000.png‘).save(‘result.png‘)
write_number(‘0000.png‘, 100).save(‘result100.png‘)