Python中yaml文件的讀寫（使用PyYAML庫）。最近在搭建自動化測試項目過程中經(jīng)常遇到y(tǒng)aml文件的讀寫，為了方便后續(xù)使用，決定記下筆記。

最近在搭建自動化測試項目過程中經(jīng)常遇到y(tǒng)aml文件的讀寫，為了方便后續(xù)使用，決定記下筆記。

一，YAML 簡介

YAML，Yet Another Markup Language的簡寫，通常用來編寫項目配置，也可用于數(shù)據(jù)存儲，相比conf等配置文件要更簡潔。

二，YAML 語法

• 支持的數(shù)據(jù)類型：

字典、列表、字符串、布爾值、整數(shù)、浮點數(shù)、Null、時間等

• 基本語法規(guī)則：

1、大小寫敏感

2、使用縮進表示層級關(guān)系

3、相同層級的元素左側(cè)對齊

4、鍵值對用冒號 “:” 結(jié)構(gòu)表示，冒號與值之間需用空格分隔

5、數(shù)組前加有 “-” 符號，符號與值之間需用空格分隔

6、None值可用null 和 ~ 表示

7、多組數(shù)據(jù)之間使用3橫杠---分割

8、# 表示注釋，但不能在一段代碼的行末尾加 #注釋，否則會報錯

注意：網(wǎng)上查找到各種博客都提到y(tǒng)aml縮進時不能使用tab鍵，但我在pycharm編輯器里實際使用時是可以使用tab鍵進行縮進的，讀寫時并沒有報錯！

三，安裝第三方y(tǒng)aml文件處理庫PyYAML

python沒有自帶的處理yaml文件的庫，需要下載第三方庫PyYAML 或 ruamel.yaml ，這里我們安裝PyYAML。

pip install pyyaml
# 下載速度慢的話加上清華鏡像源
pip install pyyaml -i https://pypi.tuna.tsinghua.edu.cn/simple

四，讀取yaml文件

1，從yaml中讀取字典

yaml中的字典格式如下：

# yaml文件，文件名為yamlData

os: Android
osVersion: 10
account:
  username: xiaoqq
  password: 123456
deviceName: null
appPackage: ~
bool1: True

讀取字典代碼：

# @author: 給你一頁白紙
import yaml
with open('./yamlData.yml', 'r', encoding='utf-8') as f:
    result = yaml.load(f.read(), Loader=yaml.FullLoader)
print(result, type(result))
print(result['os'], type(result['os']))
print(result['osVersion'], type(result['osVersion']))
print(result['account'], type(result['account']))
print(result['account']['username'])
print(result['deviceName'])
print(result['appPackage'])
print(result['bool1'], type(result['bool1']))

讀取結(jié)果：

{'os': 'Android', 'osVersion': 10, 'account': {'username': 'xiaoqq', 'password': 123456}, 'deviceName': None, 'appPackage': None} <class 'dict'>
Android <class 'str'>
10 <class 'int'>
{'username': 'xiaoqq', 'password': 123456} <class 'dict'>
xiaoqq
None
None
True <class 'bool'>

從讀取結(jié)果可以看出：

1，讀取出來的數(shù)據(jù)不會改變原數(shù)據(jù)類型，即yaml里是什么數(shù)據(jù)類型，讀出來就是什么類型。

2，Loader=yaml.FullLoader參數(shù)不寫的話對結(jié)果不會有影響，但運行時會出現(xiàn)警告信息。

3，yaml.load(f.read(), Loader=yaml.FullLoader)也可以寫成yaml.load(f, Loader=yaml.FullLoader)，讀取出來的結(jié)果相同。

2，從yaml中讀取list

yaml中l(wèi)ist格式：數(shù)據(jù)前加'-' 并使用空格與數(shù)據(jù)間隔開，如下：

# yaml文件名yamlData

- Android
- 10
- null
- ~
- True

讀取list代碼：

# @author: 給你一頁白紙
import yaml
with open('./yamlData.yml', 'r', encoding='utf-8') as f:
    result = yaml.load(f.read(), Loader=yaml.FullLoader)
print(result, type(result))

讀取結(jié)果：

['Android', 10, None, None, True] <class 'list'>

3，從yaml中讀取元組

yaml中存儲元組格式：yaml中使用!!對數(shù)據(jù)類型進行轉(zhuǎn)換，yaml中tuple由list轉(zhuǎn)換而來。如下：

# yaml文件名yamlData

!!python/tuple
- Android
- 10
- null
- ~
- True

