Python Flask 接收前端上传的图片

为了编写一个Python接口来接收前端上传的图片、将其存储到本地并返回存储地址,你可以使用Flask框架,这是一个轻量级的Web框架,非常适合快速搭建Web应用。

以下是一个简单的示例,展示了如何实现这个功能:

  1. 首先,确保你已经安装了Flask。你可以使用以下命令安装:

    pip install Flask
    
  2. 创建一个Python文件(例如 app.py),并编写以下代码:

from flask import Flask, request, jsonify
import os
import uuid

app = Flask(__name__)

# 配置上传文件夹和允许的文件扩展名
UPLOAD_FOLDER = 'uploads'
ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif'}
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER

# 确保上传文件夹存在
if not os.path.exists(UPLOAD_FOLDER):
    os.makedirs(UPLOAD_FOLDER)

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 part'}), 400
    
    file = request.files['file']
    
    # 如果用户没有选择文件,浏览器也会提交一个空的文件名,没有文件内容
    if file.filename == '':
        return jsonify({'error': 'No selected file'}), 400
    
    if file and allowed_file(file.filename):
        filename = str(uuid.uuid4()) + '_' + file.filename
        filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
        file.save(filepath)
        
        return jsonify({'message': 'File uploaded successfully', 'path': filepath}), 201
    else:
        return jsonify({'error': 'File type not allowed'}), 400

if __name__ == '__main__':
    app.run(debug=True)
  1. 运行你的Flask应用:

    python app.py
    
  2. 现在,你可以使用Postman或其他HTTP客户端发送一个POST请求到 http://127.0.0.1:5000/upload,并在请求中包含一个文件(字段名为 file)。

示例请求(使用Postman)

  • URL: http://127.0.0.1:5000/upload
  • Method: POST
  • Body: form-data
    • Key: file
    • Value: 选择你要上传的图片文件

响应示例

如果上传成功,你将收到一个JSON响应,其中包含成功信息和图片存储路径,例如:

{
    "message": "File uploaded successfully",
    "path": "uploads/xxxxxxx-xxxx-xxxx-xxxx-xxxxxxxxxxxx_yourimage.jpg"
}

如果上传失败(例如文件类型不允许或没有选择文件),你将收到一个包含错误信息的JSON响应。

这个示例展示了基本的文件上传功能,但在生产环境中,你可能需要添加更多的错误处理、安全性措施(如验证上传者身份、限制上传大小等)以及日志记录功能。

上一篇:机器学习与金融风控项目篇-day04-卡方分箱案例与模型的可解释性


下一篇:python实现投影仪自动对焦