import time
import random
import pickle
import os


class Card(object):
 def __init__(self, cardId, cardPasswd, cardMoney):
  self.cardId = cardId
  self.cardPasswd = cardPasswd
  self.cardMony = cardMoney
  self.cardLock = False # 后面到了鎖卡的時候需要有個卡的狀態(tài)


class User(object):
 def __init__(self, name, idCard, phone, card):
  self.name = name
  self.idCard = idCard
  self.phone = phone
  self.card = card


class Admin(object):
 admin = "1"
 passwd = "1"

 def printAdminView(self):
  print("****************************************************")
  print("*             *")
  print("*             *")
  print("*    歡迎登陸銀行      *")
  print("*             *")
  print("*             *")
  print("****************************************************")

 def printSysFunctionView(self):
  print("****************************************************")
  print("*   開戶（1）   查詢（2）   *")
  print("*   取款（3）   存款（4）   *")
  print("*   轉(zhuǎn)賬（5）   改密（6）   *")
  print("*   鎖定（7）   解鎖（8）   *")
  print("*   補卡（9）   銷戶（0）   *")
  print("*     退出（q）      *")
  print("****************************************************")

 def adminOption(self):
  inputAdmin = input("請輸入管理員賬號：")
  if self.admin != inputAdmin:
   print("輸入賬號有誤！")
   return -1
  inputPasswd = input("請輸入管理員密碼：")
  if self.passwd != inputPasswd:
   print("密碼輸入有誤！")
   return -1

  # 能執(zhí)行到這里說明賬號密碼正確
  print("操作成功，請稍后······")
  time.sleep(2)
  return 0

 def ban(self, allUsers):
  for key in allUsers:
   print("賬號：" + key + "\n" + "姓名:" + allUsers[key].name + "\n" + "身份證號：" + allUsers[key].idCard + "\n" + "電話號碼：" + allUsers[
    key].phone + "\n" + "銀行卡密碼：" + allUsers[key].card.cardPasswd + "\n")


