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 = '''
    <!DOCTYPE html>
    <form action="" method="post" enctype="multipart/form-data">
        请求序号: {} <br><br>
        下单日期: {} <br><br>
        接收邮箱: <input type="text" id="email" name="email"><br><br>
        需求描述: <textarea rows="5" cols="37" id="my_text" name="my_text" value=""></textarea><br><br>
        上传文件: <input type="file" name="file"><br><br>
        <b>----------------------------------------</b>
        <br> 注: 多个文件请压缩为 .rar .zip 压缩包后上传<br>
        <b>----------------------------------------</b><br><br>
        <input type="submit" value="提交订单请求">
    </form>
    '''
ref_html = '''
    <b>------------------------------------------------------</b>
    <br>您的订单已提交,感谢您的支持 <br>
    <b>------------------------------------------------------</b><br><br>
    付款订单号: {} <br><br>
    接收邮件地址: {} <br><br>
    请使用微信支付,并备注该订单号，处理完成后会第一时间邮件发送给您！ <br><br>
    ---> PS图片处理 10元 每张 <br>
    ---> 打印复印材料 0.5元 每张 -> 宿舍: 3号楼 2层 210自提 <br><br>
    <img style="width: 200px;" src="./static/web.jpeg" />
'''
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()
posted @ 2021-12-14 20:17 lyshark 阅读(792) 评论(0) 编辑收藏举报
刷新页面返回顶部
LyShark

Flask 实现简单的图片上传

公告
