Flask—04-文件上传与邮件发送(自带优化)

北城以北 提交于 2021-02-13 13:57:54

文件上传与邮件发送

  • 可以按照标题分别直接粘贴对应的文件夹,运行直接用:

    原生上传

  • 模板文件

    <form method="post" enctype="multipart/form-data">
        <input type="file" name="photo" /><br />
        <input type="submit" value="上传" />
    </form>
    
  • 视图函数

    @app.route('/upload/', methods=['GET', 'POST'])
    def upload():
        if request.method == 'POST':
            # 获取上传对象
            photo = request.files.get('photo')
            if photo:
                # 保存上传文件,参数是文件保存的路径名
                photo.save(photo.filename)
                return '文件上传成功'
            return '文件上传失败'
        return render_template('upload.html')
    
  • 优化完善

    import os
    from flask import send_from_directory
    
    # 上传文件保存位置
    app.config['UPLOADED_FOLDER'] = os.path.join(os.getcwd(), 'static/upload')
    # 请求大小(文件大小限制)
    app.config['MAX_CONTENT_LENGTH'] = 1024 * 1024 * 8
    
    # 判断是否是允许的文件类型
    def allowed_file(filename):
        return '.' in filename and filename.rsplit('.', 1)[1] in ALLOWED_SUFFIX
    
    # 展示上传的文件
    @app.route('/uploaded/<filename>')
    def upladed(filename):
        # 安全的发送文件
        return send_from_directory(app.config['UPLOADED_FOLDER'], filename)
    
    @app.route('/upload/', methods=['GET', 'POST'])
    def upload():
        img_url = None
        if request.method == 'POST':
            # 获取上传对象
            photo = request.files.get('photo')
            # 保存前验证文件的类型
            if photo and allowed_file(photo.filename):
                # 拼接文件保存的完整路径名
                pathname = os.path.join(app.config['UPLOADED_FOLDER'],
                                        photo.filename)
                # 保存上传文件,参数是文件保存的路径名
                photo.save(pathname)
                img_url = url_for('upladed', filename=photo.filename)
        return render_template('upload.html', img_url=img_url)
    

    优化:大小限定、类型限定、保存位置、查看上传文件

flask-uploads

  • 说明:极大地优化了文件上传的操作,使用非常方便

  • 安装: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 = Flask(__name__)
    # 设置文件的大小:8M=8*1024*1024k
    app.config['MAX_CONTENT_LENGTH'] = 1024 * 1024 * 8
    app.config['UPLOADED_PHOTOS_DEST'] = os.path.join(os.getcwd(), 'static/upload')
    # 创建上传对象,指定名字和上传文件的类型
    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:
                # 保存上传文件
                photos.save(photo)
                # 获取上传文件的地址
                img_url = photos.url(photo.filename)
        return render_template('upload.html', img_url=img_url)
    
    • 优化上传:生成随机文件名,生成缩略图(PIL模块,只支持py2,py3中使用需要安装pillow)
    # 生成随机字符串
    def random_string(length=32):
        import random
        base_str = 'abcdefghijklmnopqrstuvwxyz1234567890'
        return ''.join(random.choice(base_str) for i in range(length))
    
    @app.route('/upload/', methods=['GET', 'POST'])
    def upload():
        img_url = None
        if request.method == 'POST':
            # 获取上传对象
            photo = request.files.get('photo')
            if photo:
                # 提取文件后缀
                suffix = os.path.splitext(photo.filename)[1]
                # 生成随机文件名
                filename = random_string() + suffix
                # 保存上传文件
                photos.save(photo, name=filename)
                # 生成缩略图
                from PIL import Image
                # 拼接完整路径名
                pathname = os.path.join(app.config['UPLOADED_PHOTOS_DEST'], filename)
                # 打开文件
                img = Image.open(pathname)
                # 设置大小
                img.thumbnail((64, 64))
                # 保存修改后的图片
                img.save(pathname)
                # 获取上传文件的地址
                img_url = photos.url(filename)
        return render_template('upload.html', img_url=img_url)
    

