Есть ли более чистый способ цепочки проверок пустого списка в Python? - PullRequest
3 голосов
/ 12 октября 2009

У меня есть довольно сложный объект (десериализованный json, поэтому я не слишком контролирую его), который мне нужно проверить на наличие и перебрать довольно глубокие элементы, поэтому сейчас у меня есть что-то вроде :

if a.get("key") and a["key"][0] and a["key"][0][0] :
    for b in a["key"][0][0] :
        #Do something

, который работает, но довольно уродливо. Кажется, должен быть лучший способ сделать это, так что же является более элегантным решением?

Ответы [ 3 ]

14 голосов
/ 12 октября 2009
try:
  bs = a["key"][0][0]
# Note: the syntax for catching exceptions is different in old versions
# of Python. Use whichever one of these lines is appropriate to your version.
except KeyError, IndexError, TypeError:   # Python 3
except (KeyError, IndexError, TypeError): # Python 2
  bs = []
for b in bs:

И вы можете упаковать его в функцию, если не возражаете против более длинных строк:

def maybe_list(f):
  try:
    return f()
  except KeyError, IndexError, TypeError:
    return []

for b in maybe_list(lambda: a["key"][0][0]):
3 голосов
/ 12 октября 2009

Я бы написал пользовательскую функцию индексатора следующим образом:

def safe_indexer(obj, *indices):
    for idx in indices:
        if not obj: break

        if hasattr(obj, "get"):
            obj = obj.get(idx)
        else:
            obj = obj[idx]

    return obj

Использование:

a = {"key": {0: {0: "foo"} } };
print safe_indexer(a, "key", 0, 0)
print safe_indexer(a, "bad", 0, 0)

Выход:

foo
None
2 голосов
/ 12 октября 2009

Что по этому поводу:

try:
    for b in a['key'][0][0]:
        # do stuff.
except KeyError, TypeError, IndexError:
    # respond appropriately.
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...