एक pyYAML फ़ाइल कैसे लोड करें और शब्दकोश नोटेशन का उपयोग करने के बजाय विशेषताओं का उपयोग करके इसे एक्सेस करें?

Question

मैं एक YAML config कि तरह दिखता है:

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

मैं इसे लोड करने के लिए अजगर YAML मॉड्यूल का उपयोग कर रहा है, लेकिन मैं की तरह बेहतर तरीके से उपयोग करना चाहते:

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

मैं क्या चाहते करने के लिए है करने में सक्षम हो: print(config.content)।

Answer 1

आप के रूप this जवाब में चर्चा की, निम्नलिखित वर्ग का उपयोग कर शब्दकोशों के साथ वस्तु संकेतन का उपयोग कर सकते हैं:

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

कार्रवाई में इस वर्ग:

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

संपादित यह उप के साथ काम करता है रिकर्सिवली शब्दकोश, उदाहरण के लिए:

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

ऐसा करने का सबसे आसान तरीका शायद tag:yaml.org,2002:map के लिए वाईएएमएल कन्स्ट्रक्टर को ओवरराइट करना है, इसलिए यह सामान्य शब्दकोश के बजाय एक कस्टम शब्दकोश कक्षा देता है।

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 

ध्यान दें कि यह प्रक्रिया YAML का उपयोग करता है, अगर यह सब कुछ सामान्य अजगर तरीके से काम करने की उम्मीद में सब कुछ टूट जाएगा। आप एक कस्टम लोडर का उपयोग कर सकते हैं, लेकिन मुझे व्यक्तिगत रूप से PyYAML दस्तावेज़ थोड़ा भूलभुलैया मिलती है, और ऐसा लगता है कि दुष्प्रभाव एक अपवाद के बजाय एक नियम के रूप में वैश्विक और संक्रामक हैं।

आपको चेतावनी दी गई है।

एक विकल्प के रूप में, अगर अपने स्कीमा अपेक्षाकृत स्थिर आप अपने खुद के वर्गों लिख सकते हैं और उन लोगों के लिए deserialize सकता है (जैसे, class Config साथ id और name गुण)। हालांकि, यह शायद अतिरिक्त कोड की लागत के लायक नहीं होगा।