Python如何读取yaml文件多层菜单（代码）-Python教程-PHP中文网

首页

后端开发

Python教程

Python如何读取yaml文件多层菜单（代码）

不言

Oct 08, 2018 pm 03:10 PM

python

本篇文章给大家带来的内容是关于Python如何读取yaml文件多层菜单（代码），有一定的参考价值，有需要的朋友可以参考一下，希望对你有所帮助。

需要用到的Python知识点

Python的对象属性方法；
用到字典｛key：value｝值的提取；
列表的增加；
if循环结合break的使用；
yaml文件读取；

代码如下：

#!/usr/bin/python34
import sys,os,re,yaml,time
#reload(sys)
#sys.setdefaultencoding(&#39;utf-8&#39;)  

######################对input输入字符类型判断并转化#####################
def input_handle(s):
   if str.isdigit(s):                   ###对输入是否是数字进行判断
       s = int(s)                       ###如果输出的是个数字，则转化为整数类型
   return s
   
   # try:
   #     eval(s)                         ###eval将字符串str当成有效的表达式来求值并返回计算结果
   # except NameError:                   ###名称错误
   #     return s
   # except SyntaxError:                 ###语法错误
   #     return s
   # else:
   #     return eval(s)
    

####################框架函数######################
def framework(province=&#39;&#39;,city=&#39;&#39;,county=&#39;&#39;):
    os.system(&#39;cls&#39;)                             ###清屏###
    print(&#39;&#39;&#39;
******************************************************************
*                                                                *
*                     欢迎访问全国省市查询系统                   *
*                                                                *
******************************************************************


+-----------------------------------------------------------------
|           省份:  %s 
|
|           市(区): %s 
|
|           县(区): %s
+-----------------------------------------------------------------
&#39;&#39;&#39; % (province,city,county))
    
######################输出展示函数#################
def show(province_name=&#39;&#39;,city_name=&#39;&#39;,county_name=&#39;&#39;):
    output= &#39;&#39;&#39;
******************************************************************
*                                                                *
                  美丽的%s %s %s
*                           欢迎您                               *
                  这里的山美，水美，妹子更美                     
*                                                                *
******************************************************************  
&#39;&#39;&#39;
    print(output % (province_name,city_name,county_name))


###################菜单第一层省份或直辖市输出函数#################
def province_show(province_list):
  
    ############申明全局变量####################
    global P_NAME
    global C_NAME
    global X_NAME
    global FLAG_M

    province_dict = {}            
    ############对省份或直辖市列表参数进行遍历并加上数字编号############### 
    for k,v in enumerate(province_list,1):
            province_dict[k] = v
            print(&#39;%d . %s&#39; % (k,v) + &#39;\t&#39;,end=&#39;&#39;)       ###加上end参数，取消默认换行###
            if k % 4 == 0:                               ###按4列换行###
                print()
    print(&#39;\n================================================================&#39;)
    print(&#39;q : Exit&#39;)

    ###############键盘读入编号或省份，可以输入汉字#################
    province_index = input(&#39;请输入编号或省份 : &#39;)

    ###############如果输入非空，对输入进行判断并转化类型###########
    if len(province_index) != 0:
        province_index = input_handle(province_index)

    if province_index == &#39;q&#39;:                            ###如果输入为q,则退出程序###
        sys.exit(0)     
    elif province_index in province_dict.keys():         ###如果输入为数字编号，则从字典中获取具体省份或直辖市名称###
        P_NAME = province_dict[province_index]           ###对全局变量赋值省份名称###
    elif province_index in province_dict.values():       ###如果输入为具体省份，则从字典中获取具体省份或直辖市名称###
        P_NAME = province_index                          ###对全局变量赋值省份名称###
    else:
        P_NAME = &#39;&#39;                                      ###输入其他字符，赋值全局变量为空### 

    while P_NAME:                                        ###全局变量不为空进行循环###
        framework(P_NAME,C_NAME,X_NAME)                  ###调用框架###
        if type(yaml_dict[P_NAME]) is list:
            city_show(P_NAME)                            ###调用城市函数，并传入省份值###
            if FLAG_M == &#39;b&#39;:                            ###城市函数输入b，返回上一层，重新选择省份###
                break
        else:
            show(P_NAME)                                 ###调用输出展示函数### 
            time.sleep(5)
            P_NAME = &#39;&#39;
            break
    else:
        print(&#39;输入错误，请重新输入!&#39;)                   ###P_NAME为空，即输入错误，重新输入###
        time.sleep(2)
    
##############菜单第二层城市输出函数#######################
def city_show(province_name):

    ############申明全局变量###############
    global P_NAME
    global C_NAME
    global X_NAME
    global FLAG_M
    
    city_name = &#39;&#39;                                       ###定义城市变量默认为空###
    city_list = yaml_dict[province_name]                 ###定义赋值城市列表###
    city_dict = {}                                       ###定义城市编号和名称字典###
    city_county_dict = {}                                ###定义地级市和下属区县字典###

    ############对城市列表参数进行遍历并加上数字编号############### 
    for k,v in enumerate(city_list,1):
        if type(v) is str:                               ###直辖市只有二层菜单，第二层为直接的各区或县，值类型为具体字符###
            city_dict[k] = v                             ###对直辖市下的区或县进行新字典赋值，方便查询和展示###
            print(&#39;%d . %s&#39; % (k,v) + &#39;\t&#39;,end=&#39;&#39;)       ###末尾加&#39;,&#39;，取消默认换行###
        elif type(v) is dict:                            ###其他省份有三层菜单，第二层为各地级市，值类型字典###
            for kk,vv in v.items():                      ###对地级市的字典进行遍历###
                city_dict[k] = kk                        ###对其他省份下的地级市进行新字典赋值，方便查询和展示### 
                city_county_dict[kk] = vv                ###对二层地级市和三层县市重新赋值新字典###             
                print(&#39;%d . %s&#39; % (k,kk) + &#39;\t&#39;,end=&#39;&#39;)  ###加上end参数，取消默认换行### 
        else:
            pass
                
        if k % 4 == 0:                                   ###按4列换行###
            print()
    print(&#39;\n================================================================&#39;)
    print(&#39;q : Exit   b : Back&#39;)

    ###############键盘读入编号或区市，可以输入汉字#################
    city_index = input(&#39;请输入编号或区市 : &#39;)
    
    ###############如果输入非空，对输入进行判断并转化类型###########
    if len(city_index) != 0:
        city_index = input_handle(city_index)

    if city_index == &#39;q&#39;:                                ###如果输入为q,则退出程序###
        sys.exit(0)
    elif city_index == &#39;b&#39;:                              ###如果输入为b，则返回上一层，重新选择省份或直辖市###
        (P_NAME,C_NAME,FLAG_M) = (&#39;&#39;,&#39;&#39;,&#39;b&#39;)             ###全局变量P_NAME,C_NAME设置为空，FLAG_M设置为b，则返回上一层###
        return                                           ###直接返回，不进行函数以下的操作###
    elif city_index in city_dict.keys():                 ###如果输入为数字编号，则从字典中获取具体城市名称###
        city_name = city_dict[city_index]                ###赋值地级市的名称，并对全局变量进行赋值###
        (P_NAME,C_NAME,FLAG_M) = (province_name,city_name,&#39;&#39;)
    elif city_index in city_dict.values():               ###如果输入为城市名称，则从字典中获取具体省份名称###
        city_name = city_index                           ###赋值地级市的名称，并对全局变量进行赋值###
        (P_NAME,C_NAME,FLAG_M) = (province_name,city_name,&#39;&#39;)
    else:
        pass                                             ###如果输入其他字符，则不做任何操作###
    
    if city_name:                                        ###如果地级市名字不为空，即键盘输入为要求字符###
        if city_name in city_county_dict.keys():         ###判断是省份的地级市名字### 
            while C_NAME:                                ###环境变量C_NAME不为空###
                framework(P_NAME,C_NAME,X_NAME)          ###调用框架函数，并将省份名字和地级市名字传入###

                ###调用三层区县显示函数，并传入具体变量###
                county_show(P_NAME,C_NAME,city_county_dict[city_name])    
                if FLAG_N == &#39;b&#39;:                        ###三层区县函数输入b，返回上一层，重新选择地级市###
                    break
            else:
                print(&#39;输入错误，请重新输入!&#39;)           ###C_NAME为空，即输入错误，重新输入###
                time.sleep(2)
        else:                                            ###判断是直辖市的区或县名字###
           show(P_NAME,C_NAME)                           ###调用输出展示函数### 
           time.sleep(5)
    else:                                                ###输入非要求字符，提示重新输入###
        print(&#39;输入错误，请重新输入!&#39;)
        time.sleep(2)
     
##############菜单第三层区县输出函数#######################
def county_show(province_name,city_name,county_list):

    ############申明全局变量####################
    global P_NAME
    global C_NAME
    global X_NAME
    global FLAG_N

    county_name = &#39;&#39;                                     ###定义三级区县变量默认为空###
    county_dict = {}                                     ##定义赋值区县字典###
 
    ############对区县列表参数进行遍历并加上数字编号############### 
    for k,v in enumerate(county_list,1):                 
        if type(v) is str:                               ###第三层为直接的各区或县，值类型为具体字符str###
            county_dict[k] = v                           ###对区或县进行新字典赋值，方便查询和展示###
            print(&#39;%d . %s&#39; % (k,v) + &#39;\t&#39;,end=&#39;&#39;)       ###加上end参数，取消默认换行### 

        if k % 4 == 0:                                   ###按4列换行###
            print()
    print(&#39;\n================================================================&#39;)
    print(&#39;q : Exit   b : Back&#39;)

    ###############键盘读入编号或区县，可以输入汉字#################
    county_index = input(&#39;请输入编号或区县 : &#39;)

    ###############如果输入非空，对输入进行判断并转化类型###########
    if len(county_index) != 0:                           
        county_index = input_handle(county_index)

    if county_index == &#39;q&#39;:                              ###如果输入为q,则退出程序###
        sys.exit(0)
    elif county_index == &#39;b&#39;:                            ###如果输入为b，则返回上一层，重新选择第二层地级市###
        (P_NAME,C_NAME,X_NAME,FLAG_N) = (province_name,&#39;&#39;,&#39;&#39;,&#39;b&#39;)   ###全局变量C_NAME设置为空，FLAG_M设置为b，则返回上一层###
        return                                           ###直接返回，不进行函数以下的操作### 
    elif county_index in county_dict.keys():             ###如果输入为数字编号，则从字典中获取具体区县名称###
        county_name = county_dict[county_index]          ###赋值区县的名称，并对全局变量进行赋值###
        (P_NAME,C_NAME,X_NAME,FLAG_N) = (province_name,city_name,county_name,&#39;&#39;)
    elif county_index in county_dict.values():           ###如果输入为区县名称，则从字典中获取具体区县名称###
        county_name = county_index                       ###赋值区县的名称，并对全局变量进行赋值###
        (P_NAME,C_NAME,X_NAME,FLAG_N) = (province_name,city_name,county_name,&#39;&#39;)
    else:                                                ###如果输入其他字符，则不做任何操作### 
        (P_NAME,C_NAME,X_NAME,FLAG_N) = (province_name,city_name,&#39;&#39;,&#39;&#39;)

    if county_name:                                      ###如果区县名字不为空，即键盘输入为要求字符###
        show(P_NAME,C_NAME,X_NAME)                       ###调用输出展示函数### 
        time.sleep(5)
    else:
        print(&#39;输入错误，请重新输入!&#39;)                   ###输入非要求字符，提示重新输入###
        time.sleep(2)
    


##########################################主程序###############################

###############读取yaml格式文件#######################
fd = open(r&#39;G:\sicps\test\pro&#39;,encoding=&#39;utf-8&#39;)
yaml_dict = yaml.load(fd)

###############定义全局变量###########################
P_NAME = &#39;&#39;                                             ###省份或直辖市全局变量###
C_NAME = &#39;&#39;                                             ###各省地级市或直辖市区县的全局变量###
X_NAME = &#39;&#39;                                             ###各省地级市下的区或县 全局变量###
FLAG_M = &#39;&#39;                                             ###退出菜单第二层，返回上一层循环的变量###
FLAG_N = &#39;&#39;                                             ###退出菜单第三层，返回上一层循环的变量###

###############获取省份或直辖市的列表#################
prov_list = yaml_dict.keys()

###############主循环开始#############################
while True:
    framework(P_NAME,C_NAME,X_NAME)                     ###调用框架函数，显示初始状态###
    province_show(prov_list)                            ###调用第一层省份或直辖市输出函数###

登录后复制

以上是Python如何读取yaml文件多层菜单（代码）的详细内容。更多信息请关注PHP中文网其他相关文章！

本站声明

本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系admin@php.cn