Как можно переписать эту функцию для реализации OrderedDict? - PullRequest
Новая
       17

Как можно переписать эту функцию для реализации OrderedDict?

12 голосов
/ 08 ноября 2010

У меня есть следующая функция, которая выполняет грубую работу по синтаксическому анализу XML-файла в словаре.

К сожалению, поскольку словари Python не упорядочены, я не могу циклически перемещаться по узлам, как хотелось бы.

Как мне изменить это, чтобы он выводил упорядоченный словарь, который отражает исходный порядок узлов при циклическом 'for'. 

def simplexml_load_file(file):
    import collections
    from lxml import etree

    tree = etree.parse(file)
    root = tree.getroot()

    def xml_to_item(el):
        item = None
        if el.text:
            item = el.text
        child_dicts = collections.defaultdict(list)
        for child in el.getchildren():
            child_dicts[child.tag].append(xml_to_item(child))
        return dict(child_dicts) or item

    def xml_to_dict(el):
        return {el.tag: xml_to_item(el)}

    return xml_to_dict(root)

x = simplexml_load_file('routines/test.xml')

print x

for y in x['root']:
    print y

Выходы: 

{'root': {
    'a': ['1'],
    'aa': [{'b': [{'c': ['2']}, '2']}],
    'aaaa': [{'bb': ['4']}],
    'aaa': ['3'],
    'aaaaa': ['5']
}}

a
aa
aaaa
aaa
aaaaa

Как я могу реализовать collection.OrderedDict, чтобы я мог быть уверен в получении правильного порядка узлов?

XML-файл для справки: 

<root>
    <a>1</a>
    <aa>
        <b>
            <c>2</c>
        </b>
        <b>2</b>
    </aa>
    <aaa>3</aaa>
    <aaaa>
        <bb>4</bb>
    </aaaa>
    <aaaaa>5</aaaaa>
</root>

Ответы [ 3 ]

29 голосов
/ 08 ноября 2010

Вы можете использовать новый подкласс OrderedDict dict, который был добавлен к модулю collections стандартной библиотеки в версии 2.7 *.На самом деле вам нужна комбинация Ordered + defaultdict, которая не существует, но ее можно создать, создав подкласс OrderedDict, как показано ниже: 

import collections

class OrderedDefaultdict(collections.OrderedDict):
    """ A defaultdict with OrderedDict as its base class. """

    def __init__(self, default_factory=None, *args, **kwargs):
        if not (default_factory is None
                or isinstance(default_factory, collections.Callable)):
            raise TypeError('first argument must be callable or None')
        super(OrderedDefaultdict, self).__init__(*args, **kwargs)
        self.default_factory = default_factory  # called by __missing__()

    def __missing__(self, key):
        if self.default_factory is None:
            raise KeyError(key,)
        self[key] = value = self.default_factory()
        return value

    def __reduce__(self):  # optional, for pickle support
        args = (self.default_factory,) if self.default_factory else tuple()
        return self.__class__, args, None, None, self.iteritems()

    def __repr__(self):  # optional
        return '%s(%r, %r)' % (self.__class__.__name__, self.default_factory,
                               list(self.iteritems()))

def simplexml_load_file(file):
    from lxml import etree

    tree = etree.parse(file)
    root = tree.getroot()

    def xml_to_item(el):
        item = el.text or None
        child_dicts = OrderedDefaultdict(list)
        for child in el.getchildren():
            child_dicts[child.tag].append(xml_to_item(child))
        return collections.OrderedDict(child_dicts) or item

    def xml_to_dict(el):
        return {el.tag: xml_to_item(el)}

    return xml_to_dict(root)

x = simplexml_load_file('routines/test.xml')
print(x)

for y in x['root']:
    print(y)

Вывод, полученный в результате тестаФайл XML выглядит следующим образом:

Вывод: 

{'root':
    OrderedDict(
        [('a', ['1']),
         ('aa', [OrderedDict([('b', [OrderedDict([('c', ['2'])]), '2'])])]),
         ('aaa', ['3']),
         ('aaaa', [OrderedDict([('bb', ['4'])])]),
         ('aaaaa', ['5'])
        ]
    )
}

a
aa
aaa
aaaa
aaaaa

Что, я думаю, близко к тому, что вы хотите.

* Если ваша версия Python не имеет OrderedDict, который был представлен в версии 2.5, вы можете использовать вместо этого заказной словарь Раймонда Хеттингера для Py2.4 ActiveState в качестве базового класса.

Незначительное обновление:

Добавлен метод __reduce__(), который позволяет правильно и аккуратно обрабатывать экземпляры класса.Это не было необходимо для этого вопроса, но возникло в подобном одном.

1 голос
/ 10 июня 2016

У меня работает рецепт от martineau, но есть проблемы с методом copy (), унаследованным от DefaultDict.Следующий недостаток устраняет этот недостаток: 

class OrderedDefaultDict(OrderedDict):
    #Implementation as suggested by martineau

    def copy(self):
         return type(self)(self.default_factory, self)

Обратите внимание, что в этой реализации нет глубокого копирования, что представляется особенно подходящим для словарей по умолчанию, а в большинстве случаев - правильная вещь

1 голос
/ 08 ноября 2010

Существует множество возможных реализаций OrderedDict, перечисленных в ответе здесь: Как получить элементы из словаря в порядке их вставки?

Вы можете создать свойМодуль OrderedDict для использования в вашем собственном коде путем копирования одной из реализаций.Я предполагаю, что у вас нет доступа к OrderedDict из-за версии Python, которую вы используете.

Один интересный аспект вашего вопроса - возможная потребность в функциональности defaultdict.Если вам это нужно, вы можете реализовать метод __missing__, чтобы получить желаемый эффект.

...