综合使用

  • 要求:结合flask-bootstrap、flask-wtf、flask-uploads等实现文件上传
  • 使用:
    • 各种配置及对象创建
    • 注意对象创建的先后顺序
    • 上传文件的校验处理过程与上面一样

flask-mail

  • 说明:专门用来实现邮件发送的扩展库,使用非常方便。

  • 安装:pip install flask-mail

  • 使用:

    from flask_mail import Mail, Message
    import os
    
    # 邮件配置,一定要放在对象创建之前
    # 邮箱服务器:不同类型邮箱服务器不同,不知道的请百度
    app.config['MAIL_SERVER'] = 'smtp.xxx.com'
    # 用户名
    app.config['MAIL_USERNAME'] = 'xxx@xxx.com'
    # 密码:如果密码添加到了环境变量可以直接引用'MAIL_PASSWORD',否则直接把'123456'换成自己的邮箱密码。
    app.config['MAIL_PASSWORD'] = os.getenv('MAIL_PASSWORD', '123456')
    
    # 创建发送邮件的对象
    mail = Mail(app)
    
    @app.route('/send/')
    def send():
        # 创建邮件消息对象
        msg = Message('账户激活',
                      recipients=['xxx@qq.com'],
                      sender=app.config['MAIL_USERNAME'])
        # 设置邮件内容
        msg.html = '恭喜你,中奖了!'
        # 发送邮件
        mail.send(msg)
        return '邮件发送成功'
    
  • 总结:

    • 书写邮箱相关配置
    • 创建发送邮件的(Mail)对象
    • 创建邮件消息(Message)对象
    • 使用mail的send方法发送邮件
  • 封装函数发送邮件

    # 封装函数发送邮件
    def send_mail(subject, to, template, **kwargs):
        # 处理邮件接收者
        if isinstance(to, list):
            recipients = to
        elif isinstance(to, str):
            recipients = to.split(',')
        else:
            raise Exception('邮件接收者参数有误')
        # 创建邮件消息对象
        msg = Message(subject=subject, recipients=recipients,
                      sender=app.config['MAIL_USERNAME'])
        # 设置邮件内容
        msg.html = render_template(template, **kwargs)
        # 发送邮件
        mail.send(msg)
    
    @app.route('/send/')
    def send():
        # 调用函数即可发送邮件
        send_mail('账户激活', 'xxx@qq.com', 'activate.html', name='八戒')
        return '邮件发送成功'
    
  • 异步发送邮件

    from flask import current_app
    from threading import Thread
    
    # 异步发送邮件
    def async_send_mail(app, msg):
        # 邮件发送必须在程序上下文中进行
        # 新的线程没有上下文,因此需要手动创建上下文
        with app.app_context():
            mail.send(msg)   
    
    # 封装函数发送邮件
    def send_mail(subject, to, template, **kwargs):
          # 处理邮件接收者
        if isinstance(to, list):
            recipients = to
        elif isinstance(to, str):
            recipients = to.split(',')
        else:
            raise Exception('邮件接收者参数有误')
        # 从代理中获取原始对象
        app = current_app._get_current_object()
        # 创建邮件消息对象
        msg = Message(subject=subject,recipients=recipients,sender=app.config['MAIL_USERNAME'])
        # 设置邮件内容
        msg.html = render_template(template, **kwargs)
        # 发送邮件:同步发送,会阻塞运行
        # mail.send(msg)
        # 创建一个线程,在新的线程中发送邮件
        thr = Thread(target=async_send_mail, args=(app, msg))
        # 启动线程
        thr.start()
        # 返回线程
        return thr
    
  • QQ邮箱发送配置

    # 密码,QQ邮箱需要使用授权码
    app.config['MAIL_PASSWORD'] = os.getenv('MAIL_PASSWORD', '123456')
    # QQ邮箱配置
    # app.config['MAIL_PORT'] = 465
    # 加密传输
    # app.config['MAIL_USE_SSL'] = True
    
标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!