fastapi实现图片验证码

FastAPI是一个基于Python的现代、快速（高性能）的Web框架，它可以用于构建高性能的Web应用程序和API。要实现图片验证码，你可以按照以下步骤使用FastAPI：

1. 导入所需的模块和库：

from fastapi import FastAPI
from fastapi.responses import HTMLResponse
from PIL import Image, ImageDraw, ImageFont
import random

2. 创建FastAPI应用程序实例：

app = FastAPI()

3. 定义生成验证码的函数：

def generate_code():
    # 生成随机的验证码
    code = ''.join(random.choices('0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ', k=4))
    return code

4. 定义生成验证码图片的函数：

def generate_image(code):
    # 创建一个空白图片
    image = Image.new('RGB', (120, 40), color=(255, 255, 255))
    draw = ImageDraw.Draw(image)
    
    # 设置字体和字体大小
    font = ImageFont.truetype('arial.ttf', size=30)
    
    # 在图片上绘制验证码
    draw.text((10, 5), code, font=font, fill=(0, 0, 0))
    
    return image

5. 定义路由和处理函数来生成和返回验证码图片：

@app.get('/captcha', response_class=HTMLResponse)
async def get_captcha():
    # 生成验证码
    code = generate_code()
    
    # 生成验证码图片
    image = generate_image(code)
    
    # 将图片保存为临时文件
    image_path = '/tmp/captcha.png'
    image.save(image_path)
    
    # 返回HTML响应，显示验证码图片
    return f'<img src="{image_path}" alt="Captcha">'

6. 运行FastAPI应用程序：

if __name__ == '__main__':
    import uvicorn
    uvicorn.run(app, host='0.0.0.0', port=8000)

你可以通过访问`http://localhost:8000/captcha`来获取生成的验证码图片。