使用 Flask 上传文件时，我应该如何处理重复的文件名

Question

我最近遇到了这个问题并回答了https://stackoverflow.com/a/44926557/12322095 关于 Flask 文件上传。

在我再次上传同名图片之前，这一切都很好。它没有改变图像或覆盖它。

我的问题是，如果用户上传了同名的图片，我们有什么办法可以显示错误消息或自动将名称更改为其他名称。

对于自动更改名称，我研究过，可以通过 resolve_conflict 完成，但我不明白如何实现。

Answer 1

我的代码同上作为参考

在保存文件之前，您需要创建某种 uniuqe ID 以附加到文件名。

这可以通过以下方式完成：

from uuid import uuid4
def make_unique(string):
    ident = uuid4().__str__()
    return f"{ident}-{string}"

将唯一的 UUID 添加到字符串的开头：

>>> make_unique('something.txt')
'f374b80c-b09e-468f-adc6-3d9df175cee7-something.txt'

要在上传代码中使用它，只需在保存之前通过该函数运行文件名。请务必先通过secure_filename 函数输入文件名：

        if file and allowed_file(file.filename):
            original_filename = secure_filename(file.filename)
            unique_filename = make_unique(original_filename)
            file.save(os.path.join(app.config['UPLOAD_FOLDER'], unique_filename))

---

虽然这样做是为了避免重复，但在更大的应用程序中，您可能希望扩展这种方法。

如果您将original_filename 和unique_filename 的值存储在数据库中，那么这允许您在下载路径中执行以下操作：

from flask import send_file
# ...
f = os.path.join(app.config['UPLOAD_FOLDER'], unique_filename)
send_file(f, attachment_filename=original_filename)

这样做的好处是文件以唯一标识符存储在您的服务器上，但用户永远不会知道这一点，因为文件会以最初上传的文件名返回给他们。

事实上，您可能希望更进一步，只需使用 UUID 字符串保存文件，而不是附加文件名；而不是使用上面的make_unique 函数，将第三行更改为：

unique_filename = uuid4().__str__()

这仍将提供具有正确 mimetype 的文件，因为 send_file 根据提供的 attachment_filename 猜测 mimetype。