はじめに
Python の標準ライブラリの mimetypes モジュール を使って、
MIME Type を取得しようとしたら少しだけハマったのでメモを残す。
公式ドキュメント にもしっかり書いてあるが、
「ファイル名拡張子に関連付けられた MIME 型とを変換します」 となっている。
どうやらファイル名に含まれる拡張子しか見てくれていないっぽい。
今回は拡張子は無しの状態でファイルを判定したかったので、
mimetypes モジュールではダメだったというお話です。
mimetypes モジュール で試してみた
対象のファイルは png 画像
main.py
import mimetypes
IMAGE_FILE_PATH = './image.png'
print('===== .png =====')
print(f'-- mimetypes: {mimetypes.guess_type(IMAGE_FILE_PATH)[0]}')
実行結果
$ python main.py
===== .png =====
-- mimetypes: image/png
ちゃんと取れてます。
拡張子無しで試す
main.py
import mimetypes
import shutil
IMAGE_FILE_PATH = './image.png'
print('===== 拡張子無し =====')
tmp = './dummy_image'
shutil.copy(IMAGE_FILE_PATH, tmp)
print(f'-- mimetypes: {mimetypes.guess_type(tmp)[0]}')
実行結果
$ python main.py
===== 拡張子無し =====
-- mimetypes: None
None になってしまった。
.mp4 に偽装して試してみる
main.py
import mimetypes
import shutil
IMAGE_FILE_PATH = './image.png'
print('===== 拡張子を .mp4 の偽装 =====')
tmp = './dummy_image.mp4'
shutil.copy(IMAGE_FILE_PATH, tmp)
print(f'-- mimetypes: {mimetypes.guess_type(tmp)[0]}')
実行結果
$ python main.py
===== 拡張子を .mp4 の偽装 =====
-- mimetypes: video/mp4
video/mp4 と判断されてしまった。
代わりに python-magic モジュールを使用
python-magic は libmagic を使ってファイル種別を判定してくれる。
また libmagic を使用するため、実行環境に libmagic がインストールされている必要がある
インストールされていない場合は以下のように実行時にエラーが出てしまいます。
libmagicがインストールされていない環境でpython-magicを使った時のエラー
ImportError: failed to find libmagic. Check your installation
libmagic をインストール
自分は Mac を使用しているため、Homebrew でインストールしました。
libmagicをインストール
$ brew install libmagic
python-magic をインストール
まずは python-magic モジュールをインストール
$ pip install python-magic
python-magic モジュールで試してみた
試し方は前述の mimetypes モジュールと同じです。
- png 画像
- 上記をコピーした拡張子無しファイル
- 同様に拡張子を .mp4 にしたファイル
png 画像
main.py
import magic
IMAGE_FILE_PATH = './image.png'
print('===== .png =====')
print(f'-- python-magic: {magic.from_file(IMAGE_FILE_PATH, mime=True)}')
実行結果
$ python main.py
===== .png =====
-- python-magic: image/png
普通に取得できます。
拡張子無し
main.py
import shutil
import magic
IMAGE_FILE_PATH = './image.png'
print('===== 拡張子無し =====')
tmp = './dummy_image'
shutil.copy(IMAGE_FILE_PATH, tmp)
print(f'-- python-magic: {magic.from_file(tmp, mime=True)}')
実行結果
$ python main.py
===== 拡張子無し =====
-- python-magic: image/png
mimetypes では None だったのに対し、ちゃんと取得できています。
.mp4 に偽装
main.py
import shutil
import magic
IMAGE_FILE_PATH = './image.png'
print('===== 拡張子を .mp4 の偽装 =====')
tmp = './dummy_image.mp4'
shutil.copy(IMAGE_FILE_PATH, tmp)
print(f'-- python-magic: {magic.from_file(tmp, mime=True)}')
実行結果
$ python main.py
===== 拡張子を .mp4 の偽装 =====
-- python-magic: image/png
mimetypes では video/mp4 だったのに対し、ちゃんと取得できています。
まとめ
ファイル種別判定には mimetypes より python-magic を使ったほうが良い！
main.py
import mimetypes
import shutil
import magic
IMAGE_FILE_PATH = './image.png'
print('===== .png =====')
print(f'-- mimetypes: {mimetypes.guess_type(IMAGE_FILE_PATH)[0]}')
print(f'-- python-magic: {magic.from_file(IMAGE_FILE_PATH, mime=True)}')
print('')
print('===== 拡張子無し =====')
tmp = './dummy_image'
shutil.copy(IMAGE_FILE_PATH, tmp)
print(f'-- mimetypes: {mimetypes.guess_type(tmp)[0]}')
print(f'-- python-magic: {magic.from_file(tmp, mime=True)}')
print('')
print('===== 拡張子を .mp4 の偽装 =====')
tmp = './dummy_image.mp4'
shutil.copy(IMAGE_FILE_PATH, tmp)
print(f'-- mimetypes: {mimetypes.guess_type(tmp)[0]}')
print(f'-- python-magic: {magic.from_file(tmp, mime=True)}')
実行結果
$ python main.py
===== .png =====
-- mimetypes: image/png
-- python-magic: image/png
===== 拡張子無し =====
-- mimetypes: None
-- python-magic: image/png
===== 拡張子を .mp4 の偽装 =====
-- mimetypes: video/mp4
-- python-magic: image/png