python yaml用法詳解

YAML是一種直觀的能夠被電腦識別的的數據序列化格式，容易被人類閱讀，並且容易和腳本語言交互。YAML類似於XML，但是語法比XML簡單得多，對於轉化成數組或可以hash的數據時是很簡單有效的。 
PyYaml 

• 1、load() :返回一個對象 
  我們先創建一個yml文件，config.yml:

  name: Tom Smith
  age: 37
  spouse:
      name: Jane Smith
      age: 25
  children:
   - name: Jimmy Smith
     age: 15
   - name1: Jenny Smith
     age1: 12

• 讀取yml文件：

  import yaml
  f = open(r'E:\AutomaticTest\Test_Framework\config\config.yml')  #  傳入文件路徑
  y = yaml.load(f)
  print (y)

  結果：
  {'name': 'Tom Smith', 'age': 37, 'spouse': {'name': 'Jane Smith', 'age': 25}, 'children': [{'name': 'Jimmy Smith', 'age': 15}, {'name1': 'Jenny Smith', 'age1': 12}]}

  2、load_all()生成一個迭代器 
  如果string或文件包含幾塊yaml文檔，你可以使用yaml.load_all來解析全部的文檔

  import yaml
  f = '''
  ---
  name: James
  age: 20
  ---
  name: Lily
  age: 19
  '''
  y = yaml.load_all(f)
  for data in y:
      print(data)

  執行結果：
  {'name': 'James', 'age': 20}
  {'name': 'Lily', 'age': 19}

  3、yaml.dump 將一個python對象生成為yaml文檔

  import yaml
  aproject = {'name': 'Silenthand Olleander',
              'race': 'Human',
              'traits': ['ONE_HAND', 'ONE_EYE']
              }
  
  print(yaml.dump(aproject，))

  執行結果：
  name: Silenthand Olleander
  race: Human
  traits: [ONE_HAND, ONE_EYE]

• yaml.dump接收的第二個參數一定要是一個打開的文本文件或二進制文件，yaml.dump會把生成的yaml文檔寫到文件里

  import yaml
  
  aproject = {'name': 'Silenthand Olleander',
              'race': 'Human',
              'traits': ['ONE_HAND', 'ONE_EYE']
              }
  f = open(r'E:\AutomaticTest\Test_Framework\config\config.yml','w')
  yaml.dump(aproject,f)
  print(yaml.load( open(r'E:\AutomaticTest\Test_Framework\config\config.yml')))

  執行結果：
  {'name': 'Silenthand Olleander', 'race': 'Human', 'traits': ['ONE_HAND', 'ONE_EYE']}

  4、yaml.dump_all()將多個段輸出到一個文件中

  import yaml
  
  obj1 = {"name": "James", "age": 20}
  obj2 = ["Lily", 19]
  
  with open(r'E:\AutomaticTest\Test_Framework\config\config.yml', 'w') as f:
      yaml.dump_all([obj1, obj2], f)

  輸出到文件：
  {age: 20, name: James}
  --- [Lily, 19]

   5、構造器(constructors)、表示器(representers)、解析器(resolvers ) 

• yaml.YAMLObject 
  yaml.YAMLObject用元類來注冊一個構造器（也就是代碼里的 init() 方法），讓你把yaml節點轉為Python對象實例，用表示器（也就是代碼里的 repr() 函數）來讓你把Python對象轉為yaml節點，看代碼：

  import yaml
  class Person(yaml.YAMLObject):
      yaml_tag = '!person'
  
      def __init__(self, name, age):
          self.name = name
          self.age = age
  
      def __repr__(self):
          return '%s(name=%s, age=%d)' % (self.__class__.__name__, self.name, self.age)
  
  james = Person('James', 20)
  
  print (yaml.dump(james))  # Python對象實例轉為yaml
  
  lily = yaml.load('!person {name: Lily, age: 19}')
  
  print (lily)  # yaml轉為Python對象實例

  輸出：
  !person {age: 20, name: James}
  
  Person(name=Lily, age=19)

• yaml.add_constructor 和 yaml.add_representer 
  你可能在使用過程中並不想通過上面這種元類的方式，而是想定義正常的類，那么，可以用這兩種方法

  import yaml
  
  
  class Person(object):
      def __init__(self, name, age):
          self.name = name
          self.age = age
  
      def __repr__(self):
          return 'Person(%s, %s)' % (self.name, self.age)
  
  james = Person('James', 20)
  print (yaml.dump(james))  # 沒加表示器之前
  
  
  def person_repr(dumper, data):
      return dumper.represent_mapping(u'!person', {"name": data.name, "age": data.age})  # mapping表示器，用於dict
  
  yaml.add_representer(Person, person_repr)  # 用add_representer方法為對象添加表示器
  print (yaml.dump(james))  # 加了表示器之后
  
  
  def person_cons(loader, node):
      value = loader.construct_mapping(node)  # mapping構造器，用於dict
      name = value['name']
      age = value['age']
      return Person(name, age)
  
  yaml.add_constructor(u'!person', person_cons)  # 用add_constructor方法為指定yaml標簽添加構造器
  lily = yaml.load('!person {name: Lily, age: 19}')
  print (lily)

  !!python/object:__main__.Person {age: 20, name: James}
  !person {age: 20, name: James}
  Person(Lily, 19)

  下面添加了構造器，能夠把 !person 標簽轉化為Person對象

 

原文地址：https://www.cnblogs.com/klb561/p/9326677.html