[Python3] nullチェックをきれいに書く方法を教えてください

Question

pythonで文字列が0のとき、Noneのとき、dict形式のkeyが無いときに引っ掛けるif文を作ったのですが、もう少し短く書く方法は無いでしょうか

# 以下のパターンを引っ掛けたい
dic = {"hello": "hello"}
dic = {"test": ""}
dic = {"test": " "}
dic = {"test": None}

# このif文をきれいに書きたい

# dic['test']がNoneの場合エラーになる
if 'test' in dic and len(dic['test']) == 0 and dic['test'] is not None :
    test = dic['test']

# 追記
if 'test' in dic and dic['test'] is not None and len(dic['test']) == 0 :
    test = dic['test']

else:
    print("error")

よろしくお願いします。

Answer 1

Noneが入っているのが最大の問題だと思いますが、
短く書きたいなら

if (dic.get('test') or '').strip():
   ...
else:
   ...  # こっちがエラー

でしょうかね。

短く書かないほうが読みやすいケースだと思いますけれども。

---

https://docs.python.org/ja/3/reference/expressions.html#boolean-operations

a or bを評価すると、aが真値ならaと、偽値ならbとなります。
dic.get('test')で、キーがなくてNoneになるか、Noneや空文字がとりだされたときはorの右の空文字で評価されます。

Comments

[cray123]

ご回答ありがとうございます。

確かに短く書くことで可読性が損なわれる場合もあるみたいですね..
参考にさせていただきます

また、この書き方も初めて見るのですが、どのような評価が行われているのでしょうか
それについて解説のある参考サイトでもいいので教えていただけると嬉しいです

Answer 2

if dic.get('test') == '': とかですかね。

######get(key[, default])

key が辞書にあれば key に対する値を、そうでなければ default を返します。 default が与えられなかった場合、デフォルトでは None となります。そのため、このメソッドは KeyError を送出することはありません。

引用元：Python 標準ライブラリ » 組み込み型 - マッピング型 --- dict

コメントを受けて

dic = {"hello": "hello"}

dic = {"test": ""}
dic = {"test": " "}
dic = {"test": None}
挙げているdicのパターンはすべてerrorにしたいです

条件が複雑で面倒なので、判定用の関数を用意しましょう。

def is_valid_dic(dic):
    if 'test' not in dic:
        return False
    
    value = dic['test']
    if value is None:
        return False
    
    if value.strip() == '':
        return False
    
    return True

あるいは辞書を作るとき、空白のみの文字列を書き込まずにNoneを使うようにしても良いです。
これなら if not dic.get('test'): で済みます。

Comments

[cray123]

ご回答ありがとうございます。
` if not dic.get('test') in {None, '', ' '} `
としたら短くかけそうですね!

[LouiS0616]

それだと元のコードと真偽変わっちゃいませんか？

- testをキーに持ち
- その要素が空文字列である
という条件では無かったのですか？

[cray123]

やりたかった判定は以下ですが
・dicに'test'キーが存在しない場合
・要素がNoneの場合
・要素のlengthが0の場合　
・要素がスペースの場合

先のコードだと問題があるということでしょうか?

[LouiS0616]

ええと、
dic = {"hello": "hello"}
dic = {"test": ""}
dic = {"test": " "}
dic = {"test": None}

どれがokでどれがerrorなんですか？
元のコードだと error/ok/error/実行時例外 だったのですが。

[cray123]

すみません　書き方が悪かったようです
挙げているdicのパターンはすべてerrorにしたいです

[cray123]

今気づいたのですが、質問に書いた
dic['test'] is not None

をif文の最後に書いているのでdic['test']がNoneだった場合にもlen()を取得してしまうので例外になるようです

スペースの場合の考慮も漏れてますね...

[LouiS0616]

一行で書くなら if 'test' not in dic or dic['test'] is None or dic['test'].strip() == '': ですけれど、読めたものでは無いので追記したような関数を噛ませた方が良いでしょう。

[cray123]

追記ありがとうございます

strip()で空白を削除して比較というのは思いつきませんでした。
ご提示いただいた関数も参考にさせていただきます

[cray123]

ちなみに
if not dic.get('test') in {None, '', ' '}
では問題があるのでしょうか?

[LouiS0616]

複数空白がある場合に対応できません。
dic = {'test': '空白 空白 空白'} とか。

[cray123]

たしかにそうですね!!

ありがとうございます

Answer 3

こうですかね。

item = dic.get("test")
if item and item.strip():
    test = item
else:
    print("error")

これは

item = dic.get("test")
if (item is not None) and (item.strip() != ""):
    test = item
else:
    print("error")

とだいたい同じです。

Comments

[cray123]

ありがとうございます

前者の書き方は知らなかったです
左辺と右辺でどのような評価が行われているのでしょうか

それについて解説のある参考サイトでもいいので教えていただけると嬉しいです

[tachikoma]

ここに似たような例がありました。
https://www.pythoncentral.io/how-to-check-if-a-list-tuple-or-dictionary-is-empty-in-python/