Python3で複数階層の辞書型を存在チェックして取得する方法

Question

以下のような辞書型のオブジェクトがあるとします。
「token が存在すれば token の値を取得、なければ例外を出さずに None(undefined, nil) を取得」したいです。

obj = {
  "facebook": {
    "token": "xxx"
  },
  "twitter": {
  }
}

JavaScript で書くとこうなります

const fb = obj["facebook"] && obj["facebook"]["token"];  // => 'xxx'
const tw = obj["twitter"] && obj["twitter"]["token"];        // => undefined
const ig = obj["instgram"] && obj["instagram"]["token"]; // => undefined

Ruby なら Safe Navigation Operator があるのでもっとラクです

const fb = obj["facebook"]&.["token"];  # => 'xxx'
const tw = obj["twitter"]&.["token"];     # => nil
const ig = obj["instgram"]&.["token"];   # => nil

Python3 ではこのように書くしかないんでしょうか
非常に冗長で読みにくいです

const fb = obj["facebook"]["token"] if "facebook" in obj and "token" in obj["facebook"] else None    # => 'xxx'
const tw = obj["twitter"]["token"] if "twitter" in obj and "token" in obj["twitter"] else None              # => None
const ig = obj["instagram"]["token"] if "instagram" in obj and "token" in obj["instagram"] else None # => None

煩雑なのでもっと簡易な書き方があれば教えてください

Answer 1

getをつかうとKeyErrorを出しません。またなかった場合の代替値を指定できます。

facebookやtwitterという第一階層が必ずあるのであれば単に

obj['facebook'].get('token')  # ない場合はNone
obj['facebook'].get('token', 'notoken')  # ない場合はnotokenという文字列

でいけます。第一階層もあるか不定であれば

obj.get("facebook", {}).get("token")

みたいにするのもいいかもしれません。

Comments

[tachikoma]

最後のコードで一回目のget失敗時にdictを返さないといけないのが何か惜しいですね。単なるコメントですが。

[denzow]

一応`defaultdict`使えばそこも省略できますが、すこし混乱しそうなので控えました汗

```
In [6]: from collections import defaultdict

In [7]: data = defaultdict(dict)

In [8]: data['facebook']['token'] = 'my token'

In [9]: data['facebook'].get('token')
Out[9]: 'my token'

In [10]: data['twitter'].get('token')
```

[jnst]

「なければ RuntimeError にならないように empty で初期化してから使う」というスニペットはスクリプト言語ではよく見るものなので、自分にとっては違和感ありませんでした????

obj.get("facebook", {}).get("token")

これからこのコードを活用させてもらいます。