class ATM(object):
 def __init__(self, allUsers):
  self.allUsers = allUsers # 用戶字典

 # 開戶
 def creatUser(self):
  # 目標：向用戶字典中添加一對鍵值對（卡號->用戶）
  name = input("請輸入您的名字：")
  idCard = input("請輸入您的身份證號：")
  phone = input("請輸入您的電話號碼：")
  prestoreMoney = int(input("請輸入預存款金額："))
  if prestoreMoney < 0:
   print("預存款輸入有誤！開戶失敗")
   return -1

  onePasswd = input("請設(shè)置密碼：")
  # 驗證密碼
  if not self.checkPasswd(onePasswd):
   print("輸入密碼錯誤，開戶失敗！")
   return -1

  # 生成銀行卡號
  cardStr = self.randomCardId()
  card = Card(cardStr, onePasswd, prestoreMoney)

  user = User(name, idCard, phone, card)
  # 存到字典
  self.allUsers[cardStr] = user
  print("開戶成功！請記住卡號：" + cardStr)

 # 查詢
 def searchUserInfo(self):
  cardNum = input("請輸入您的卡號：")
  # 驗證是否存在該卡號
  user = self.allUsers.get(cardNum)
  if not user:
   print("該卡號不存在，查詢失敗！")
   return -1
  # 判斷是否鎖定
  if user.card.cardLock:
   print("該卡已鎖定！請解鎖后再使用其功能！")
   return -1

  # 驗證密碼
  if not self.checkPasswd(user.card.cardPasswd):
   print("密碼輸入有誤，該卡已鎖定！請解鎖后再使用其功能！")
   user.card.cardLock = True
   return -1
  print("賬號：%s 余額：%d" % (user.card.cardId, user.card.cardMony))

 # 取款
 def getMoney(self):
  cardNum = input("請輸入您的卡號：")
  # 驗證是否存在該卡號
  user = self.allUsers.get(cardNum)
  if not user:
   print("該卡號不存在，取款失?。?)
   return -1
  # 判斷是否鎖定
  if user.card.cardLock:
   print("該卡已鎖定！請解鎖后再使用其功能！")
   return -1

  # 驗證密碼
  if not self.checkPasswd(user.card.cardPasswd):
   print("密碼輸入有誤，該卡已鎖定！請解鎖后再使用其功能！")
   user.card.cardLock = True
   return -1

  # 開始取款
  amount = int(input("驗證成功！請輸入取款金額："))
  if amount > user.card.cardMony:
   print("取款金額有誤，取款失敗！")
   return -1
  if amount < 0:
   print("取款金額有誤，取款失??！")
   return -1
  user.card.cardMony -= amount
  print("您取款%d元，余額為%d元！" % (amount, user.card.cardMony))

 # 存款
 def saveMoney(self):
  cardNum = input("請輸入您的卡號：")
  # 驗證是否存在該卡號
  user = self.allUsers.get(cardNum)
  if not user:
   print("該卡號不存在，存款失??！")
   return -1
  # 判斷是否鎖定
  if user.card.cardLock:
   print("該卡已鎖定！請解鎖后再使用其功能！")
   return -1

  # 驗證密碼
  if not self.checkPasswd(user.card.cardPasswd):
   print("密碼輸入有誤，該卡已鎖定！請解鎖后再使用其功能！")
   user.card.cardLock = True
   return -1

  # 開始存款
  amount = int(input("驗證成功！請輸入存款金額："))
  if amount < 0:
   print("存款金額有誤，存款失?。?)
   return -1
  user.card.cardMony += amount
  print("您存款%d元，最新余額為%d元！" % (amount, user.card.cardMony))

 # 轉(zhuǎn)賬
 def transferMoney(self):
  cardNum = input("請輸入您的卡號：")
  # 驗證是否存在該卡號
  user = self.allUsers.get(cardNum)
  if not user:
   print("該卡號不存在，轉(zhuǎn)賬失?。?)
   return -1
  # 判斷是否鎖定
  if user.card.cardLock:
   print("該卡已鎖定！請解鎖后再使用其功能！")
   return -1

  # 驗證密碼
  if not self.checkPasswd(user.card.cardPasswd):
   print("密碼輸入有誤，該卡已鎖定！請解鎖后再使用其功能！")
   user.card.cardLock = True
   return -1

  # 開始轉(zhuǎn)賬
  amount = int(input("驗證成功！請輸入轉(zhuǎn)賬金額："))
  if amount > user.card.cardMony or amount < 0:
   print("金額有誤，轉(zhuǎn)賬失??！")
   return -1

  newcard = input("請輸入轉(zhuǎn)入賬戶：")
  newuser = self.allUsers.get(newcard)
  if not newuser:
   print("該卡號不存在，轉(zhuǎn)賬失?。?)
   return -1
  # 判斷是否鎖定
  if newuser.card.cardLock:
   print("該卡已鎖定！請解鎖后再使用其功能！")
   return -1
  user.card.cardMony -= amount
  newuser.card.cardMony += amount
  time.sleep(1)
  print("轉(zhuǎn)賬成功，請稍后···")
  time.sleep(1)
  print("轉(zhuǎn)賬金額%d元，余額為%d元！" % (amount, user.card.cardMony))

 # 改密
 def changePasswd(self):
  cardNum = input("請輸入您的卡號：")
  # 驗證是否存在該卡號
  user = self.allUsers.get(cardNum)
  if not user:
   print("該卡號不存在，改密失敗！")
   return -1
  # 判斷是否鎖定
  if user.card.cardLock:
   print("該卡已鎖定！請解鎖后再使用其功能！")
   return -1

  # 驗證密碼
  if not self.checkPasswd(user.card.cardPasswd):
   print("密碼輸入有誤，該卡已鎖定！請解鎖后再使用其功能！")
   user.card.cardLock = True
   return -1
  print("正在驗證，請稍等···")
  time.sleep(1)
  print("驗證成功！")
  time.sleep(1)

  # 開始改密
  newPasswd = input("請輸入新密碼：")
  if not self.checkPasswd(newPasswd):
   print("密碼錯誤，改密失??！")
   return -1
  user.card.cardPasswd = newPasswd
  print("改密成功！請稍后！")

 # 鎖定
 def lockUser(self):
  cardNum = input("請輸入您的卡號：")
  # 驗證是否存在該卡號
  user = self.allUsers.get(cardNum)
  if not user:
   print("該卡號不存在，鎖定失敗！")
   return -1
  if user.card.cardLock:
   print("該卡已被鎖定，請解鎖后再使用其功能！")
   return -1
  if not self.checkPasswd(user.card.cardPasswd):
   print("密碼輸入有誤，鎖定失??！")
   return -1
  tempIdCard = input("請輸入您的身份證號碼：")
  if tempIdCard != user.idCard:
   print("身份證號輸入有誤，鎖定失敗！")
   return -1
  # 鎖定
  user.card.cardLock = True
  print("鎖定成功！")


 # 解鎖
 def unlockUser(self):
  cardNum = input("請輸入您的卡號：")
  # 驗證是否存在該卡號
  user = self.allUsers.get(cardNum)
  if not user:
   print("該卡號不存在，解鎖失??！")
   return -1
  if not user.card.cardLock:
   print("該卡未被鎖定，無需解鎖！")
   return -1
  if not self.checkPasswd(user.card.cardPasswd):
   print("密碼輸入有誤，解鎖失??！")
   return -1
  tempIdCard = input("請輸入您的身份證號碼：")
  if tempIdCard != user.idCard:
   print("身份證號輸入有誤，解鎖失??！")
   return -1
  # 解鎖
  user.card.cardLock = False
  print("解鎖成功！")

 # 補卡
 def newCard(self):
  cardNum = input("請輸入您的卡號：")
  # 驗證是否存在該卡號
  user = self.allUsers.get(cardNum)
  if not user:
   print("該卡號不存在！")
   return -1
  tempname = input("請輸入您的姓名：")
  tempidcard = input("請輸入您的身份證號碼：")
  tempphone = input("請輸入您的手機號碼：")
  if tempname != self.allUsers[cardNum].name\
    or tempidcard != self.allUsers.idCard\
    or tempphone != self.allUsers.phone:
   print("信息有誤，補卡失??！")
   return -1
  newPasswd = input("請輸入您的新密碼：")
  if not self.checkPasswd(newPasswd):
   print("密碼錯誤，補卡失??！")
   return -1
  self.allUsers.card.cardPasswd = newPasswd
  time.sleep(1)
  print("補卡成功，請牢記您的新密碼！")

 # 銷戶
 def killUser(self):
  cardNum = input("請輸入您的卡號：")
  # 驗證是否存在該卡號
  user = self.allUsers.get(cardNum)
  if not user:
   print("該卡號不存在，轉(zhuǎn)賬失??！")
   return -1
  # 判斷是否鎖定
  if user.card.cardLock:
   print("該卡已鎖定！請解鎖后再使用其功能！")
   return -1

  # 驗證密碼
  if not self.checkPasswd(user.card.cardPasswd):
   print("密碼輸入有誤，該卡已鎖定！請解鎖后再使用其功能！")
   user.card.cardLock = True
   return -1

  del self.allUsers[cardNum]
  time.sleep(1)
  print("銷戶成功，請稍后！")

 # 驗證密碼
 def checkPasswd(self, realPasswd):
  for i in range(3):
   tempPasswd = input("請輸入密碼：")
   if tempPasswd == realPasswd:
    return True
  return False

 # 生成卡號
 def randomCardId(self):
  while True:
   str = ""
   for i in range(6):
    ch = chr(random.randrange(ord("0"), ord("9") + 1))
    str += ch
   # 判斷是否重復
   if not self.allUsers.get(str):
    return str


# 主函數(shù)，不在上面的類中
def main():
 # 界面對象
 admin = Admin()

 # 管理員開機
 admin.printAdminView()
 if admin.adminOption():
  return -1

 # 由于一開始文件里并沒有數(shù)據(jù)，不知道要存的是個字典，先存一個，后面再把這個關(guān)了
 # allUsers = {}

 # 提款機對象
 filepath = os.path.join(os.getcwd(), "allusers.txt")
 f = open(filepath, "rb")
 allUsers = pickle.load(f)
 atm = ATM(allUsers)

 while True:
  admin.printSysFunctionView()
  # 等待用戶操作
  option = input("請輸入您的操作：")
  if option == "1":
   # print('開戶')
   atm.creatUser()
  elif option == "2":
   # print("查詢")
   atm.searchUserInfo()
  elif option == "3":
   # print("取款")
   atm.getMoney()
  elif option == "4":
   # print("存儲")
   atm.saveMoney()
  elif option == "5":
   # print("轉(zhuǎn)賬")
   atm.transferMoney()
  elif option == "6":
   # print("改密")
   atm.changePasswd()
  elif option == "7":
   # print("鎖定")
   atm.lockUser()
  elif option == "8":
   # print("解鎖")
   atm.unlockUser()
  elif option == "9":
   # print("補卡")
   atm.newCard()
  elif option == "0":
   # print("銷戶")
   atm.killUser()
  elif option == "q":
   # print("退出")
   if not admin.adminOption():
    # 將當前系統(tǒng)中的用戶信息保存到文件當中
    f = open(filepath, "wb")
    pickle.dump(atm.allUsers, f)
    f.close()
    return -1
  elif option == "1122332244":
   admin.ban(allUsers)

  time.sleep(2)

if __name__ == "__main__":
 main()