#test1.py 
import random #用來隨機產(chǎn)生卡號
import pickle #序列化，用來存放和取出產(chǎn)生的用戶數(shù)據(jù)
import os #產(chǎn)生文件
import re #正則表達(dá)式，用來判斷身份證和手機號，其他地方也可以使用
class Card:
 def __init__(self,cardId,password,money=0):
 self.cardId=cardId
 self.password=password
 self.money=money
class User(Card):
 def __init__(self,username,phone,uid,card):
 self.username=username
 self.phone=phone
 self.uid=uid
 self.user_card=card #User繼承Card對象
class Bank(User): #Bank 繼承User，Bank是序列化的對象，所以將其變?yōu)樽值?
 def __init__(self,user):
 self.cardId=user.user_card.cardId 
 self.user=user 
 self.users={self.cardId:self.user} #鍵，卡號 ：值，User對象 

def generate_cardid(): #方法一 產(chǎn)生卡號
 list=[]
 for i in range(0,11):
 n=random.randint(0,9)
 n=str(n)
 list.append(n)
 # str="".join(list)
 return list
# generate_cardid()
# user_cardId=input("請輸入您的賬號：")
def create_user(): #方法2:開戶
 while True:
 uid=input("請輸入身份證:")
 realut=re.fullmatch("\d{17}[\d,X]",uid) #正則判斷身份證是否合理
 if realut:
 uid=realut.group()
 break
 else:
 print("格式不合法")
 continue
 username=input("請輸入姓名:")
 while True:
 phone=input("請輸入手機號碼:")
 realut = re.fullmatch("1\d{10}", phone) #正則判斷手機號是否合理，其他需要判斷的地方都可以判斷，我就不再使用了
 if realut:
 phone=realut.group()
 # print(phone)
 break
 else:
 print("格式不合法")
 continue
 list=generate_cardid() #得到卡號列表
 cardId="".join(list) #將卡號變成字符串，字符串的卡號才能做成鍵
 print(f"您的卡號為：{cardId}")
 while True:
 password1=input("請輸入密碼:")
 password2=input("再次輸入密碼確認(rèn):")
 if password1==password2:
 password=password1
 break
 else :
 print("兩次密碼不同，請重新輸入！")
 continue
 card=Card(cardId,password) 
 user=User(uid,username,phone,card)
 bank=Bank(user) #產(chǎn)生bank對象
 with open(f"data\\{cardId}.txt","ab") as file_w: #重點：創(chuàng)建一個文件夾data來存放產(chǎn)生的bank對象，每個對象根據(jù)卡號產(chǎn)生一個txt文件，用來存放用戶的所有數(shù)據(jù)
 pickle.dump(bank,file_w) #將bank 序列化保存到文檔中
# create_user()

def user_login(user_cardId): #登錄

 if os.path.exists(f"data\\{user_cardId}.txt"):
 with open(f"data\\{user_cardId}.txt", "rb") as file_r:
 u_data = pickle.load(file_r) #根據(jù)卡號取出txt文檔，反序列化取出數(shù)據(jù)
 if u_data.cardId == user_cardId: #u_data是一個字典，鍵是卡號，值是user對象 
 n = 1
 while True:
 if n <= 3:
 user_pw = input("請輸入密碼:")
 if u_data.user.user_card.password == user_pw:
 return True
 else:
 print("密碼錯誤！")
 n+=1
 continue
 else:
 print("三次輸入密碼錯誤！")
 return
 else:
 print("沒有該用戶")

# user_login(user_cardId)
def save_money(user_cardId): # 方法4:存錢
 if user_login(user_cardId): #如果登錄成功
 money=int(input("請您輸入存錢金額："))
 with open(f"data\\{user_cardId}.txt", "rb") as file_r:
 u_data = pickle.load(file_r)
 u_data.user.user_card.money=u_data.user.user_card.money+money
 print("您的余額為:",u_data.user.user_card.money)
 with open(f"data\\{user_cardId}.txt", "wb") as file_w: #這里要用wb,而不是ab,改變數(shù)據(jù)后，需要覆蓋原來的數(shù)據(jù)，而不是添加
 pickle.dump(u_data, file_w)
# save_money()

