上传文件后端代码
时间: 2023-11-12 19:15:20 浏览: 109
上传文件的后端代码可以使用以下的 Python Flask 框架实现:
```python
from flask import Flask, request, jsonify
import os
app = Flask(__name__)
# 设置上传文件的保存路径
app.config['UPLOAD_FOLDER'] = './uploads'
# 设置允许上传的文件类型
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 jsonify({'error': 'No file uploaded!'})
file = request.files['file']
# 检查文件名是否合法
if file.filename == '':
return jsonify({'error': 'No selected file!'})
if not allowed_file(file.filename):
return jsonify({'error': 'Invalid file type!'})
# 将文件保存到指定目录
filename = os.path.join(app.config['UPLOAD_FOLDER'], file.filename)
file.save(filename)
return jsonify({'success': 'File uploaded successfully!'})
if __name__ == '__main__':
app.run()
```
该代码定义了 `/upload` 路由来处理文件上传请求。当请求中包含文件时,先检查文件名是否合法,然后将文件保存到指定的目录中。最后返回上传成功的消息。
阅读全文