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

软件发布|下载排行|最新软件

当前位置:首页IT学院IT技术

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

梦想橡皮擦   2023-02-02 我要评论

基础环境配置

再 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() 函数将字节流返回到前台。

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

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

Copyright 2022 版权所有 软件发布 访问手机版

声明:所有软件和文章来自软件开发商或者作者 如有异议 请与本站联系 联系我们