[Python]三级联动查询

# 以字典的形式 保存相关数据
menu = {
    '本部': {
        '工程': {
            '工程三課'}
    },
    '五': {
        '工程': {
            '技术1',
            '技术2',
            '技术3',
            '技术4',
            '技术5',
        },
        '製造': {
            '生產1',
            '生產2'
        },
        '專案': {
            '品保1',
            '品保2',
            '品保3'
        }
    }
}
back_flag = False  # 用于返回上一层
exit_flag = False  # 用于退出整个程序
while not back_flag and not exit_flag:
    for key in menu:
        print(key)
    choice = input(">>").strip()
    if choice in menu:
        while not back_flag and not exit_flag:  # 让程序停在第二层
            for key2 in menu[choice]:
                print(key2)
            choice2 = input('>>').strip()
            if choice2 == 'q':
                back_flag = True
            if choice2 in menu[choice]:
                while not back_flag and not exit_flag:
                    for key3 in menu[choice][choice2]:
                        print(key3)
                    choice3 = input('>>').strip()
                    if choice3 == 'q':
                        back_flag = True
                    else:
                        back_flag = False
                else:
                    back_flag = False
        else:
            back_flag = False