情人节闲着无聊写个python程序

用python在图片上写字

'''
依赖，需要安装Pillow
pip install Pillow
'''
from PIL import Image, ImageDraw, ImageFont
import sys #判断参数个数，可以不要
import os  #判断文件是否存在，可以不要

image_path = "1.jpg"
font_path = "SIMLI.TTF"
text = "情人节快乐！"
font_size = 10


###########这几个判断用来支持命令行参数#######
if len(sys.argv)>1:
    path = sys.argv[1]
    if os.path.exists(path):
        image_path = path
        
if len(sys.argv)>2:
    path = sys.argv[2]
    if os.path.exists(path):
        font_path = path
        
if len(sys.argv)>3:
    text = sys.argv[3]
    
if len(sys.argv)>4:
    if sys.argv[4].isdigit():
        font_size = int(sys.argv[4])
#############################################

def generator_new_image(image_path, font_path, text, font_size):
    img_origin = Image.open(image_path)
    img_array = img_origin.load()
    img_new = Image.new("RGB", img_origin.size, (0,0,0))
    draw = ImageDraw.Draw(img_new)
    font = ImageFont.truetype(font_path, font_size)
    
    index = 0
    for y in range(0, img_origin.size[1], font_size):
        for x in range(0,img_origin.size[0], font_size):
            index = index % len(text)
            draw.text((x,y), text[index], font=font, fill=img_array[x,y], direction=None)
            index = index + 1
            
    img_new.convert("RGB").save("holiday.jpg")

generator_new_image(image_path, font_path, text, font_size)

效果