Python 生成微信头图

前两天个人朋友肖和我说 “完啦完啦，如今微信公众号无法弄了”。我就纳闷了，问她怎么回事。她告诉我 “黑洞图片事件你不知道，好多微信公众号号都有被告图片侵权，赔偿还很多。虽然我一直注意，可是有些图片根本不知道有没有版权啊。本身的小号，被告了可赔偿啊不起。” 我原本想告诉她，你想多了，就你那破号，可是这样怕是无法作朋友了。机智的我转说她 “太简单了，我帮你作个图片小程序吧，自动生成图片，一劳永逸”。肖立马说 “好啊，好啊，到时候赚钱了分你！” 因而就有了黑洞微信头图生成器。

使用 Pillow 处理图片

Pillow 是python的图片处理库，能够用来处理图片。核心代码大概以下:

from PIL import Image, ImageDraw, ImageFont

...

# 生成图片
image = Image.new("RGBA", (WIDTH, HEIGHT), background_color)
draw = ImageDraw.Draw(image)

# 设置字体大小
font = ImageFont.truetype(font_name, 20)
w, h = font.getsize(word)

# 文字垂直和水平居中
draw.text(((WIDTH - w) / 2, (HEIGHT - h) / 2), word, fill=font_color, font=font)
# 生成图片文件
image.save("output/{}_{}.png".format(filename, idx, "png"))
复制代码

核心功能都是使用 Pillow 提供的api，生成一个rgb图片，而后往上贴文本，步骤和Photoshop建立图片相似。建立的图片效果有:

英文会生成10种，中文生成4种。更多效果图，请见github/game404

使用 click 封装cli

python的命令行工具备 docopt, fire 和 click。 比较了一下，以为 click 更适合本项目。

@click.command(context_settings=CONTEXT_SETTINGS)
@click.option("--word", prompt="the word",
              help="图片文字")
@click.option("--bcolor", default="white", help="图片背景色，默认白色，支持颜色值['red', 'cyan', 'green', ...]及rgb hex值#ffffff")
@click.option("--fcolor", default="#2496ed",
              help="文字颜色，默认docker蓝，支持颜色值['red', 'cyan', 'green', ...]及rgb hex值#ffffff")
@click.option("--fpath", default=None, help="字体文件路径，默认空，使用预制字体。也许相对路径更合适。")
def generator(word, bcolor, fcolor, fpath):
    """ 黑洞图片生成器，协助生成微信公众号头图。自由不侵权，game404 出品。 python运行示例: python blackhole.py --word docker python blackhole.py --word docker --bcolor black --fcolor #ffffff python blackhole.py --word docker --fpath some.ttf Docker运行示例: docker run -it --rm game404/blackhole docker ... """
复制代码

运行效果以下:

(pywork-Y7-meC5z) C:\pywork\blackhole>python blackhole.py --help
Usage: blackhole.py [OPTIONS] 

  黑洞图片生成器，协助生成微信公众号头图。自由不侵权，game404 出品。

  python运行示例:

      python blackhole.py --word docker

      python blackhole.py --word docker --bcolor black --fcolor #ffffff

      python blackhole.py --word docker --fpath some.ttf

  Docker运行示例:

      docker run -it --rm game404/blackhole docker

      ...

Options:
  --word TEXT    图片文字
  --bcolor TEXT  图片背景色，默认白色，支持颜色值['red', 'cyan', 'green', ...]及rgb hex值#ffffff
  --fcolor TEXT  文字颜色，默认docker蓝，支持颜色值['red', 'cyan', 'green', ...]及rgb
                 hex值#ffffff
  --fpath TEXT   字体文件路径，默认空，使用预制字体。也许相对路径更合适。
  -h, --help     Show this message and exit.

复制代码

作完程序，我正在想怎么发给肖，封装成pypi package呢，仍是exe呢。 这时候旁边的肖恩顿，忽然插了一句。“直接编成docker镜像吧，多方便啊，还跨平台”。原来他都知道啊，但愿分账的消息他不知道。

应用docker镜像制做

Dockerfile 以下：

FROM python:3.7-alpine3.9
LABEL maintainer="shawn <studyoo@foxmail.com>"

WORKDIR /app

COPY requirements.txt /app

RUN apk add --no-cache jpeg-dev \
                       zlib-dev \
                       freetype-dev \
                       lcms2-dev \
                       openjpeg-dev \
                       tiff-dev \
                       tk-dev \
                       tcl-dev \
                       harfbuzz-dev \
                       fribidi-dev \
                       gcc \
                       musl-dev \
    && echo "[global] \ index-url = http://pypi.tuna.tsinghua/simple \ trusted-host = pypi.tuna.tsinghua \ timeout = 120 \ " > /etc/pip.conf \
    && pip install --upgrade pip \
    && pip install -r requirements.txt \
    && mkdir output

COPY . /app

ENTRYPOINT ["python","blackhole.py"]
复制代码

pillow涉及的dev包比较多，属于比较难装的package。

封装完成后，提交到docker hub，肖直接可使用:

docker run -it -v $PWD/output:/app/output --rm game404/blackhole --word docker
复制代码

和python方式运行相似，区别在于只用输入--word docker参数

总结

简单总结一下：

1. 使用pillow制做图片文件。
2. 使用click封装代码命令行工具。
3. 编写docker镜像方便使用。