Flask上传图片限制后缀|极客笔记

Flask上传图片限制后缀

在Web开发中，允许用户上传图片是一个非常常见的功能。然而，为了保证网站的安全性和稳定性，我们通常会对用户上传的内容进行一定的限制，比如限制上传的图片格式。本文将详细介绍如何使用Flask框架实现图片上传功能，并限制上传图片的格式。

准备工作

在开始之前，需要确保已经安装了Flask框架和相关的扩展库。可以使用以下命令安装：

pip install Flask
pip install Flask-Uploads

创建Flask应用

首先，我们需要创建一个基本的Flask应用，并配置上传文件的存储路径。具体代码如下：

from flask import Flask, request
from flask_uploads import UploadSet, configure_uploads, IMAGES

app = Flask(__name__)

# 配置上传文件的存储路径
app.config['UPLOADS_DEFAULT_DEST'] = '/path/to/uploaded_files'

photos = UploadSet('photos', IMAGES)
configure_uploads(app, photos)

@app.route('/upload', methods=['POST'])
def upload():
    if 'photo' in request.files:
        filename = photos.save(request.files['photo'])
        return f'File {filename} uploaded successfully!'
    else:
        return 'No file uploaded!'

在上面的代码中，我们首先引入了Flask和Flask-Uploads模块。然后配置了上传文件的存储路径为/path/to/uploaded_files，并创建了一个名为photos的上传集合，限制只能上传图片格式的文件。

接着，我们定义了一个路由/upload，当用户上传文件时，会调用upload函数进行处理。在upload函数中，我们首先检查用户是否上传了文件，如果上传了文件，则调用photos.save()方法保存文件，并返回上传成功的信息；如果没有上传文件，则返回提示信息。

限制上传图片的格式

为了限制用户上传图片的格式，我们可以通过设置ALLOWED_EXTENSIONS来限制允许上传的图片格式。具体代码如下：

ALLOWED_EXTENSIONS = {'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():
    if 'photo' in request.files:
        file = request.files['photo']
        if file and allowed_file(file.filename):
            filename = photos.save(file)
            return f'File {filename} uploaded successfully!'
        else:
            return 'Invalid file format!'
    else:
        return 'No file uploaded!'

在上面的代码中，我们定义了ALLOWED_EXTENSIONS变量，包含允许上传的图片格式。然后定义了allowed_file函数，用于检查用户上传的文件是否符合格式要求。在upload函数中，我们先获取用户上传的文件，然后通过allowed_file函数进行格式检查，只有通过检查的文件才会被保存。

运行Flask应用

最后，我们需要运行Flask应用，启动服务器，并监听指定端口。具体代码如下：

if __name__ == '__main__':
    app.run()

运行上述代码，Flask应用将会启动，监听默认端口，并等待用户上传图片。当用户上传图片时，会根据我们设置的限制条件对图片格式进行检查，并保存符合要求的图片文件。

示例代码

为了演示上述功能，我们可以编写一个简单的HTML页面，包含一个文件上传表单，以便用户上传图片。具体代码如下：

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Upload Image</title>
</head>
<body>
    <h1>Upload Image</h1>
    <form action="/upload" method="post" enctype="multipart/form-data">
        <input type="file" name="photo">
        <input type="submit" value="Upload">
    </form>
</body>
</html>

运行Flask应用后，访问上述HTML页面，选择一张图片文件上传，如果文件格式符合要求，即可看到上传成功的提示信息。

总之，通过Flask框架的帮助，我们可以很方便地实现图片上传功能，并限制上传图片的格式，确保网站的安全性和稳定性。