Flask 实现简单的图片上传工具

通过使用Flask框架实现一个简单的图片上传工具，当用户提交图片后会自动将图片保存到upload目录下，代码如下

import os
from flask import Flask, request
import random
import string
import datetime

ALLOWED_EXTENSIONS = set(['rar', 'zip' ,'xls','xlsx','docx','doc','html','png', 'jpg', 'jpeg', 'gif'])
app = Flask(__name__)
app.config['UPLOAD_FOLDER'] = "./upload_file"                    # 将文件上传到当前目录
app.config['MAX_CONTENT_LENGTH'] = 32 * 1024 * 1024              # 限制只能上传16M的文件

html = '''
    
    

        请求序号: {} 


        下单日期: {} 


        接收邮箱: 


        需求描述: 


        上传文件: 


        ----------------------------------------
        
 注: 多个文件请压缩为 .rar .zip 压缩包后上传

        ----------------------------------------


        
    
    '''
ref_html = '''
    ------------------------------------------------------
    
您的订单已提交,感谢您的支持 

    ------------------------------------------------------


    付款订单号: {} 


    接收邮件地址: {} 


    请使用微信支付,并备注该订单号，处理完成后会第一时间邮件发送给您！ 


    ---> PS图片处理 10元 每张 

    ---> 打印复印材料 0.5元 每张 -> 宿舍: 3号楼 2层 210自提 


    
'''
def GenPassword(length):
    chars=string.ascii_letters+string.digits
    return ''.join([random.choice(chars) for i in range(length)])
def allowed_file(filename):
    return '.' in filename and filename.rsplit('.', 1)[1] in ALLOWED_EXTENSIONS
@app.route('/', methods=['GET', 'POST'])
def upload_file():
    if request.method == 'GET':
        this_uuid = GenPassword(20)
        now_time = datetime.datetime.now().strftime('%Y-%m-%d %H:%M')
        return html.format(this_uuid,now_time)
    if request.method == 'POST':
        ud = GenPassword(5)
        try:
            file = request.files.get("file")
            email = request.form.get("email")
            my_text = request.form.get("my_text")
            if file == None or email == None or my_text == None:
                return "接口错误"
            if file and allowed_file(file.filename):
                file.save(os.path.join(app.config['UPLOAD_FOLDER'], "{}_{}".format(ud,file.filename)))
                print("*" * 200)
                print("订单号: {} ---> 接收邮箱: {} ---> 文件名: {}_{} ---> 需求: {} \n\n".format(ud,email,ud,file.filename, my_text))
                print("*" * 200)
                return ref_html.format(ud,email)
        except Exception:
            pass
    return "接口错误"

if __name__ == '__main__':
    app.run()