标签：code randint image random height width FastAPI.6

FastAPI.6

FastApi用pillow生成随机四位数字验证码图片

1. 导入模块

   import random
   from fastapi import FastAPI,Response
   from PIL import Image, ImageDraw, ImageFont
   from io import BytesIO
   

2. 代码

   def code():
   # 生成四位随机数字
       code = ""
       for _ in range(4):
           digit = str(random.randint(0, 9))
           code += digit
       return code
   
   
   @app.get('/code')
   def generate_image():
       number = code()
       width, height = 200, 100
       bg_color = (random.randint(10, 200), random.randint(20, 200), random.randint(50, 250))
       image = Image.new("RGB", (width, height), bg_color)
       draw = ImageDraw.Draw(image, 'RGB')
       # 如果没有Ceria Lebaran.otf字体文件，请替换为已有字体文件的路径
       font = ImageFont.truetype("Ceria Lebaran.otf", 80)
       num_particles = 200
       particle_size = 3  # 调整粒子大小
   
       # 随机绘制指定数量的粒子
       for _ in range(num_particles):
           x = random.randint(0, width-1)
           y = random.randint(0, height-1)
           # 随机选取颜色
           color = (random.randint(0, 255), random.randint(0, 255), random.randint(0, 255))
           # 在指定位置绘制像素点
           draw.ellipse([(x-particle_size, y-particle_size), (x+particle_size, y+particle_size)], fill=color)
   
       text_width, text_height = draw.textsize(number, font=font)
       x = (width - text_width) // 2
       y = (height - text_height) // 2
   
       draw.text((x, y), number, font=font, fill=(0, 0, 0))
       # 将生成的图片保存到内存中
       image_io = BytesIO()
       image.save(image_io, 'PNG')
       image_io.seek(0)
   
       image.save("verification_code.png")  # 可以在本地看见生成的图片
       return Response(content=image_io.getvalue(), media_type='image/png')
   

   

标签：code,randint,image,random,height,width,FastAPI.6
From： https://www.cnblogs.com/Zhang614/p/17679015.html