Xpath-подобный запрос для вложенных словарей Python

Не совсем красиво, но вы можете использовать что-то вроде

def xpath_get(mydict, path):
    elem = mydict
    try:
        for x in path.strip("/").split("/"):
            elem = elem.get(x)
    except:
        pass

    return elem

Конечно, это не поддерживает такие вещи xpath, как индексы ... не говоря уже об указанной ловушке / ключа unutbu.

Одна из лучших библиотек, которую мне удалось выделить, которая, к тому же, очень активно развивается, — это проект, извлеченный из boto: JMESPath. У него очень мощный синтаксис для выполнения действий, для выражения которых обычно требуются страницы кода.

Вот некоторые примеры:

search('foo | bar', {"foo": {"bar": "baz"}}) -> "baz"
search('foo[*].bar | [0]', {
    "foo": [{"bar": ["first1", "second1"]},
            {"bar": ["first2", "second2"]}]}) -> ["first1", "second1"]
search('foo | [0]', {"foo": [0, 1, 2]}) -> [0]

Сейчас есть более простой способ сделать это.

http://github.com/akesterson/dpath-python

$ easy_install dpath
>>> dpath.util.search(YOUR_DICTIONARY, "morefoo/morebar")

... сделано. Или, если вам не нравится возвращать результаты в представление (объединенный словарь, сохраняющий пути), вместо этого выведите их:

$ easy_install dpath
>>> for (path, value) in dpath.util.search(YOUR_DICTIONARY, "morefoo/morebar", yielded=True)

... и готово. 'value' будет содержать {'bacon': 'foobar'} в этом случае.

Существует более новая библиотека jsonpath-rw, поддерживающая JSONPATH, но для словарей и массивов Python, как вы пожелаете.

Таким образом, ваш 1-й пример становится:

from jsonpath_rw import parse

print( parse('$.morefoo.morebar').find(foo) )

И 2-й:

print( parse("$.morefoo[0].morebar.bacon").find(foo) )

PS: альтернативная более простая библиотека, также поддерживающая словари, — python-json-pointer с более похожим на XPath синтаксисом.

словарь › jmespath

Вы можете использовать JMESPath, который является языком запросов для JSON и имеет реализация Python.

import jmespath # pip install jmespath

data = {'root': {'section': {'item1': 'value1', 'item2': 'value2'}}}

jmespath.search('root.section.item2', data)
Out[42]: 'value2'

Синтаксис запроса jmespath и живые примеры: http://jmespath.org/tutorial.html

словарь › xml › xpath

Другим вариантом может быть преобразование ваших словарей в XML с помощью чего-то вроде dicttoxml, а затем использование обычных выражений XPath, например. через lxml или любую другую библиотеку, которую вы предпочитаете.

from dicttoxml import dicttoxml  # pip install dicttoxml
from lxml import etree  # pip install lxml

data = {'root': {'section': {'item1': 'value1', 'item2': 'value2'}}}
xml_data = dicttoxml(data, attr_type=False)
Out[43]: b'<?xml version="1.0" encoding="UTF-8" ?><root><root><section><item1>value1</item1><item2>value2</item2></section></root></root>'

tree = etree.fromstring(xml_data)
tree.xpath('//item2/text()')
Out[44]: ['value2']

Json указатель

Еще одним вариантом является Json Pointer, который представляет собой спецификация IETF с реализацией Python:

• https://github.com/stefankoegl/python-json-pointer

Из руководства по jsonpointer-python:

from jsonpointer import resolve_pointer

obj = {"foo": {"anArray": [ {"prop": 44}], "another prop": {"baz": "A string" }}}

resolve_pointer(obj, '') == obj
# True

resolve_pointer(obj, '/foo/another%20prop/baz') == obj['foo']['another prop']['baz']
# True

>>> resolve_pointer(obj, '/foo/anArray/0') == obj['foo']['anArray'][0]
# True

Если вам нравится краткость:

def xpath(root, path, sch='/'):
    return reduce(lambda acc, nxt: acc[nxt],
                  [int(x) if x.isdigit() else x for x in path.split(sch)],
                  root)

Конечно, если у вас есть только словари, то это проще:

def xpath(root, path, sch='/'):
    return reduce(lambda acc, nxt: acc[nxt],
                  path.split(sch),
                  root)

Удачи в поиске ошибок в спецификации пути ;-)

Другая альтернатива (помимо предложенной jellybean) это:

def querydict(d, q):
  keys = q.split('/')
  nd = d
  for k in keys:
    if k == '':
      continue
    if k in nd:
      nd = nd[k]
    else:
      return None
  return nd

foo = {
  'spam':'eggs',
  'morefoo': {
               'bar':'soap',
               'morebar': {'bacon' : 'foobar'}
              }
   }
print querydict(foo, "/morefoo/morebar")

Потребуется дополнительная работа над тем, как будет работать XPath-подобный селектор. '/' является действительным ключом словаря, так как же

foo={'/':{'/':'eggs'},'//':'ham'}

обрабатываться?

foo.select("///")

было бы неоднозначно.

Есть ли у вас какая-либо причина запрашивать его так же, как шаблон XPath? Как предположил комментатор вашего вопроса, это просто словарь, поэтому вы можете получить доступ к элементам в виде гнезда. Кроме того, учитывая, что данные представлены в формате JSON, вы можете использовать модуль simplejson для их загрузки и доступа к элементам.

Существует проект JSONPATH, который пытается помочь людям делать противоположное тому, что вы собираетесь делать ( учитывая XPATH, как сделать его легко доступным через объекты python), что кажется более полезным.

def Dict(var, *arg, **kwarg):
  """ Return the value of an (imbricated) dictionnary, if all fields exist else return "" unless "default=new_value" specified as end argument
      Avoid TypeError: argument of type 'NoneType' is not iterable
      Ex: Dict(variable_dict, 'field1', 'field2', default = 0)
  """
  for key in arg:
    if isinstance(var, dict) and key and key in var:  var = var[key]
    else:  return kwarg['default'] if kwarg and 'default' in kwarg else ""   # Allow Dict(var, tvdbid).isdigit() for example
  return kwarg['default'] if var in (None, '', 'N/A', 'null') and kwarg and 'default' in kwarg else "" if var in (None, '', 'N/A', 'null') else var

foo = {
  'spam':'eggs',
  'morefoo': {
               'bar':'soap',
               'morebar': {'bacon' : 'foobar'}
              }
   }
print Dict(foo, 'morefoo', 'morebar')
print Dict(foo, 'morefoo', 'morebar', default=None)

Имейте функцию SaveDict(value, var, *arg), которая может даже добавляться к спискам в dict...