mime.py

# Copyright The IETF Trust 2020, All Rights Reserved
# -*- coding: utf-8 -*-


import magic
import re


def get_mime_type(content):
    # try to fixup encoding
    if hasattr(magic, "open"):
        m = magic.open(magic.MAGIC_MIME)
        m.load()
        filetype = m.buffer(content)
    else:
        m = magic.Magic()
        m.cookie = magic.magic_open(
            magic.MAGIC_NONE | magic.MAGIC_MIME | magic.MAGIC_MIME_ENCODING
        )
        magic.magic_load(m.cookie, None)
        filetype = m.from_buffer(content)
    # Work around silliness in libmagic on OpenSUSE 15.1
    filetype = filetype.replace("text/x-Algol68;", "text/plain;")
    filetype = filetype.replace("application/vnd.hp-HPGL;", "text/plain;")
    if ";" in filetype and "charset=" in filetype:
        mimetype, charset = re.split("; *charset=", filetype)
    else:
        mimetype = re.split(";", filetype)[0]
        charset = "utf-8"
    return mimetype, charset