Python利用Flask动态生成汉字头像

时间:2023-02-02 梦想橡皮擦人气:0

基础环境配置

再 Python 的 Flask 框架中，我们使用 Python 的图片处理库（例如 Pillow）来生成一个汉字图片。

首先看一个简单的示例，然后再此基础上，进行细节优化。

提前安装 PIL 模块

pip install pillow

接下来的视图函数和代码框架，可参考爬虫训练场系列博客中的任意一篇，这里橡皮擦在 app/apis 目录的 index.py 文件中进行操作。

建立基础代码结构，如下所示：

@api.route('/avatars')
def index():

    # 生成随机汉字
    def get_random_common_char():
        # 读取文件中的常用汉字

        with open('demo.txt', 'r', encoding='utf-8') as f:
            common_chars = f.read()
            # 去除空格
            common_chars = common_chars.replace(' ', '')
            common_chars = common_chars.strip()

            # 创建字符映射表
            translator = {ord(c): None for c in common_chars if unicodedata.category(c).startswith('P')}

            # 使用字符映射表去除标点符号
            s = common_chars.translate(translator)

        return random.choice(s)

	hanchar = get_random_char()

	# 生成图片
    image = Image.new('RGB', (200, 200), (255, 255, 255))
    draw = ImageDraw.Draw(image)
    font = ImageFont.truetype('arial.ttf', 36)
    draw.text((50, 50), char, font=font, fill=(0, 0, 0))

    # 将图片转换为二进制数据
    image_binary = image.tobytes()

    # 设置响应头，返回图片
    response = make_response(image_binary)
    response.headers['Content-Type'] = 'image/jpeg'
    return response

此时的代码是不能正确返回图片的，里面存在很多问题，需要一一进行修改。

加载本地 txt 文件，用于随机生成汉字

第一个问题 demo.txt 被我放置在 static 文件夹中，直接读取肯定是无法读取到的，需要用到 os 库。

# 获取根目录
app_root = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))

# 拼接 static 文件夹路径
static_path = os.path.join(app_root, 'static')

此时可以读取本地字体文件，代码如下所示：

os.path.join(static_path, 'demo.txt')

读取完毕，根目录变量 app_root 要放到内部函数外，后续还要使用。

由于我们需要两个汉字，所以使用最简单的字符串拼接，准备好 2 个汉字。

# 可以生成任意内容
han_char1 = get_random_common_char()
han_char2 = get_random_common_char()
han_char = han_char1 + han_char2

Pillow 生成图片

在 Flask 中生成图片非常简单，稍微麻烦的是直接将图片以二进制流格式返回到前端。

图片生成的代码如下所示。

# 生成图片
image = Image.new('RGB', (64, 64), (255, 255, 255))
draw = ImageDraw.Draw(image)
font = ImageFont.truetype(os.path.join(static_path, 'font/msyh.ttf'), 28)

上述代码使用了 Image.new() 方法，其语法结构如下所示。

Image.new(mode, size, color=0)

其各参数含义如下：

mode：图像模式。例如：‘RGB’、‘RGBA’、‘L’ 等。
size：图像大小，以像素为单位。例如：(800, 600)。
color：图像背景色。例如：(255, 255, 255)。

这里加载了本地字体文件 msyh.ttf，后续大家可以去 GitCode 进行下载。

除此之外，为了达到最终效果，还需要绘制一个同心圆。

# 绘制一个圆形
# 正方形的中心坐标
x = 32
y = 32
# 正方形的边长
length = 64
# 计算半径
r = length / 2
# 计算圆的左上角和右下角的坐标
left = x - r
top = y - r
right = x + r
bottom = y + r
draw.ellipse((left, top, right, bottom), fill=(0, 129, 255))

绘制文字到图片，注意调整居中参数。

draw.text((4, 12), han_char, font=font, fill=(255, 255, 255))

将图片传递到前端

这里我们使用 io 库的 BytesIO 类将图像转换成字节流，示例代码如下所示。

# 将图像转换成字节流
buf = io.BytesIO()
image.save(buf, format='png')
buf.seek(0)

return send_file(buf, mimetype='image/png')

除此之外，还使用了 send_file() 函数将字节流返回到前台。

运行完整代码，在浏览器访问指定接口，得到下述图片。

基于该内容，可以扩展更多参数，由用户自行配置，这里不再一一说明。

加载全部内容