Python簡單實現多級菜單

# -*- coding: utf-8 -*-
# @Time    : 2018-06-01 13:40
# @Author  : 超人
# @Email   : [e-mail]
# @File    : 簡單實現多級菜單功能 (輸入s返回首菜單,輸入b返回上一菜單,輸入o終止程序)
# @Software: PyCharm

current_layer = menu#字典,多級菜單
parent_layers=[]
while True:
    for key in current_layer:
        print(key)
    choice = input('>>>>>>>>>>>').strip()
    if len(choice) == 0: continue
    if choice in current_layer:
        # print(choice)
        parent_layers.append(current_layer)
        current_layer = current_layer[choice]
    elif choice == "b":
        if not len(parent_layers)==0:
            current_layer = parent_layers.pop()
        else:
            current_layer=menu
    elif choice =="s":
        current_layer = menu
    elif choice == "o":
        break
    else:
        print('沒有此選項,重新輸入'.center(50,"-"))