Python ATM功能实现代码实例


Posted in Python onMarch 19, 2020

编写ATM程序实现下述功能,数据来源于文件db.txt

1、充值功能:用户输入充值钱数,db.txt中该账号钱数完成修改

2、转账功能:用户A向用户B转账1000元,db.txt中完成用户A账号减钱,用户B账号加钱

3、提现功能:用户输入提现金额,db.txt中该账号钱数减少

4、查询余额功能:输入账号查询余额

登录功能

用户登录成功后,内存中记录下该状态,上述功能以当前登录状态为准,必须先登录才能操作

代码如下

import os

user_staus = {'username': None}


def ad_credit(username, amount):
  """
  充值功能
  :param username:
  :param amount:
  :return:
  """
  if user_staus['username'] is None:
    login()
  with open('db.txt', 'rt', encoding='utf-8')as f1, \
      open('db.txt.swap', 'wt', encoding='utf-8')as f2:
    while True:
      cont = f1.readline()
      if len(cont) == 0:
        break
      name, remain = cont.strip().split(':')
      if username in name:
        remain = int(remain) + int(amount)
        f2.write('{}:{}\n'.format(name, remain))
        print('充值成功,{}的余额为{}'.format(username, remain))
      else:
        f2.write(cont)
  os.remove('db.txt')
  os.rename('db.txt.swap', 'db.txt')


def transfer(user_out, user_in, amount):
  '''3
  转账功能
  :param user_out:
  :param user_in:
  :param amount:
  :return:
  '''
  if user_staus['username'] is None:
    login()
  with open('db.txt', 'rt', encoding='utf-8')as f1, \
      open('db.txt.swap', 'wt', encoding='utf-8')as f2:

    userinfo = {}
    for line in f1:
      name, remind = line.strip().split(':')
      userinfo[name] = int(remind)
    if user_out not in userinfo:
      print('用户不存在')
      return
    if user_in not in userinfo:
      print('收款方不存在')
      return
    if user_out in userinfo and user_in in userinfo:
      if userinfo[user_out] >= int(amount):
        userinfo[user_out] -= int(amount)
        userinfo[user_in] += int(amount)
        print('转账成功,已成功从{}向{}汇款{}'.format(user_out, user_in, amount))
      elif userinfo[user_out] < amount:
        print('余额不足')
        return
    for name, remind in userinfo.items():
      f2.write('{}:{}\n'.format(name, remind))

  os.remove('db.txt')
  os.rename('db.txt.swap', 'db.txt')


def cashon(username, amount):
  '''
  提现功能
  :param username:
  :param amount:
  :return:
  '''
  if user_staus['username'] is None:
    login()
  with open('db.txt', 'rt', encoding='utf-8')as f1, \
      open('db.txt.swap', 'wt', encoding='utf-8')as f2:
    userinfo = {}
    for line in f1:
      name, remind = line.strip().split(':')
      userinfo[name] = int(remind)
    if username not in userinfo:
      print('用户不存在')
      return
    if username in userinfo and userinfo[username] >= int(amount):
      userinfo[username] -= int(amount)
      print('已从余额中取出{},现余额为{}'.format(amount, userinfo[username]))
    elif userinfo[username] < amount:
      print('余额不足,提现失败')
      return
    for name, remind in userinfo.items():
      f2.write('{}:{}\n'.format(name, remind))
  os.remove('db.txt')
  os.rename('db.txt.swap', 'db.txt')


def check(username):
  '''
  余额查询功能
  :param username:
  :return:
  '''
  if user_staus['username'] is None:
    login()
  with open('db.txt', 'rt', encoding='utf-8')as f:
    userinfo = {}
    for line in f:
      name, remind = line.strip().split(':')
      userinfo[name] = remind
    if username not in userinfo:
      print('用户不存在')
      return
    if username in userinfo:
      print('当前余额为:{}'.format(userinfo[username]))


def login():
  username = input('输入用户名')
  userpassword = input('输入密码')
  with open('login.txt', 'rt', encoding='utf-8')as login_f:
    login = {}
    for line in login_f:
      name, psd = line.strip().split(':')
      login[name] = psd
      if username in login:
        if login[username] == userpassword:
          print('登陆成功')
          user_staus['username'] = username
          break
      elif username not in login:
        print('用户名不存在')
        return


def logout():
  user_staus['username'] = None
  print('已成功登出')
  return


login()
tag = True
while tag:
  cmd = input('''
  请输入你想使用的功能序号
  1:充值
  2:转账
  3:提现
  4:查询余额
  0:登出
  ''')
  if cmd == '1':
    username = input('输入用户名:')
    amount = input('输入充值金额:')
    ad_credit(username, amount)
  elif cmd == '2':
    user_out = input('请输入转账方:')
    user_in = input('请输入接收方:')
    amount = input('输入转账金额:')
    transfer(user_out, user_in, amount)
  elif cmd == '3':
    username = input('输入用户名:')
    amount = input('输入提现金额')
    cashon(username, amount)
  elif cmd == '4':
    username = input('输入用户名:')
    check(username)
  elif cmd == '0':
    logout()
    tag = False
  else:
    print('请正确输入序号')

