1. 程式人生 > 實用技巧 >Python讀取yaml檔案

Python讀取yaml檔案

yaml簡介

1.yaml [ˈjæməl]: Yet Another Markup Language :另一種標記語言。yaml 是專門用來寫配置檔案的語言,非常簡潔和強大,之前用ini也能寫配置檔案,看了yaml後,發現這個更直觀,更方便,有點類似於json格式。在自動化測試用的相當多所以需要小夥伴們要熟練掌握

2.yaml基本語法規則:

  • 大小寫敏感
  • 使用縮排表示層級關係
  • 縮排時不允許使用Tab鍵,只允許使用空格。
  • 縮排的空格數目不重要,只要相同層級的元素左側對齊即可
  • #表示註釋,從這個字元一直到行尾,都會被解析器忽略,這個和python的註釋一樣

3.yaml支援的資料結構有三種:

  • 物件:鍵值對的集合,又稱為對映(mapping)/ 雜湊(hashes) / 字典(dictionary)
  • 陣列:一組按次序排列的值,又稱為序列(sequence) / 列表(list)
  • 純量(scalars):單個的、不可再分的值。字串、布林值、整數、浮點數、Null、時間、日期

安裝yaml

1.使用pip安裝pyyaml模組

pip install pyyaml

1.python各資料型別在yaml中的書寫格式

1.yaml裡面的鍵值對,也就是python裡面的字典(dict)資料型別,比如python裡面的字典,用yaml如何去寫

# python3.6
{
"user": '測試',
"psw": 123456,
}

在yaml檔案裡可以這樣寫:

# yaml
user:測試
psw: 123456 

字典巢狀字典

# python3.6
"nb1": {
        "user": "測試",
        "psw": "123456,
        }

3.yaml檔案裡可以這樣寫:
# yaml
nb1:
    user: 測試
    psw: 123456

序列(list)

1.yaml裡面寫一個數組,前面加一個‘-’符號,如下

- admin1: 123456
- admin2: 111111
- admin3: 222222

對應python裡面的list

[{'admin1': 123456},
{'admin2': 111111}, 
{'admin3': 222222}]

注意:數字讀出來的是int或float型別

純量(str)

1.int和float型別的數字

n1: 12.30 

對應python

{'n1': 12.3}

2.布林值用true和false表示

n2: true 
n3: false 

對應python

{'n2': True, 'n3': True}

3.None用~表示

n4: ~

對應python

{'n4': None}

4.時間採用 ISO8601 格式。

time1: 2001-12-14t21:59:43.10-05:00 

對應python

{'time1': datetime.datetime(2001, 12, 15, 2, 59, 43, 100000)}

5.日期採用複合 iso8601 格式的年、月、日表示。

date1: 2017-07-31

對應python

{'date1': datetime.date(2017, 7, 31)}

6.使用兩個感嘆號,強制轉換資料型別。

# int轉str
n6: !!str 123  

對應python

{'n6': '123'}
# bool值轉str
n7: !!str true

對應python

{'n7': 'true'}

7.在yaml檔案寫入以下內容:

n1: 12.30
n2: true
n3: false
n4: ~
time1: 2018-04-18t21:59:43.10+08:00
date1: 2018-04-18
n6: !!str 123
n7: !!str true

python讀取結果:

{'n1': 12.3, 
'n2': True, 
'n3': False, 
'n4': None, 
'time1': datetime.datetime(2018, 4, 18, 13, 59, 43, 100000), 
'date1': datetime.date(2018, 4, 18), 
'n6': '123', 
'n7': 'true'}


混合使用

1.list巢狀dict,在yaml裡面寫入如下內容:

- user: admin1
  psw: '123456'

- user: admin2
  psw: '111111'

- user: admin3
  psw: '222222'

用python讀取出來的結果:

[{'user': 'admin1', 'psw': '123456'},
{'user': 'admin2', 'psw': '111111'},
{'user': 'admin3', 'psw': '222222'}]

2.dict巢狀list,在yaml裡面寫入如下內容:

nub1:
    - admin1
    - '123456'

nb2:
    - admin2
    - '111111'

nb3:
    - admin3
    - '222222'

用python讀取出來的結果:

{'nub1': ['admin1', '123456'],
'nb2': ['admin2', '111111'], 
'nb3': ['admin3', '222222']}


3.讀取yaml檔案

讀取login.yaml檔案的內容

3.寫入資料到yaml檔案