本案例是 爬虫训练场 项目的衍生案例，用于渲染用户头像，大家也可以将该接口用于其它目的。
本文会使用随机汉字生成技术，如未掌握，可以查看 Python 随机生成一个汉字
文章目录
- 基础环境配置
- 加载本地 txt 文件，用于随机生成汉字
- Pillow 生成图片
- 将图片传递到前端
基础环境配置
再 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() 函数将字节流返回到前台。
运行完整代码，在浏览器访问指定接口，得到下述图片。
基于该内容，可以扩展更多参数，由用户自行配置，这里不再一一说明。
API 体验地址：爬虫训练场工具-头像生成器
📢📢📢📢📢📢
💗 你正在阅读 【梦想橡皮擦】 的博客
👍 阅读完毕，可以点点小手赞一下
🌻 发现错误，直接评论区中指正吧
📆 橡皮擦的第 823 篇原创博客
从订购之日起，案例 5 年内保证更新
- ⭐️ Python 爬虫 120，点击订购 ⭐️
- ⭐️ 爬虫 100 例教程，点击订购 ⭐️
