示例#1
0
def main():
    # 管理员对象
    admin = Admin()
    # 管理员开机
    admin.printAdminView()
    if admin.adminOption():
        print("登录失败!请稍后再试……")
        return -1
    # 创建提款机对象
    atm = ATM()
    fileDir = os.path.join(os.getcwd(), "data")
    if os.path.exists(fileDir):
        filePath = os.path.join(fileDir, "data.txt")
        with open(filePath, "rb") as f:
            atm.allUsers = pickle.load(f)

    while True:
        # 存储所有用户的信息
        admin.printSysFunctionView()
        # 等待用户操作
        option = input("请输入您的操作:")
        if option == "1":
            #开户
            atm.creatUser()
        elif option == "2":
            atm.searchUserInfo()
        elif option == "3":

            print("取款")
        elif option == "4":

            print("存款")
        elif option == "5":

            print("转账")
        elif option == "6":

            print("改密")
        elif option == "7":

            atm.lockUser()
        elif option == "8":

            atm.unlockUser()
        elif option == "9":

            print("补卡")
        elif option == "0":

            print("销户")
        elif option == "t":
            if not admin.adminOption():  #841968
                # 将当前系统中的用户保存到文件中
                fileDir = os.path.join(os.getcwd(), "data")
                if not os.path.exists(fileDir):
                    os.mkdir("data")
                file = os.path.join(fileDir, "data.txt")
                allUsersDict = atm.allUsers
                f = open(file, "wb")
                pickle.dump(allUsersDict, f)
                f.close()
                print("退出成功!")
                return -1
        time.sleep(2)