帮我写一个文件上传代码
时间: 2024-02-01 11:36:48 浏览: 60
文件上传代码
这里提供一个简单的文件上传代码,使用Python的Flask框架实现:
```python
from flask import Flask, request, redirect, url_for
import os
app = Flask(__name__)
# 设置上传文件保存的目录
UPLOAD_FOLDER = 'uploads'
if not os.path.exists(UPLOAD_FOLDER):
os.makedirs(UPLOAD_FOLDER)
# 允许上传的文件扩展名
ALLOWED_EXTENSIONS = {'txt', 'pdf', 'png', 'jpg', 'jpeg', 'gif'}
# 检查文件扩展名是否允许上传
def allowed_file(filename):
return '.' in filename and \
filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
# 文件上传路由
@app.route('/upload', methods=['POST'])
def upload_file():
# 检查是否有文件上传
if 'file' not in request.files:
return redirect(request.url)
file = request.files['file']
# 检查文件是否符合要求
if file.filename == '':
return redirect(request.url)
if not allowed_file(file.filename):
return '不允许上传该类型的文件'
# 保存上传的文件
filename = file.filename
file.save(os.path.join(UPLOAD_FOLDER, filename))
return '文件上传成功'
if __name__ == '__main__':
app.run(debug=True)
```
在上传文件时,将文件POST到/upload路由即可。该代码会将上传的文件保存到uploads目录下,并检查文件扩展名是否符合要求。如果上传成功,会返回“文件上传成功”提示。
阅读全文