Python Flask 实现云盘代码：含文件上传与类型检查及目录创建逻辑

```python # 导入所需模块 import os from flask import Flask, request, send_from_directory 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 '没有文件部分', 400 file = request.files['file'] if file.filename == '': return '没有选择文件', 400 if file and allowed_file(file.filename): filename = file.filename file.save(os.path.join(UPLOAD_FOLDER, filename)) return '文件上传成功', 200 else: return '不允许的文件类型', 400 @app.route('/files/<filename>') def uploaded_file(filename): return send_from_directory(UPLOAD_FOLDER, filename) if __name__ == '__main__': app.run(debug=True) ``` ### 需要注意的逻辑点： 1. 文件上传时，检查文件是否为空或未选择。 2. 确保上传的文件类型在允许的文件类型列表中，避免上传恶意文件。 3. 使用 `os.makedirs` 创建上传目录，确保即使目录不存在也能正常运行。 ######[AI写代码神器 | 351点数解答 | 2024-11-28 21:36:19]