Python: request.filesで取得した値の場合分け

Question

困っていること

Pythonでユーザー画像の変更ができるようにフォームを作成しています。
その際に、pngやjpegなどの画像拡張子に変更しているダミー画像（拡張子は画像だが中身は別ファイル）を弾くようにしたいと考えています。

自分で調べて見たところ、imghdrを使うと中身を見てくれることがわかりました。
また、アップロードした画像の中身が画像ファイルでなかった時Noneを返すという性質があることもわかりました。

そこで、Noneが返ってこなかったら正常な処理を、Noneだったらエラーメッセージを表示というように処理を書いています。

しかし、これだと特に画像を変更をせず変更完了ボタンを押した際にもNoneを返すため、エラーメッセージを表示してしまいます。

そこで、imghdrで中身を見る前に、そもそも画像が選択されているのかを判定したいのですが、どうif文を書いたらいいかわからず困っています。

現在書いているコード

image = request.files['image']
print(image)
imageDecision = imghdr.what(image)
print(imageDecision)

if imageDecision is not None:
~~~

print(image)の処理結果

1. 何も選択せずに変更完了ボタンを押す

<FileStorage: '' ('application/octet-stream')>

1. ダミー画像を選択して変更完了ボタンを押す

<FileStorage: 'ダミー.png' ('image/png')>

print(imageDecision)の処理結果

1. 何も選択せずに変更完了ボタンを押す

None

1. ダミー画像を選択して変更完了ボタンを押す

None

FileStorageが空だったらみたいに書けばいいのかなとは思うのですが、具体的にどう指定したら良いのかわからず困っています。。。。
以上です。よろしくお願いします。

Answer 1

プロフィール変更画面でしょうか？

もし画像のアップロードが必須ではない（ニックネームやパスワードだけ変更する等）のなら、"もし画像がアップロードされたら" チェック処理とユーザー画像の変更処理を行うという流れにするのが自然かなと思います。

request.files['image'] からわかるように、アップロードされたファイルは request.files にリストとして格納されています。Python ではリストに特定の要素が含まれているかを <要素> in <リスト> で検索できるので、その判定を入れてあげればよいと思います。

"""プロフィール画像がアップロードされたら差し替える"""
if 'image' in request.files:
    image = request.files['image']
    imageDecision = imghdr.what(image)
    if imageDecision is None:
        flash('ファイルは画像ではないか壊れています', 'error' )
        return redirect(request.url)

またユーザー画像変更だけをするページであれば、ファイルをアップロードしないのはそもそもユーザー側の操作ミスなので、即エラーとするのがよいですね。

if 'image' in request.files:
    flash('ファイルを選択して下さい', 'error' )
    return redirect(request.url)

image = request.files['image']
imageDecision = imghdr.what(image)

if imageDecision is None:
    flash('ファイルは画像ではないか壊れています', 'error' )
    return redirect(request.url)

---

余談ですがHTMLのフォーム側にも適切なバリデーションを入れてあげると親切です。

たとえば、画像の選択を必須にしたいのならば required 属性を付与したり、ファイル形式を限定したいのであれば accept 属性を付与します。

<form>
    <input type="file" accept=".jpg,.gif,.png" required>
</form>

ブラウザによっては画像だけピックアップしてくれたりとユーザーの利便性が向上します。

Comments

[gomakichi]

遅くなってすみません。上記の内容で無事解決いたしました。ありがとうございます。