python yaml文件读写

#test.yaml

people_info:
- Age: 28
  name:
    End_name: Xin
    First_name: zhao
- Age: 27
  name:
    End_name: yue
    First_name: sun
- Age: 29
  name:
    End_name: xiaobing
    First_name: peng
- Age: 25
  name:
    End_name: yuanchang
    First_name: liu

import yaml
import os

## 读取yaml文件
def readyaml(file):
    if os.path.isfile(file):
        fr = open(file, 'r')
        yaml_info = yaml.load(fr)
        fr.close()
        return yaml_info
    return None
##  向yaml文件中写入配置
def writeyaml(file, data):
    fr = open(file, 'w')
    yaml.dump(data, fr)
    fr.close()
##显示yaml文件
def display_info(data_list):
    for tmp in data_list:
        print (tmp)
def main():
    yaml_file = "test.yaml"

    yaml_info =readyaml(yaml_file)
    # print(yaml_info)
    # print ('old_yaml_info is ......')
    # print ("-"*150)
    display_info(yaml_info['people_info'])
    # print ('*'*150)
    test_data = {'Age': 21, 'name': {'First_name': 'tan', 'End_name': 'Xiao'}}
    yaml_info['people_info'].append(test_data)
    writeyaml(yaml_file, yaml_info)
    print ("new_yaml_info is ......")
    print(readyaml(yaml_file))
    # display_info(readyaml(yaml_file)['people_info'])
if __name__ == '__main__':
    main()