Flask Flask-RESTful – 上传图片

在本文中，我们将介绍如何使用Flask和Flask-RESTful来实现图片上传功能。Flask是一个Python的轻量级Web框架，而Flask-RESTful是一个用于构建RESTful API的扩展。

阅读更多：Flask 教程

准备工作

在开始之前，我们需要安装Flask和Flask-RESTful。可以使用以下命令进行安装：

pip install flask flask-restful

另外，我们还需要使用到Pillow库来处理图片。

pip install pillow

创建Flask应用

首先，我们需要创建一个Flask应用，并配置一些基本的设置。

from flask import Flask
from flask_restful import Api, Resource

app = Flask(__name__)
api = Api(app)
app.config['UPLOAD_FOLDER'] = 'uploads'

在以上代码中，我们创建了一个Flask应用和一个Flask-RESTful的API对象。我们还设置了一个文件上传的目录。

创建文件上传API

接下来，我们可以创建一个用于上传图片的API。

from werkzeug.utils import secure_filename
from flask import request
from PIL import Image

class UploadImage(Resource):
    def post(self):
        file = request.files['file']
        if file:
            filename = secure_filename(file.filename)
            file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
            return {'message': 'File uploaded successfully'}
        else:
            return {'message': 'No file uploaded'}

在以上代码中，我们定义了一个名为UploadImage的资源类，并且实现了post方法来处理上传图片的请求。首先，我们从请求中获取上传的文件，然后使用secure_filename方法来保证文件名的安全性。最后，我们将文件保存到指定的目录中，并返回一个成功上传的消息。

注册API路由

要让我们创建的API生效，我们需要将它注册到我们的Flask应用中。

api.add_resource(UploadImage, '/upload')

在以上代码中，我们使用了add_resource方法将UploadImage资源类注册到/upload路由上。

测试API

现在，我们可以使用Postman或者其他工具来测试我们的API了。

发送POST请求到http://localhost:5000/upload，并且选择一个图片文件进行上传。如果上传成功，我们将会得到一个包含成功消息的响应。

总结

在本文中，我们介绍了使用Flask和Flask-RESTful来实现图片上传功能。我们创建了一个Flask应用，并使用Flask-RESTful创建了一个处理图片上传的API。通过这个API，我们可以方便地让用户上传图片，并保存到指定的目录中。

通过学习本文，希望读者能够掌握使用Flask和Flask-RESTful进行图片上传的基本方法，为开发更强大的Web应用打下基础。