【PyYaml】yml教程 pyyaml庫介紹及yml寫 yml讀取

Yaml

yaml語言教程

PyYAML

源碼： https://github.com/yaml/pyyaml

安裝

# pip命令行安裝
pip install PyYAML

# 下載源碼的安裝
python setup.py install

導入

import yaml

讀取yaml文件

def read_yaml(yml_file, mode='r', encoding='utf-8'):
    """ yaml中內容讀取並轉化爲Python對象

    :param yml_file:
    :param mode:
    :param encoding:
    :return:
    """
    # safe_load_all() 打開多個文檔
    with open(yml_file, mode=mode, encoding=encoding) as y_file:
        # .load 是非推薦的 不安全的編碼方式
        # content = yaml.load(y_file.read(), yaml.FullLoader)
        # .safe_load 安全編碼方式
        # If you don't trust the input stream, you should use:
        return yaml.safe_load(y_file)

寫入yaml文件

def write_yaml(yaml_file, data, mode='w', encoding='utf-8', is_flush=True):
    """ Python對象轉換爲 yaml

    :param yaml_file:
    :param data:
    :param mode:
    :param encoding:
    :param is_flush:
    :return:
    """
    with open(yaml_file, mode=mode, encoding=encoding) as y_file:
        # yaml.dump(data, stream=y_file)
        # allow_unicode 解決寫入亂碼的問題
        yaml.safe_dump(data, stream=y_file, allow_unicode=True)
        if is_flush:
            y_file.flush()