如何从二进制字段中查找文件 mimetype

Question

我有一个带有二进制字段的文件模型

class File(models.Model):
    id = models.AutoField(primary_key=True)
    file = models.BinaryField()
    name = models.TextField(null=False, blank=False)

我创建了一个按 id 呈现文件的视图

def media(request, file_id):
    file = get_object_or_404(File, id=file_id)
    content = file.file
    return FileResponse(content, as_attachment=False, filename=file.name)

这是我得到的输出

%PDF-1.4 3 0 obj >>>>] /Contents 4 0 R>> endobj 4 0 obj 流 x�3R��2�35W(�*T0P�R0T(�Y @���@QC= PA�J���1Tp�W�+R@&�ҹ6f� endstream endobj 1 0 obj endobj 5 0 obj /Group /Length 19219 >> stream x ��}K�m�U����F�?��{9���������h�(�U��+�[�z;��A�����HI '4h#z��

当文件类型已知时，我可以手动将content_type 设置为application/pdf、image/png... 但这是BinaryField，我不知道用户上传了什么。无论文件类型如何，如何获得正确的 mimetype？

Answer 1

您可以使用python-magic package [pypi] 来猜测 MIME 类型。你安装这个（在你的虚拟环境中）：

pip3 install <b>python-magic</b>

然后你可以根据字节流猜测mimetype

from io import BytesIO
from magic import from_buffer as mime_from_buffer

def media(request, file_id):
    file = get_object_or_404(File, id=file_id)
    content = file.file
    with BytesIO(content) as f:
        mimetype = mime_from_buffer(f.read(), mime=True)
    # …
    return FileResponse(content, as_attachment=False, filename=file.name)

但这是一个猜测，并非所有文件都有一个标头，可以清楚地表明它们携带的是哪种数据，所以最终猜测可能是错误的。

我有一个带有二进制字段的文件模型

在数据库中存储大型二进制流通常不是一个好主意。数据库在存储大量（二进制）数据方面通常不是很好，并且数据库在检索、存储等文件方面通常比文件系统慢。这就是为什么FileField [Django-doc] 会将检索文件的路径存储在数据库中，并将文件内容本身存储在文件系统或专用存储服务器上。