pyyaml 3.11 将字典传递给迭代器？

Question

我使用以下 YAML 数据：

Document:
 InPath: /home/me
 OutPath: /home/me
 XLOutFile: TestFile1.xlsx

Sheets: 
  - Sheet: Test123
    InFile: Test123.MQSC
    Server: Testsystem1
  - Sheet: Test345
    InFile: Test345.MQSC
    Server: Testsystem2

Title:
    A: "Server Name"
    B: "MQ Version"
    C: "Broker Version"

Fields:
    A: ServerName
    B: MQVersion
    C: BrokerVersion

和以下代码：

import yaml

class cfgReader():
    def __init__(self):
        self.stream = ""
        self.ymldata = ""
        self.ymlkey = ""
        self.ymld = ""

    def read(self,infilename):
        self.stream = self.stream = file(infilename, 'r') #Read the yamlfile
        self.ymldata = yaml.load(self.stream)    #Instanciate yaml object and parse the input "stream".

    def docu(self):
        print self.ymldata
        print self.ymldata['Sheets']
        for self.ymlkey in self.ymldata['Document']: #passes String to iterator
            print self.ymlkey
        for sheets in self.ymldata['Sheets']:  #passes Dictionary to iterator
            print sheets['Sheet']
        for title in self.ymldata['Title']:
            print title
        for fields in self.ymldata['Fields']:
            print fields

print 输出是：

{'Fields': {'A': 'ServerName', 'C': 'BrokerVersion', 'B': 'MQVersion'}, 'Document': {'XLOutFile': 'TestFile1.xlsx', 'InPath': '/home/me', 'OutPath': '/home/me'}, 'Sheets': [{'Sheet': 'Test123', 'InFile': 'Test123.MQSC', 'Server': 'Testsystem1'}, {'Sheet': 'Test345', 'InFile': 'Test345.MQSC', 'Server': 'Testsystem2'}], 'Title': {'A': 'Server Name', 'C': 'Broker Version', 'B': 'MQ Version'}}
[{'Sheet': 'Test123', 'InFile': 'Test123.MQSC', 'Server': 'Testsystem1'}, {'Sheet': 'Test345', 'InFile': 'Test345.MQSC', 'Server': 'Testsystem2'}]
X
I
O
Test123
Test345
A
C
B
A
C
B

我找不到如何控制数据传递给迭代器的方式。我想要的是将它作为字典传递，以便我可以通过键访问值。这适用于 "Sheets" 但我不明白为什么。文档没有清楚地描述它：http://pyyaml.org/wiki/PyYAMLDocumentation

Answer 1

在您的代码中 self.ymldata['Sheets'] 是一个字典列表，因为您的 YAML 源代码：

  - Sheet: Test123
    InFile: Test123.MQSC
    Server: Testsystem1
  - Sheet: Test345
    InFile: Test345.MQSC
    Server: Testsystem2

是一系列映射（这是 YAML 文件中顶级映射的键 Sheets 的值）。

其他顶级键的值都是映射（而不是映射序列），它们加载为 Python dict。如果你像你一样迭代 dict，你会得到 key 值。

如果您不想遍历这些字典，则不应启动 for 循环。您可能想测试顶层键的值是什么，然后采取相应的行动，例如打印出从 YAML 文件加载的所有字典，除了顶级映射：

import ruamel.yaml as yaml

class CfgReader():
    def __init__(self):
        self.stream = ""
        self.ymldata = ""
        self.ymlkey = ""
        self.ymld = ""

    def read(self, infilename):
        self.stream = open(infilename, 'r') # Read the yamlfile
        self.ymldata = yaml.load(self.stream)    # Instanciate yaml object and parse the input "stream".

    def docu(self):
        for k in self.ymldata:
            v = self.ymldata[k]
            if isinstance(v, list):
                for elem in v:
                    print(elem)
            else:
                print(v)

cfg_reader = CfgReader()
cfg_reader.read('in.yaml')
cfg_reader.docu()

打印：

{'InFile': 'Test123.MQSC', 'Sheet': 'Test123', 'Server': 'Testsystem1'}
{'InFile': 'Test345.MQSC', 'Sheet': 'Test345', 'Server': 'Testsystem2'}
{'B': 'MQVersion', 'A': 'ServerName', 'C': 'BrokerVersion'}
{'B': 'MQ Version', 'A': 'Server Name', 'C': 'Broker Version'}
{'XLOutFile': 'TestFile1.xlsx', 'InPath': '/home/me', 'OutPath': '/home/me'}

还请注意一些一般性的事情，你应该知道关闭

我使用 ruamel.yaml（免责声明：我是该包的作者），它支持 YAML 1.2（PyYAML 支持 2005 年的 1.1 标准）。对于您的目的，它们的行为相同。
不要使用 file() 它在 Python3 中不可用，请使用 open()
将相同的值两次赋给同一个属性是没有意义的(self.stream = self.stream = ...)

您打开的 file/stream 永远不会关闭，您可能需要考虑使用

with open(infilename) as self.stream:
    self.ymldata = yaml.load(self.stream)

class 按照惯例，名称应以大写字符开头。

pyyaml 3.11 将字典传递给迭代器？

pyyaml 3.11 pass dictionary to iterator?

python

dictionary

iterator

pyyaml

python-2.7