Come caricare un file pyYAML e accedervi usando gli attributi invece di usare la notazione del dizionario?

Question

Ho una configurazione YAML che assomiglia:

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

Sto usando modulo Python YAML di caricarlo ma voglio accedervi in ​​modo migliore come:

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

Quello che voglio è quello di essere in grado di fare: print(config.content).

Answer 1

È possibile utilizzare la notazione oggetto con dizionari che utilizzano la seguente classe, come discusso in this risposta:

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

Questa classe in azione:

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

Modifica Questo funziona in modo ricorsivo con sub- dizionari, ad esempio:

>>> 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

Il modo più semplice per eseguire questa operazione è probabilmente sovrascrivere il costruttore YAML per tag:yaml.org,2002:map in modo che restituisca una classe dizionario personalizzata anziché un dizionario normale.

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 

Si noti che questo si romperà tutto il resto del processo che utilizza YAML, se si aspetta di tutto per lavorare nel modo Python normale. È possibile utilizzare un caricatore personalizzato, ma personalmente trovo la documentazione PyYAML un po 'labirintica, e sembra che gli effetti collaterali siano globali e contagiosi come una regola piuttosto che un'eccezione.

Sei stato avvisato.

In alternativa, se lo schema è relativamente statico si potrebbe scrivere le proprie classi e deserializzare a quelli (ad esempio, con class Configid e name proprietà). Probabilmente non varrebbe la pena il costo del codice extra, comunque.