Flask / Python. Obtenez mimetype de fichier uploadé

Question

J'utilise Flask micro-framework 0.6 et Python 2.6

Je dois obtenir le type MIME d'un fichier téléchargé pour que je puisse le stocker.

Voici le relevent code Python / Flask:

@app.route('/upload_file', methods=['GET', 'POST'])
def upload_file():
    if request.method == 'POST':
        file = request.files['file']
        mimetype = #FIXME
        if file:
            file.save(os.path.join(UPLOAD_FOLDER, 'File-Name')
            return redirect(url_for('uploaded_file'))
        else:
            return redirect(url_for('upload'))

Et voici le code de la page web:

<form action="upload_file" method=post enctype=multipart/form-data> 
Select file to upload: <input type=file name=file> 
<input type=submit value=Upload> 
</form> 

Les œuvres de code, mais je dois être en mesure d'obtenir les mimetype quand il télécharge. J'ai eu un coup d'œil à la documentation Flask ici: http: // flacon. pocoo.org/docs/api/#incoming-request-data
Alors, je sais qu'il ne soit le type MIME, mais je ne peux pas trouver comment récupérer - comme une chaîne de texte, par exemple 'Txt / plain'.

Toutes les idées?

Merci.

Answer 1

De la docs , file.content_type contient la le type complet avec le codage, mimetype ne contient que le type MIME.

@app.route('/upload_file', methods=['GET', 'POST'])
def upload_file():
    if request.method == 'POST':
        file = request.files.get('file')
        if file:
            mimetype = file.content_type
            filename = werkzeug.secure_filename(file.filename)
            file.save(os.path.join(UPLOAD_FOLDER, filename)
            return redirect(url_for('uploaded_file'))
        else:
            return redirect(url_for('upload'))

Answer 2

Vous pourriez en théorie utiliser request.files['YOUR_FILE_KEY'].content_type, mais la mise en œuvre (inclus ci-dessous, trouvée dans werkzeug.datastructures ) soit confiance quel que soit le client fournit, ou utilise mimetypes.guess_type qui ne vérifie que l'extension de fichier (voir doc Python ici ).

class FileMultiDict(MultiDict):

    """A special :class:`MultiDict` that has convenience methods to add
    files to it.  This is used for :class:`EnvironBuilder` and generally
    useful for unittesting.
    .. versionadded:: 0.5
    """

    def add_file(self, name, file, filename=None, content_type=None):
        """Adds a new file to the dict.  `file` can be a file name or
        a :class:`file`-like or a :class:`FileStorage` object.
        :param name: the name of the field.
        :param file: a filename or :class:`file`-like object
        :param filename: an optional filename
        :param content_type: an optional content type
        """
        if isinstance(file, FileStorage):
            value = file
        else:
            if isinstance(file, string_types):
                if filename is None:
                    filename = file
                file = open(file, 'rb')
            if filename and content_type is None:
                content_type = mimetypes.guess_type(filename)[0] or \
                    'application/octet-stream'
            value = FileStorage(file, filename, name, content_type)

        self.add(name, value)

En fonction de votre cas d'utilisation, vous voudrez peut-être utiliser python-magic qui utilisera le fichier réel pour obtenir le type MIME. Il serait quelque chose comme ça:

import magic


def get_mimetype(data: bytes) -> str:
    """Get the mimetype from file data."""
    f = magic.Magic(mime=True)
    return f.from_buffer(data)


get_mimetype(request.files['YOUR_FILE_KEY'].stream.read(MAX_LENGTH))