python2.7 读取yaml文件

前言

之前看了很多帖子，都在使用yaml配置测试用例，一直没有尝试，最近整理openapi脚本，第一次使用了yaml配置接口条件，于是顺便学习了一下yaml语法。总结的不到位的地方，欢迎批评指正。
备注：这篇文章主要参考了阮一峰的文章，这是传送门,以及官方wiki文档
语言介绍部分就不多说了，网上有很多介绍，直接从数据结构开始说吧。
同时记录中文读取显示乱码的解决方法。

对象

这应该算是最简单的一种格式了，就是一组键值对

1. 新建data.yaml文件，内容如下：

 name: waiqin365
 code: cus001
 type: market

脚本如下：

import yaml

f=open('data.yaml')
data = yaml.load(f)
print data

转换为python后：

{'code': 'cus001', 'type': 'market', 'name': 'waiqin365'}

修改yaml文件，格式如下：

 name: waiqin365
 detail:
  code: cus001
  type: market

转换为python后：

{'name': 'waiqin365', 'detail': {'code': 'cus001', 'type': 'market'}}

数组

一组连词线开头的行，构成一个数组

1.新建data.yaml,格式如下：

- cus001
- cus002
- cus003

转换为python后：

['cus001', 'cus002', 'cus003']

修改yaml文件，格式如下：

- cus001:
    code: 001
- cus002:
    code: 002
    type: 0022
- cus003

转换为python后：

[{'cus001': {'code': 'code001'}}, {'cus002': {'code': 'code002', 'type': 'tyoe002'}}, 'cus003']

多个对象

1.新建data.yaml,格式如下：

备注：这是官方wiki文档的例子

---
name: The Set of Gauntlets 'Pauraegen'
description:
    A set of handgear with sparks that crackle across its knuckleguards.
---
name: The Set of Gauntlets 'Paurnen'
description:
    A set of gauntlets that gives off a foul,acrid odour yet remains untarnished.
---
name: The Set of Gauntlets 'Paurnimmen'
description:
    A set of handgear, freezing with unnatural cold.

需要修改python脚本,改用load_all来解析文档：

import yaml

f=open('data.yaml')
for data in yaml.load_all(f):
    print data

转换为python后：

{'name': "The Set of Gauntlets 'Pauraegen'", 'description': 'A set of handgear with sparks that crackle across its knuckleguards.'}
{'name': "The Set of Gauntlets 'Paurnen'", 'description': 'A set of gauntlets that gives off a foul,acrid odour yet remains untarnished.'}
{'name': "The Set of Gauntlets 'Paurnimmen'", 'description': 'A set of handgear, freezing with unnatural cold.'}

对象和数组结合

1.新建data.yaml,格式如下：

cus:
  - cus001
  - cus002
  - cus003
detail:
  type: {type1: a}
  dept: dept1

转换为python后：

{'cus': ['cus001', 'cus002', 'cus003'], 'detail': {'dept': 'dept1', 'type': {'type1': 'a'}}}

读取中文显示问题

当data.yaml文件中，含有中文时，读取后显示有问题：
如下，data.yaml文件为：

cus:
  - 终端001
  - 终端002
  - 终端003
detail:
  type: {type1: a}
  dept: dept1

用上面脚本，直接打印出来结果为：

{'cus': [u'\u7ec8\u7aef001', u'\u7ec8\u7aef002', 'cus003'], 'detail': {'dept': 'dept1', 'type': {'type1': 'a'}}}

如何才能直接将 cus:['终端001','终端002','终端003'] 打印出来呢？

修改脚本如下：

f=open('data.yaml',)
data = yaml.load(f)
print data
print (repr(data).decode('unicode-escape'))

对比一下，打印结果如下：

{'cus': [u'\u7ec8\u7aef001', u'\u7ec8\u7aef002', 'cus003'], 'detail': {'dept': 'dept1', 'type': {'type1': 'a'}}}
{'cus': [u'终端001', u'终端002', 'cus003'], 'detail': {'dept': 'dept1', 'type': {'type1': 'a'}}}

ok问题解决，解决方法，主要参考知乎一片文章