用Python写一个点餐的小程序

需求：

1 需要一个二级菜单返回键
2 我应该可以选择商品，输入购买数量
3 商品需要价格，购买进行价格核算
4 取消已经购买的商品

代码：

dicts = {
    '1':{'主食':[['馒头',1],['米饭',2],['花卷',2],['挂面',4]]},
    '2':{'凉菜':[['拌黄瓜',1],['拌猪耳朵',1],['皮蛋豆腐',1],['雷焦茄子',1],['五彩拉皮',1]]},
    '3':{'饮料':[['可乐',1],['雪碧',1],['芬达',1]]},
    '4':{'热菜':[['鱼香肉丝',1],['宫保鸡丁',1],['老爆三',1],['糖醋里脊',1],['八珍豆腐',1]]}
}
menu = {}
flag=0
while flag==0:
    print("+"*15+'欢迎光临大地瓜餐厅，请点餐'+'+'*15)
    for type in dicts:
        print(' '*15+type+'.'+list(dicts[type].keys())[0])

    c1 = input('请您输入您选择的序号')
    n = 1
    for i in list(dicts[c1].values())[0]:
        print(' '*15+str(n)+'.'+i[0]+' '+str(i[1])+'元')
        n+=1
    while True:
        c2 = input('请您输入您选择的序号(输入q结算输入w返回上级菜单)')
        if c2=='w':
            break
        if c2=='q':
            flag = 1
            print(' '*15+'商品名称','购买数量', '商品总价')
            total_price = 0
            for a,b in menu.items():
                total_price+=b[1]
                print(' '*15+a+' '*5+str(b[0])+' '*5+str(b[1]))
            print('商品总价为：'+str(total_price))
            break
        goods_name = list(dicts[c1].values())[0][int(c2)-1][0]
        goods_price = list(dicts[c1].values())[0][int(c2)-1][1]
        c2 = input('请您输入购买数量')
        if goods_name in menu:
            menu[goods_name][0] = menu[goods_name][0]+int(c2)
            menu[goods_name][1] = menu[goods_name][0]*goods_price
        else:
            menu[goods_name] = [int(c2),int(c2)*goods_price]