Python利用Flask动态生成汉字头像
作者：梦想橡皮擦
基础环境配置
再 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() 函数将字节流返回到前台。
运行完整代码，在浏览器访问指定接口，得到下述图片。
基于该内容，可以扩展更多参数，由用户自行配置，这里不再一一说明。
到此这篇关于Python利用Flask动态生成汉字头像的文章就介绍到这了,更多相关Python Flask动态生成汉字头像内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！