讀取元組代碼：

# @author: 給你一頁白紙
import yaml
with open('./yamlData.yml', 'r', encoding='utf-8') as f:
    result = yaml.load(f.read(), Loader=yaml.FullLoader)
print(result, type(result))

讀取結(jié)果：

('Android', 10, None, None, True) <class 'tuple'>

在實際使用中，很多的時候往往是多種類型嵌套的數(shù)據(jù)。如下yaml數(shù)據(jù)

# yaml文件名yamlData
os: Android
osVersion: 10
account:
  - username1: xiaoqq
  - password1: 123456
  - username2: Lilei
  - password2: 888888
deviceName: null
appPackage: ~
bool1: True

讀取結(jié)果：

{'os': 'Android', 'osVersion': 10, 'account': [{'username1': 'xiaoqq'}, {'password1': 123456}, {'username2': 'Lilei'}, {'password2': 888888}], 'deviceName': None, 'appPackage': None, 'bool1': True}

4，從yaml中讀取多組數(shù)據(jù)

yaml多組數(shù)據(jù)時，每組數(shù)據(jù)之間需要用3橫杠分隔'---'，如下：

os: Android
osVersion: 10
account1:
  username1: xiaoqq
  password1: 123456
---
os: ios
osVersion: 12
account1:
  username2: Lilei
  password2: 888888

從yaml中讀取多組數(shù)據(jù)時需要使用yaml.load_all()方法，返回結(jié)果為一個生成器，需要使用for循環(huán)語句獲取每組數(shù)據(jù)。代碼如下：

# @author: 給你一頁白紙
import yaml
with open('./yamlData.yml', 'r', encoding='utf-8') as f:
    result = yaml.load_all(f.read(), Loader=yaml.FullLoader)
    print(result, type(result))
    for i in result:
        print(i)

讀取結(jié)果：

<generator object load_all at 0x000001F78EBD5B48> <class 'generator'>
{'os': 'Android', 'osVersion': 10, 'account1': {'username1': 'xiaoqq', 'password1': 123456}}
{'os': 'ios', 'osVersion': 12, 'account1': {'username2': 'Lilei', 'password2': 888888}}

五，寫入yaml文件

1，單組數(shù)據(jù)寫入yaml文件

使用yaml.dump()方法，加入allow_unicode=True參數(shù)防止寫入的中文亂碼，如下：

# @author: 給你一頁白紙
import yaml
apiData = {
   "page": 1,
   "msg": "地址",
   "data": [{
      "id": 1,
      "name": "學(xué)校"
   }, {
      "id": 2,
      "name": "公寓"
      "id": 3,
      "name": "流動人口社區(qū)"
   }],
}
with open('./writeYamlData.yml', 'w', encoding='utf-8') as f:
   yaml.dump(data=apiData, stream=f, allow_unicode=True)

寫入結(jié)果：

data:
- id: 1
  name: 學(xué)校
- id: 2
  name: 公寓
- id: 3
  name: 流動人口社區(qū)
msg: 地址
page: 1

2，多組數(shù)據(jù)寫入yaml文件

使用yaml.dump_all()方法，如下：

# @author: 給你一頁白紙
import yaml
apiData1 = {
   "page": 1,
   "msg": "地址",
   "data": [{
      "id": 1,
      "name": "學(xué)校"
   }, {
      "id": 2,
      "name": "公寓"
      "id": 3,
      "name": "流動人口社區(qū)"
   }],
}
apiData2 = {
   "page": 2,
      "name": "酒店"
      "name": "醫(yī)院"
      "name": "養(yǎng)老院"
with open('./writeYamlData.yml', 'w', encoding='utf-8') as f:
   yaml.dump_all(documents=[apiData1, apiData2], stream=f, allow_unicode=True)

寫入結(jié)果：

data:
- id: 1
  name: 學(xué)校
- id: 2
  name: 公寓
- id: 3
  name: 流動人口社區(qū)
msg: 地址
page: 1
---
data:
- id: 1
  name: 酒店
- id: 2
  name: 醫(yī)院
- id: 3
  name: 養(yǎng)老院
msg: 地址
page: 2

在Python中除了PyYAML庫之外，還有ruamel.yaml庫也可以對yaml文件進行讀寫操作，后續(xù)再記筆記進行介紹。

到此這篇關(guān)于Python使用PyYAML庫讀寫yaml文件的方法的文章就介紹到這了,更多相關(guān)Python讀寫yaml文件內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！