python day2 作业

#!/usr/bin/env python
# -*- coding:utf-8 -*-


import os

Dir = os.getcwd()
Status = True
Dicts = {}
shopping_list = []



if os.path.getsize(Dir+"\shopping_cart.txt"):
    with open(Dir+"\shopping_cart.txt","w+") as sp:
        sp.truncate()

with open(Dir+"\menu.txt","r+") as f:
    Menu = f.readlines()
    for n in Menu:
        k,v = n.strip().split(":")
        v = v.split()
        v[1] = int(v[1])
        Dicts[int(k)] = v

if not os.path.getsize(Dir+"\\balance.txt"):
    Balance = input("please input balance:")
    with open(Dir + "\\balance.txt", "w+") as f1:
        f1.write(Balance)

else:
    with open(Dir+"\\balance.txt","r+") as b:
        Balance = b.read()
Balance = int(Balance)
#print(Balance,type(Balance))



while Status:
    print("Product List".center(40, "#"))
    for k in Dicts:
        print(k,Dicts[k])
    choice = input("please input you choice,enter [q] quit:")
    if choice.isdigit():
        choice = int(choice)
        if 0 < choice <= len(Dicts):
            Price = Dicts[choice]
            #with open(Dir+"\\balance.txt","r+") as f5:
                #Balance = int(f5.read())
            if Price[1] <= Balance:
                with open(Dir+"\shopping_cart.txt","a+") as f2:
                    f2.write(str(Price))
                #with open(Dir+"\\balance.txt","w+") as f3:
                    #Balance = int(f3.read())
                Balance -= Price[1]
                    #f3.write(str(Balance))
                print("shopping is %s,balance is \033[31;1m %s \033[0m"%(Price,Balance))

            else:
                print("Insufficient Balance!!!")
        else:
            print("No such item")
    elif choice == 'q' or choice == 'Q':
        with open(Dir + "\shopping_cart.txt", "r+") as s:
            s_list = s.readlines()
        print("shopping is %s,balance is \033[31;1m %s \033[0m" % (s_list, Balance))
        with open(Dir + "\\balance.txt","w+") as f4:
            f4.write(str(Balance))
        Status = False

    else:
        print("please input integer")