Python, magicパッケージを用いてファイルの種類の判定を行いたい

Question

前提・実現したいこと

PythonとFlaskでWebアプリケーションの作成をしています。
フォームから受け取ったファイルの形式をpython-magicパッケージを用いて、
より正確に判定しようとしています。

追記：
ファイルを保存する前に判定し、望んだ形式であれば保存するといったことを想定しています。

発生している問題・エラーメッセージ

下記のコードでファイルの種類の判定を行おうとすると以下のエラーメッセージが出てしまう。
TypeError: expected str, bytes or os.PathLike object, not FileStorage

該当のソースコード

from flask import Flask, redirect, request
from flask import render_template, send_file
import magic

FILE_FORMAT = ['text/plain', 'image/png', 'image/jpeg', 'application/pdf']

@app.route('/upload', methods=['POST'])
def upload():
    # アップロードしたファイルの確認
    upfile = request.files.get('upfile', None)
    mime = magic.from_file(upfile, mime=True)
    if mime in FILE_FORMAT:
        return msg('成功')

試したこと

htmlのformから受け取れるファイルの実態は
werkzeug.datastructures.FileStorageとのことで以下のように変更し試してみました。
formからはテキストファイル(.txt)を送信しました。

試したコード1
mime = magic.from_file(upfile.filename, mime=True)
エラーメッセージ1
FileNotFoundError: [Errno 2] No such file or directory: 'a.txt'

試したコード2
mime = magic.from_file(upfile.mymetype, mime=True)
エラーメッセージ2
FileNotFoundError: [Errno 2] No such file or directory: 'text/plain'

補足情報

以下のバージョンを使用しています
Python 3.6.4
Flask 1.1.1
Werkzeug 1.0.1

Answer 1

最初に示されているエラーメッセージで「TypeError: expected str, bytes or os.PathLike object, not FileStorage」とあるように、「str, bytes, os.PathLike」オブジェクトが期待されているところへ「FileStorage」オブジェクトを渡しているのでエラーが発生しています。

一例ですが、ファイル内容をbytesとして取り出して処理する場合は以下のような感じになります

（省略）

@app.route('/upload', methods=['POST'])
def upload():
    # アップロードしたファイルの確認
    upfile = request.files.get('upfile', None)
    mime = magic.from_buffer(upfile.read(), mime=True)
    app.logger.debug("mime: {}".format(mime))
    if mime in FILE_FORMAT:
        # 「read」実行でファイルの最後にカーソルが移動するので、必要なら戻す
        upfile.seek(0) 
        # 必要なら、ここで保存処理など
        return '成功'

（以下省略）

Answer 2

ネットで調べただけで実験はしていませんが、以下を試してみてください。

mime = magic.from_buffer(upfile, mime=True)

もしだめなら、

mime = magic.from_buffer(upfile.encode(encoding='utf-8'), mime=True)

と、

mime = magic.from_buffer(upfile.encode(encoding='cp932'), mime=True)

を試してみてください。

Answer 3

No such file or directory
ってのは、そのファイルがないってことです。

ファイルの判定を行いたいなら、ちゃんと存在するファイルを判定しましょうよ

Comments

[usako111]

お早い回答ありがとうございます。状況の説明が不十分でした。
ファイルは保存せずに判定することを想定していましたので質問の方に追記しておきました。
また、保存して判定を行う方法も考慮してみます。