要将PDF文件和图片文件上传到Flask中的不同上传文件夹,可以按照以下步骤操作:
from flask import Flask, request, redirect, url_for
from werkzeug.utils import secure_filename
import os
ALLOWED_EXTENSIONS = {'pdf', 'png', 'jpg', 'jpeg', 'gif'}
UPLOAD_FOLDER = os.path.join(os.getcwd(), 'pdf_files') # PDF文件夹路径
app = Flask(__name__)
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
def allowed_file(filename):
return '.' in filename and \
filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
@app.route('/upload', methods=['GET', 'POST'])
def upload_file():
if request.method == 'POST':
file = request.files['file']
if file and allowed_file(file.filename):
filename = secure_filename(file.filename)
file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
return '文件上传成功'
else:
return '不支持的文件类型'
return '''
<!doctype html>
<title>文件上传</title>
<h1>上传文件</h1>
<form method="POST" enctype="multipart/form-data">
<input type="file" name="file">
<input type="submit" value="上传">
</form>
'''
http://localhost:5000/upload
即可看到文件上传页面。选择一个PDF或图片文件进行上传。注意:上述代码只提供了基本的文件上传功能,没有进行文件大小、安全性等方面的验证。如果需要更复杂的功能,可以进一步完善代码。关于Flask的更多信息,请参考Flask官方文档。
腾讯云相关产品和产品介绍链接地址:
领取专属 10元无门槛券
手把手带您无忧上云