取款
这是接着昨天存款的功能写的,我发现存款和取款就是差在一个加一个减而已。
def withdrawal(self):
"""用来实现在线取款功能"""
money = input("请输入金额:")
if self._db.update_user(self._current_user,
"balance", int(money) -
self._current_user.get_balance()):
self._current_user.set_balance(int(money) -
self._current_user.get_balance())
status = "成功"
print("取成功")
else:
status = "失败"
print("取款失败")
item = Item(self._current_user, "取款", int(money), status)
if self._db.insert_log(item):
return True
接下来是查询余额、和退出
但是这两个功能实现起来太简单了,就直接代码了,也懒得测试了。
# -----------------查询余额------------------
def check_money(self):
print(self._current_user)
return True
# -----------------退出账号------------------
def quit(self):
self._current_user = None
接下来就是写run()方法来调用这些个方法了,如何让他们之间的逻辑完整。
ATM机首先应该是插卡,然后识别卡号,输入密码登录的这样的一种功能,但是我现在肯定不能直接去做一个RFID卡来,那不现实,所以,就把流程改成:
如果有卡则先登陆。没有卡先注册,所以一开始就只有两个选项,所以就是下面这样。
def run(self):
while True:
print("欢迎使用------DK--ATM 机-------------\n")
print("1. 登录\n")
print("2. 注册\n")
print("------------------------------------\n")
code = input("(1/2):")
if code == '1':
self._login()
break
elif code == '2':
self.open_an_account()
break
self.menu()
下面的menu是一个现实所有功能的方法:
def menu(self):
""""""
while True:
print(self._current_user)
print("1. 存款\n")
print("2. 取款\n")
print("3. 查询\n")
print("4. 退出\n")
print('-' * 36)
code = input("(1/2/3/4):")
if code == '1':
self.deposit()
elif code == '2':
self.withdrawal()
elif code == '3':
self.check_money()
elif code == '4':
self.quit()
break
最后使用的效果如下:
image.png image.png image.png image.png image.png
。。。emmmmm好像不太对,怎么取了200就成 - 的了呢????
奥。原来是前面的取款的时候减错了,。。
现在改过来:
def withdrawal(self):
"""用来实现在线取款功能"""
money = input("请输入金额:")
if self._db.update_user(self._current_user,
"balance", self._current_user.get_balance() -
int(money)):
self._current_user.set_balance(self._current_user.get_balance() -
int(money))
status = "成功"
print("取款成功")
else:
status = "失败"
print("取款失败")
item = Item(self._current_user, "取款", int(money), status)
if self._db.insert_log(item):
return True
现在用前面注册的来登录:
image.png image.png image.png image.png image.pngemmmm 还有点小瑕疵。就是,我退出当前账号后并不是想退出程序,而是换一个账号,现在需要把他换成退出登陆的当前的账号后,返回那一个登录和注册的界面。
def quit(self):
self._current_user = None
self.run()
return True
现在改成这样就好了,在推车当前登录的账号后,就直接回到第一个菜单。
image.png就像现在这样,嗯么么么,差不对都完成了吧!!!!
恭喜恭喜,同喜同喜。
奥对了,还有数据库的相关记录还么有看。
image.png
网友评论