在上一篇Python接口自動(dòng)化測(cè)試系列文章：Python接口自動(dòng)化淺析數(shù)據(jù)驅(qū)動(dòng)原理，主要介紹openpyxl操作excel，結(jié)合ddt實(shí)現(xiàn)數(shù)據(jù)驅(qū)動(dòng)。

在自動(dòng)化過程中，需要使用配置文件儲(chǔ)存數(shù)據(jù)，比如數(shù)據(jù)庫信息、賬號(hào)信息、域名等。

其中，yaml文件是一種配置文件類型，相比較ini，conf配置文件來說，更加的簡潔，操作也更加簡單，同時(shí)可以存放不同類型的數(shù)據(jù)。

以下主要介紹yaml語法、yaml存儲(chǔ)數(shù)據(jù)，封裝類讀寫yaml配置文件。

一、yaml介紹及使用

01 yaml簡介

YAML 語言（ /ˈjæməl/ ）的設(shè)計(jì)目標(biāo)，就是方便人類讀寫。

它實(shí)質(zhì)上是一種通用的數(shù)據(jù)串行化格式。

YAML 是一種非常靈活的格式，幾乎是 JSON 的超集。

除了支持注釋、換行符分隔、多行字符串、裸字符串和更靈活的類型系統(tǒng)之外，YAML 也支持引用文件，以避免重復(fù)代碼。

02 yaml語法規(guī)則

基本語法規(guī)則：

• 大小寫敏感；
• 使用縮進(jìn)表示層級(jí)關(guān)系；
• 縮進(jìn)時(shí)不允許使用Tab鍵，只允許使用空格；
• 縮進(jìn)的空格數(shù)目不重要，只要相同層級(jí)的元素左側(cè)對(duì)齊即可；
• # 表示注釋，從這個(gè)字符一直到行尾，都會(huì)被解析器忽略；

03 yaml數(shù)據(jù)結(jié)構(gòu)

• 對(duì)象：鍵值對(duì)的集合，又稱為映射（mapping）、哈希（hashes） 、字典（dictionary）；
• 數(shù)組：一組按次序排列的值，又稱為序列（sequence） 、列表（list）；
• 純量（scalars）：單個(gè)的、不可再分的值；

對(duì)象

Map（屬性和值）（鍵值對(duì)）的形式：

key:(空格)value ：表示一對(duì)鍵值對(duì)，空格不可省略。

person:
    name: vivi
    age: 18

一行寫法：

person:{name: vivi，age: 18}

相當(dāng)于JSON格式：

{"name":"vivi","age":18}

數(shù)組

一組連詞線開頭的行，構(gòu)成一個(gè)數(shù)組。數(shù)組前加有 “-” 符號(hào)，符號(hào)與值之間需用空格分隔。

color:
   - red
   - blue
   - green

一行寫法：

color: [red,blue,green]

相當(dāng)于JSON：

["red","blue","green"]

純量

單個(gè)的、不可再分的值。（如：字符串、bool值、整數(shù)、浮點(diǎn)數(shù)、時(shí)間、日期、null等）

n1: 8
n2: 8.8
n3: true
n4: false
n5: 'vivi'

二、yaml配置文件的使用

01 yaml配置文件準(zhǔn)備

在項(xiàng)目下新建一個(gè)目錄config，在目錄下新建一個(gè)文件config.yaml。

在config.yaml配置文件中寫入數(shù)據(jù)庫配置。

mysql:
  host: "127.0.0.1"
  port: 3306
  user: "vivi"
  password: "123456"
  db: "ITester"
  charset: "utf8"

02 yaml配置文件格式校驗(yàn)

剛了解yaml的同學(xué)，可能對(duì)yaml格式掌握不熟練，出現(xiàn)格式錯(cuò)誤，這里推薦一個(gè)在線網(wǎng)站，用于校驗(yàn)我們寫的yaml文件格式是否正確。

在線地址：

http://www.bejson.com/validators/yaml/

三、yaml配置文件讀寫

python中讀取yaml文件前需要安裝pyyaml和導(dǎo)入yaml模塊。

01 安裝pyYaml

在python中讀取yaml文件，需要用到第三方模塊PyYaml。

安裝命令：

pip install pyYaml

02 yaml模塊源碼解析

從yaml模塊中提取出load、dump函數(shù)。

def load(stream, Loader=None):
    """
    Parse the first YAML document in a stream
    and produce the corresponding Python object.
    """
    if Loader is None:
        load_warning('load')
        Loader = FullLoader
    loader = Loader(stream)
    try:
        return loader.get_single_data()
    finally:
        loader.dispose()
        
def dump(data, stream=None, Dumper=Dumper, **kwds):
    """
    Serialize a Python object into a YAML stream.
    If stream is None, return the produced string instead.
    """
    return dump_all([data], stream, Dumper=Dumper, **kwds)

load:

將yaml流轉(zhuǎn)化為python字典；

dump:

將python對(duì)象轉(zhuǎn)化為yaml流；

03 讀寫yaml配置文件

將讀寫yaml配置文件的類進(jìn)行封裝。

在common目錄下新建一個(gè)文件，config_handler.py用于讀寫yaml。

config_handler.py

import yaml
 
class YamlHandler:
    def __init__(self,file):
        self.file = file
 
    def read_yaml(self,encoding='utf-8'):
        """讀取yaml數(shù)據(jù)"""
        with open(self.file, encoding=encoding) as f:
            return yaml.load(f.read(), Loader=yaml.FullLoader)
            
    def write_yaml(self, data, encoding='utf-8'):
        """向yaml文件寫入數(shù)據(jù)"""
        with open(self.file, encoding=encoding, mode='w') as f:
            return yaml.dump(data, stream=f, allow_unicode=True)
 
if __name__ == '__main__':
    data = {
        "user":{
            "username": "vivi",
            "password": "123456"
        }
    }
    # 讀取config.yaml配置文件數(shù)據(jù)
    read_data = YamlHandler('../config/config.yaml').read_yaml()
    # 將data數(shù)據(jù)寫入config1.yaml配置文件
    write_data = YamlHandler('../config/config1.yaml').write_yaml(data)
    print(read_data)
 

運(yùn)行結(jié)果為：

{'mysql': 
    {
           'host': '127.0.0.1', 
           'port':3306, 
           'user': 'vivi', 
        password': '123456', 
             'db': 'ITester', 
        'charset': 'utf8'
   }
}

寫入到config1.yaml的數(shù)據(jù)。

config1.yaml

user:
  password: '123456'
  username: vivi

到此這篇關(guān)于Python接口自動(dòng)化淺析yaml配置文件原理及用法的文章就介紹到這了,更多相關(guān)Python接口自動(dòng)化yaml配置文件內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評(píng)論