python生成yaml文件_Python读取yaml生成菜单

需求:

每个location有对应的env, 每个env有对应的info

location1: dev:bar qa:foo uat:xxx

location2: dev:xxx qa:xxx

需要生成菜单，可以选择任意location中的env

Please choose location:

1: location1

2: location2

Enter your choice [1-2] : 1

1: dev

2: qa

3: uat

Enter your choice [1-3] : 2

You chosen location1 qa foo

思路及实现:

将配置信息存入配置文件中，在此选用yaml

利用python字典及列表的操作获取相应的值。

config.yml

location1:

- env: dev

info: bar

- env: qa

info: foo

- env: uat

info: xxx

location2:

- env: dev

info: xxx

- env: qa

info: xxx

menu.py

# -*- coding: utf-8 -*-

#author: firxiao

#date:20180115

#generate a menu from yaml config file.

import yaml

#读取配置文件 cfg为字典

with open("config.yml", 'r') as ymlfile:

cfg = yaml.load(ymlfile)

print("Please choose location:")

#打印带序号排序过的字典

for i,location in enumerate(sorted(cfg)):

print('%d: %s'% (i + 1,location))

#获取字典长度供菜单使用

lr = len(cfg)

choice = int(input('Enter your choice [1-%d] : ' % (lr)))

while True:

if not choice: break

# 将选项变为列表并根据输入序号选取

location = list(sorted(cfg))[choice - 1]

# 打印字典中key的value

for i,d in enumerate(cfg[location]):

print('%d: %s'% (i+1,d['env']))

le = len(cfg[location])

choice = int(input('Enter your choice [1-%d] : ' % (le)))

# 同理,将选项转换为列表并取出相应的value

print('You chosen %s %s %s'% (location,list(cfg[location])[choice - 1]['env'],list(cfg[location])[choice - 1]['info']))

env = list(cfg[location])[choice - 1]['env']

info = list(cfg[location])[choice - 1]['info']

break