python脚本实现判断yaml脚本格式是否正确

脚本实现输入yaml格式的文件名 然后如果输入格式正确则返回该yaml格式的列表。

#!/usr/bin/python2.7 
# -*- coding: utf-8 -*-
# @Date    : 2018-07-21 23:23:56
# @Author  : Zhou Jiaan
# @Version : V1.0
import os
import yaml
import sys

def playbookmonitor(playbook):
  with open(playbook) as f:
    print(yaml.load(f))

def main():
    try:
        test_playbook = sys.argv[1]
        print(test_playbook)
    except Exception as e:
        print(e)
        test_playbook = input("请输入需要检测的playbook:")

    else:
        print("输出test_playbook:{}".format(test_playbook))
    finally:
        playbook = test_playbook
    #    print(playbook)
        playbookmonitor(playbook)

if __name__ == '__main__':
    main()