以下是模拟结果

'''
/Users/chenfeng/PycharmProjects/ATM/venv/bin/python /Users/chenfeng/PycharmProjects/ATM/main.py
输入用户名xilou
输入密码666
登陆成功

请输入你想使用的功能序号
1:充值
2:转账
3:提现
4:查询余额
0:登出
1
输入用户名:xilou
输入充值金额:200
充值成功,xilou的余额为700

请输入你想使用的功能序号
1:充值
2:转账
3:提现
4:查询余额
0:登出
2
请输入转账方:xilou
请输入接收方:heiren
输入转账金额:200
转账成功,已成功从xilou向heiren汇款200

请输入你想使用的功能序号
1:充值
2:转账
3:提现
4:查询余额
0:登出
3
输入用户名:xilou
输入提现金额100
已从余额中取出100,现余额为400

请输入你想使用的功能序号
1:充值
2:转账
3:提现
4:查询余额
0:登出
4
输入用户名:xilou
当前余额为:400

请输入你想使用的功能序号
1:充值
2:转账
3:提现
4:查询余额
0:登出
0
已成功登出

Process finished with exit code 0

'''以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python下如何查询CS反恐精英的服务器信息
Jan 17 Python
python/sympy求解矩阵方程的方法
Nov 08 Python
python中import与from方法总结(推荐)
Mar 21 Python
如何安装并使用conda指令管理python环境
Jul 10 Python
Django框架基础模板标签与filter使用方法详解
Jul 23 Python
如何安装并在pycharm使用selenium的方法
Apr 30 Python
sklearn的predict_proba使用说明
Jun 28 Python
使用PyCharm官方中文语言包汉化PyCharm
Nov 18 Python
python - asyncio异步编程
Apr 06 Python
Opencv实现二维直方图的计算及绘制
Jul 21 Python
python的变量和简单数字类型详解
Sep 15 Python
python3中apply函数和lambda函数的使用详解
Feb 28 Python
详解django使用include无法跳转的解决方法
Mar 19 #Python
Django模板标签中url使用详解(url跳转到指定页面)
Mar 19 #Python
关于Python turtle库使用时坐标的确定方法
Mar 19 #Python
Windows下Pycharm远程连接虚拟机中Centos下的Python环境(图文教程详解)
Mar 19 #Python
Pycharm内置终端及远程SSH工具的使用教程图文详解
Mar 19 #Python
python开发实例之Python的Twisted框架中Deferred对象的详细用法与实例
Mar 19 #Python
mac 上配置Pycharm连接远程服务器并实现使用远程服务器Python解释器的方法
Mar 19 #Python
You might like
在IIS上安装PHP4.0正式版
2006/10/09 PHP
php+ajax实现的点击浏览量加1
2015/04/16 PHP
深入理解PHP之源码目录结构与功能说明
2016/06/01 PHP
javascript 贪吃蛇实现代码
2008/11/22 Javascript
IE6下CSS图片缓存问题解决方法
2010/12/09 Javascript
js复制到剪切板的实例方法
2013/06/28 Javascript
JQuery结合CSS操作打印样式的方法
2013/12/24 Javascript
js判断字符长度以及中英文数字等
2013/12/31 Javascript
jQuery插件Tooltipster实现漂亮的工具提示
2015/04/12 Javascript
浅谈下拉菜单中的Option对象
2015/05/10 Javascript
Bootstrap进度条组件知识详解
2016/05/01 Javascript
利用jquery实现下拉框的禁用与启用
2016/12/07 Javascript
ng-repeat指令在迭代对象时的去重方法
2018/10/02 Javascript
浅析js中mvvm模式实现的原理
2018/10/06 Javascript
了解Javascript中函数作为对象的魅力
2019/06/19 Javascript
js实现时间日期校验
2020/05/26 Javascript
JS简易计算器实例讲解
2020/06/30 Javascript
js实现直播点击飘心效果
2020/08/19 Javascript
react实现复选框全选和反选组件效果
2020/08/25 Javascript
vue穿梭框实现上下移动
2021/01/29 Vue.js
python中的一些类型转换函数小结
2013/02/10 Python
浅谈python对象数据的读写权限
2016/09/12 Python
浅谈pandas中shift和diff函数关系
2018/04/08 Python
Python网络编程使用select实现socket全双工异步通信功能示例
2018/04/09 Python
对Python 3.2 迭代器的next函数实例讲解
2018/10/18 Python
python并发编程多进程之守护进程原理解析
2019/08/20 Python
Django实现celery定时任务过程解析
2020/04/21 Python
GWT的应用有哪两种部署模式
2012/12/21 面试题
田径运动会开幕式及主持词
2014/03/28 职场文书
投资协议书范本
2014/04/21 职场文书
幼儿园安全责任书范本
2014/07/24 职场文书
学习党代会心得体会
2014/09/05 职场文书
2016年高校自主招生自荐信范文
2015/03/24 职场文书
某学校的2019年度工作报告范本
2019/10/11 职场文书
python迷宫问题深度优先遍历实例
2021/06/20 Python
JavaScript圣杯布局与双飞翼布局实现案例详解
2022/08/05 Javascript