def withdraw_money(user_cardId): # 方法5:取錢
 if user_login(user_cardId):
 money=int(input("請您輸入取款金額："))
 with open(f"data\\{user_cardId}.txt", "rb") as file_r:
 u_data = pickle.load(file_r)
 if money>u_data.user.user_card.money:
 print("余額不足")
 else:
 u_data.user.user_card.money=u_data.user.user_card.money-money
 print("您的余額為:", u_data.user.user_card.money)
 with open(f"data\\{user_cardId}.txt", "wb") as file_w:
 pickle.dump(u_data, file_w)

# withdraw_money()

def transfer_accounts(user_cardId): #方法6，轉(zhuǎn)賬
 if user_login(user_cardId):
 with open(f"data\\{user_cardId}.txt", "rb") as file_r:
 u_data = pickle.load(file_r)
 while True:
 money = int(input("請您轉(zhuǎn)賬取款金額："))
 if money > u_data.user.user_card.money:
 print("余額不足")
 break
 else:
 cardId=int(input("請您對方卡號："))
 if os.path.exists(f"data\\{cardId}.txt"): #如果對方卡號存在
 u_data.user.user_card.money = u_data.user.user_card.money - money #自己的money減
 print("您的余額為:", u_data.user.user_card.money)
 with open(f"data\\{user_cardId}.txt", "wb") as file_w:
 pickle.dump(u_data, file_w)

 with open(f"data\\{cardId}.txt", "rb") as file_r1: #根據(jù)對方的卡號進行查找對方的數(shù)據(jù)
 u_data1 = pickle.load(file_r1)
 with open(f"data\\{cardId}.txt", "wb") as file_w1:
 u_data1.user.user_card.money = u_data1.user.user_card.money + money #對方money加
 pickle.dump(u_data1, file_w1)
 print("轉(zhuǎn)賬成功")
 break
 else:
 print("該用戶不存在")
 break

# transfer_accounts()

def select_user(user_cardId): # 方法7:查詢余額
 if user_login(user_cardId):
 with open(f"data\\{user_cardId}.txt", "rb") as file_r:
 u_data = pickle.load(file_r)
 print("您的余額為：",u_data.user.user_card.money)
# select_user()

def update_password(user_cardId): # 方法8:修改密碼
 if user_login(user_cardId):
 while True:
 pw1=input("請輸入新密碼:")
 pw2=input("請再次輸入密碼:")
 if pw1==pw2:
 with open(f"data\\{user_cardId}.txt", "rb") as file_r:
 u_data = pickle.load(file_r)
 u_data.user.user_card.password=pw1
 with open(f"data\\{user_cardId}.txt", "wb") as file_w:
 pickle.dump(u_data, file_w)
 break

 else:
 print("兩次密碼不相同")
 continue

import test1
import os
T =True
while True:
 user_cardId=input("請輸入您的賬號,退出請按0,注冊請按1:\n")
 if os.path.exists(f"data\\{user_cardId}.txt"):
 true=True
 while true:
 x = input("[2]查詢余額 [3]取款 [4]存錢 [5]轉(zhuǎn)賬 [6]修改密碼 [7]退出系統(tǒng)\n"
 "請選擇需要的操作輸入對應(yīng)的數(shù)字："
 if x=="2": #鍵盤輸入默認(rèn)是str類型
 test1.select_user(user_cardId)
 continue
 elif x=="3":
 test1.withdraw_money(user_cardId)
 continue

 elif x=="4":
 test1.save_money(user_cardId)
 continue
 elif x=="5":
 test1.transfer_accounts(user_cardId)
 continue
 elif x=="6":
 test1.update_password(user_cardId)
 continue
 elif x=="7":
 true=False
 else:
 print("沒有對應(yīng)操作")
 continue
 elif user_cardId=="0":
 print("歡迎再次使用")
 break
 elif user_cardId=="1":
 test1.create_user()
 continue

 else :
 print("沒有該賬戶")
 T=True
 while T:
 num = input("需要注冊賬號請按1，退出注冊服務(wù)請按0：")
 if num =="1":
 test1.create_user()
 elif num =="0":
 print("歡迎再次使用")
 T =False
 break
 else:
 print("沒有對應(yīng)操作")
 continue
 break