Flask——flask-uploads(文件上传)+Pillow(图形验证码)

目录

1. 文件上传
2. 图形验证码

---

1. 文件上传

   ---

   # 安装： pip install flask-uploads
   # 配置
   from flask_uploads import UploadSet, IMAGES
   from flask_uploads import configure_uploads
   from flask_uploads import patch_request_class
   import os
   app.config['UPLOADED_PHOTOS_DEST'] = os.getcwd()
   app.config['MAX_CONTENT_LENGTH'] = 8 * 1024 * 1024
   # 创建上传对象
   photos = UploadSet('photos', IMAGES)
   # 配置上传对象
   configure_uploads(app, photos)
   # 配置上传文件大小，默认为64M，
   # 若设置为None，则以MAX_CONTENT_LENGTH配置为准
   patch_request_class(app, size=None)
   
   # 视图函数
   @app.route('/upload/', methods=['GET', 'POST'])
   def upload():
       img_url = None
       if request.method == 'POST':
       # 获取上传对象
       photo = request.files.get('photo')
       if photo:
       # 保存上传⽂件，返回⽂件名
       filename = photos.save(photo)
       # 根据⽂件名获取上传⽂件的URL
       img_url = photos.url(filename)
       return render_template('upload.html', img_url=img_url)

2. 图形验证码

   ---

   # pip install pillow
   import os
   from io import BytesIO
   from random import randint
   
   from PIL import Image, ImageFont, ImageDraw
   
   
   class VerifyCode:
       def __init__(self, width=100, height=40, size=4):
           self.width = width
           self.height = height
           self.size = size
           self.__code = ''  # 验证码字符串
           self.pen = None  # 画笔
   
       @property
       def code(self):
           return self.__code
   
       def generate(self):
           # 1)、创建画布
           im = Image.new("RGB", (self.width, self.height), self.__rand_color(150))
           self.pen = ImageDraw.Draw(im)
           # 2)、生成验证码字符串
           self.rand_string()
           # 3)、画验证码
           self.__draw_code()
           # 4)、画干扰点
           self.__draw_point()
           # 5)、画干扰线
           self.__rand_line()
           # 6)、返回验证码图片
           # 缓冲区
           buf = BytesIO()
           # 把图片放到缓冲区
           im.save(buf, 'png')
           # 获取图片的二进制
           res = buf.getvalue()
           buf.close()
           return res
           # im.save("vc.png")
   
       def __rand_color(self, min=0, max=255):
           return randint(min, max), randint(min, max), randint(min, max)
   
       # 验证码字符串
       def rand_string(self):
           self.__code = ''
           for i in range(self.size):
               self.__code += str(randint(0, 9))
   
       # 画验证码
       def __draw_code(self):
           # 加载字体
   
           path = os.path.join(os.getcwd(), 'static/fonts/SIMLI.TTF')
           print(path)
           font1 = ImageFont.truetype(path, size=20, encoding="utf-8")
   
           # 计算字符宽度
           width = (self.width - 20) // self.size
   
           # 逐个字符画
           for i in range(len(self.__code)):
               x = 13 + width * i  # 计算每个字符的x坐标
               self.pen.text((x, 9), self.__code[i], font=font1, fill=self.__rand_color(0, 80))
   
       # 画点
       def __draw_point(self):
           for i in range(100):
               self.pen.point((randint(1, self.width - 1), randint(1, self.height - 1)), self.__rand_color(30, 100))
   
       def __rand_line(self):
           for i in range(5):
               self.pen.line([(randint(1, self.width - 1), randint(1, self.height - 1)),
                              (randint(1, self.width - 1), randint(1, self.height - 1))], fill=self.__rand_color(50, 150),
                             width=2)
   
   
   # 单例属性
   vc = VerifyCode()
   
   if __name__ == '__main__':
       pass