Flask 如何在 Flask Web 框架中重命名上传的文件

在本文中，我们将介绍如何在 Flask Web 框架中重命名上传的文件。Flask 是一个基于 Python 的轻量级 Web 开发框架，提供了很多方便的功能，包括文件上传。但是，默认情况下，Flask 不会对上传的文件进行重命名，而是将其保存在原始的文件名下。如果我们想要给上传的文件起一个新的名称，就需要进行一些额外的操作。

阅读更多：Flask 教程

使用 Werkzeug 模块实现文件重命名

Flask 使用 Werkzeug 模块处理文件上传，而 Werkzeug 提供了一个方便的函数 secure_filename 来确保文件名的安全性。我们可以使用这个函数来生成一个安全的新文件名，并将上传的文件重命名保存。

from werkzeug.utils import secure_filename
from flask import Flask, request

app = Flask(__name__)

@app.route('/upload', methods=['POST'])
def upload_file():
    # 获取上传的文件
    file = request.files['file']

    # 生成新的文件名
    filename = secure_filename(file.filename)

    # 将文件重命名并保存
    file.save('uploads/' + filename)

    return '文件上传成功！'

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

在上面的例子中，我们首先导入了 secure_filename 函数和 Flask 模块。然后，我们创建了一个名为 upload_file 的视图函数，用于处理文件上传的请求。在该函数中，我们通过 request.files['file'] 获取上传的文件对象。然后，我们使用 secure_filename 函数生成一个安全的文件名。最后，我们将文件重命名并保存到指定的目录中。

需要注意的是，需要在运行 Flask 应用程序的目录中创建名为 uploads 的文件夹。而且，在进行文件上传之前，需要确保该文件夹的权限设置为可写。

自定义文件命名规则

除了使用 secure_filename 函数生成安全的文件名之外，我们还可以根据自己的需要定义文件命名规则。例如，我们可以使用当前时间戳作为文件名，以确保文件名的唯一性。

import time
from flask import Flask, request

app = Flask(__name__)

@app.route('/upload', methods=['POST'])
def upload_file():
    # 获取上传的文件
    file = request.files['file']

    # 生成新的文件名
    timestamp = str(int(time.time()))
    filename = timestamp + '.jpg'

    # 将文件重命名并保存
    file.save('uploads/' + filename)

    return '文件上传成功！'

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

上述例子中，我们导入了 time 模块，并在文件名中加入了当前的时间戳，然后将其作为新的文件名进行保存。

处理文件名冲突

在文件上传时，可能会遇到文件名冲突的问题。如果两个文件具有相同的文件名，但是内容不同，那么新上传的文件将会覆盖原有的文件。为了避免这种情况的发生，我们可以在文件名中加入一些随机数或者其他唯一标识符。

import uuid
from flask import Flask, request

app = Flask(__name__)

@app.route('/upload', methods=['POST'])
def upload_file():
    # 获取上传的文件
    file = request.files['file']

    # 生成新的文件名
    filename = str(uuid.uuid4()) + '.jpg'

    # 将文件重命名并保存
    file.save('uploads/' + filename)

    return '文件上传成功！'

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

在上面的例子中，我们使用了 uuid 模块生成了一个全局唯一的标识符，并将其作为新的文件名进行保存。

总结

在本文中，我们介绍了如何在 Flask Web 框架中重命名上传的文件。通过使用 Werkzeug 模块中的 secure_filename 函数，我们可以确保文件名的安全。另外，我们还可以根据自己的需求定义文件命名规则，并处理文件名冲突的问题。通过这些方法，我们可以灵活地对上传的文件进行重命名，以满足我们的需要。在实际开发中，我们可以根据具体的业务需求，选择合适的文件命名方式来进行处理。