PyYAML YAML load_all到Objects(很好)

时间:2016-05-17 07:47:25

标签: python pyyaml

我有以下Python:

class Instrument:
    def quandl_get(self):
        return quandl.get(self.quandl)

def yaml_load_all(name):
    with open('yaml/' + str(name) + '.yaml', 'r') as ymlfile:
        return list(yaml.load_all(ymlfile))

instruments = yaml_load_all("instruments")

以下YAML:

!!python/object:v.Instrument
vstoxx:
    fractional_contracts: False
    multiplier: 100
    slippage: 0.0017
    per_contract_cost: 0.0211
    quandl: 'CHRIS/EUREX_FVS1'
---  
!!python/object:v.Instrument
corn:
    multiplier: 5000
    symbol: 'zc'
    months_traded: [3, 5, 7, 9, 12]
    quandl: 'CHRIS/CME_C2'
    forecasts: ['ewmac']

我想要的是乐器的词典,所以我可以做instruments['vstoxx'].quandl_get()

作为奖励,我不想每次都在YAML文件中重复!!python/object:v.Instrument

目前我收到一份文书清单。我想我需要做两个步骤,将对象加载到列表中,然后将它们映射到dict。这真的是最干净的方式吗?

1 个答案:

答案 0 :(得分:0)

如果您在文件中使用load_all和多个文档(使用---执行此操作),则始终会获得单独数据元素的列表。

您可以做的是将YAML文件更改为:

!!python/object:v.Instruments
vstoxx:
    fractional_contracts: False
    multiplier: 100
    slippage: 0.0017
    per_contract_cost: 0.0211
    quandl: 'CHRIS/EUREX_FVS1'
corn:
    multiplier: 5000
    symbol: 'zc'
    months_traded: [3, 5, 7, 9, 12]
    quandl: 'CHRIS/CME_C2'
    forecasts: ['ewmac']

(请不要在背后包含最后一个s)然后实现一个类Instruments,当它加载YAML时创建单个Instrument个实例并填充它们,并且存储键值对的属性self._lookup(值是Instrument的实例)。然后在Instruments创建:

 __getitem__(self, key):
     return self._lookup[key]