¿Cómo cargar un archivo pyYAML y acceder a él usando atributos en lugar de usar la notación del diccionario?

Question

Tengo una configuración YAML que se parece a:

config: 
- id: foo 
- name: bar 
content: 
- run: xxx 
- remove: yyy 

estoy usando módulo de Python YAML para cargarlo pero quiero acceder a él en mejores maneras como:

stream = open(filename) 
config = load(stream, Loader=Loader) 
print(config['content']) 

Lo que yo quiero es poder hacer: print(config.content).

Answer 1

Puede utilizar notación de objetos con los diccionarios utilizando la siguiente clase, como se discutió en this respuesta:

class DictAsMember(dict): 
    def __getattr__(self, name): 
     value = self[name] 
     if isinstance(value, dict): 
      value = DictAsMember(value) 
     return value 

Esta clase de acción:

>>> my_dict = DictAsMember(one=1, two=2) 
>>> my_dict 
{'two': 2, 'one': 1} 
>>> my_dict.two 
2 

Editar Esto funciona de forma recursiva con sub diccionarios, por ejemplo:

>>> my_dict = DictAsMember(one=1, two=2, subdict=dict(three=3, four=4)) 
>>> my_dict.one 
1 
>>> my_dict.subdict 
{'four': 4, 'three': 3} 
>>> my_dict.subdict.four 
4 

Answer 2

La forma más sencilla de hacerlo es, probablemente, sobrescribir el constructor YAML para tag:yaml.org,2002:map, por lo que devuelve una clase de diccionario personalizada en lugar de un diccionario normal.

import yaml 

class AttrDict(object): 
    def __init__(self, attr): 
     self._attr = attr 
    def __getattr__(self, attr): 
     try: 
      return self._attr[attr] 
     except KeyError: 
      raise AttributeError 

def construct_map(self, node): 
    # WARNING: This is copy/pasted without understanding! 
    d = {} 
    yield AttrDict(d) 
    d.update(self.construct_mapping(node)) 

# WARNING: We are monkey patching PyYAML, and this will affect other clients!  
yaml.add_constructor('tag:yaml.org,2002:map', construct_map) 

YAML = """ 
config: 
    - id: foo 
    - name: bar 
content: 
    - run: xxx 
    - remove: yyy 
""" 

obj = yaml.load(YAML) 

print(obj.config[0].id) # prints foo 

Tenga en cuenta que esto va a romper todo lo demás en el proceso que utiliza YAML, si se espera que todo funcione de la manera normal de Python. Puede usar un cargador personalizado, pero personalmente encuentro que la documentación PyYAML es un poco laberíntica, y parece que los efectos colaterales son globales y contagiosos como una regla en lugar de una excepción.

Has sido advertido.

Como alternativa, si el esquema es relativamente estática podría escribir sus propias clases y deserializar a aquellos (por ejemplo, class Config con id y name propiedades). Sin embargo, probablemente no valdría la pena el costo del código adicional.