python atm小脚本

grerww

用python写的一个自助信用卡的小脚本,有'取现,查询记录,转账,还款,购物'功能,用到了picked序列化,prettytable等功能

下面是测试和代码

root@ubuntu:~# python atm.py plz enter your card id: 0000plz enter your password:                                     欢迎登录信用卡系统!            可选操作:                1. 取现: 不能超限额,手续费5%           4. 转账: 可转账到不同用户                2. 查询: 查询余额和交易明细            5. 购物                3. 还款: 现金还款                      6. 退出        选择操作: 1请输入金额: 5000cash:5000,FEE:250.0                                    欢迎登录信用卡系统!            可选操作:                1. 取现: 不能超限额,手续费5%           4. 转账: 可转账到不同用户                2. 查询: 查询余额和交易明细            5. 购物                3. 还款: 现金还款                      6. 退出        选择操作: 3请输入金额: 10000repay 10000                                    欢迎登录信用卡系统!            可选操作:                1. 取现: 不能超限额,手续费5%           4. 转账: 可转账到不同用户                2. 查询: 查询余额和交易明细            5. 购物                3. 还款: 现金还款                      6. 退出        选择操作: 4请输入对方账号: 0001请输入金额: 2000transfer to 0001, 2000RMB                                    欢迎登录信用卡系统!            可选操作:                1. 取现: 不能超限额,手续费5%           4. 转账: 可转账到不同用户                2. 查询: 查询余额和交易明细            5. 购物                3. 还款: 现金还款                      6. 退出        选择操作: 5        商品列表:            1. iphone       标价: 5000RMB            2. MacBook      标价: 12000RMB            3. iMac         标价: 20000RMB            4. car          标价: 50000RMB            5. 退出购物车        输入商品标签: 2Spend: 12000RMB        商品列表:            1. iphone       标价: 5000RMB            2. MacBook      标价: 12000RMB            3. iMac         标价: 20000RMB            4. car          标价: 50000RMB            5. 退出购物车        输入商品标签: 5欢迎下次光临!                                    欢迎登录信用卡系统!            可选操作:                1. 取现: 不能超限额,手续费5%           4. 转账: 可转账到不同用户                2. 查询: 查询余额和交易明细            5. 购物                3. 还款: 现金还款                      6. 退出        选择操作: 2+---------------------+----------+----------------------------+---------+|       datetime      |  action  |            log             | balance |+---------------------+----------+----------------------------+---------+| 2015-02-04 15:22:03 |   cash   |     cash:500,FEE:25.0      | 14475.0 || 2015-02-04 15:22:55 |  repay   |        repay 50000         | 64475.0 || 2015-02-04 15:23:37 | tramsfer | transfer to 0001, 20000RMB | 44475.0 || 2015-02-04 15:23:48 | shopping |       Spend: 5000RMB       | 39475.0 || 2015-02-04 16:09:45 |   cash   |    cash:5000,FEE:250.0     | 34225.0 || 2015-02-04 16:10:05 |  repay   |        repay 10000         | 44225.0 || 2015-02-04 16:10:19 | tramsfer | transfer to 0001, 2000RMB  | 42225.0 || 2015-02-04 16:10:29 | shopping |      Spend: 12000RMB       | 30225.0 |+---------------------+----------+----------------------------+---------+                                    欢迎登录信用卡系统!            可选操作:                1. 取现: 不能超限额,手续费5%           4. 转账: 可转账到不同用户                2. 查询: 查询余额和交易明细            5. 购物                3. 还款: 现金还款                      6. 退出        选择操作: 6Welcome to again!# coding: utf-8import pickleimport timefrom prettytable import PrettyTabledef userInfo():    # 读取文件,生成用户字典    userDict = {}    with open('user_info.txt', 'r') as f:        for line in f.xreadlines():            Id, pwd, status = line.split('|')            userDict[Id] = [pwd, status]    return userDictdef login(card_id):    # 用户登录认证    import getpass    if card_id in userDict and userDict[card_id][1] != '1':        for i in xrange(3):            p = getpass.getpass('plz enter your password: ')            if userDict[card_id][0] == p:                return 1            else:                print 'password error.'        else:            print 'id locked!'            userDict[card_id][1] = '1'    else:        print 'no card id or id locked.'def operat(card_id):    # 功能菜单    tag = True    while tag:        print u'''                                    欢迎登录信用卡系统!            可选操作:                1. 取现: 不能超限额,手续费5%           4. 转账: 可转账到不同用户                2. 查询: 查询余额和交易明细            5. 购物                3. 还款: 现金还款                      6. 退出        '''        choose = raw_input('选择操作: ')        if choose == '1':            cash(card_id)        elif choose == '2':            showLog(card_id)        elif choose == '3':            repay(card_id)        elif choose == '4':            transfer(card_id)        elif choose == '5':            shopping(card_id)        elif choose == '6':            tag = False            print 'Welcome to again!'        else:            print u'请输入正确的选项'def getBalance(card_id):   # 获取用余额    try:        id_info = pickle.load(open('idInfo', 'rb'))    except Exception:        id_info = {card_id: [float(15000)]}        pickle.dump(id_info, open('idInfo', 'wb'))        return id_info[card_id][0]    else:        if card_id in id_info:            return id_info[card_id][0]        else:            id_info[card_id] = [float(15000)]            pickle.dump(id_info, open('idInfo', 'wb'))            return id_info[card_id][0]def insertBal(card_id, num):    # 插入用户余额    id_info = pickle.load(open('idInfo', 'rb'))    id_info[card_id] = [float(num)]    pickle.dump(id_info, open('idInfo', 'wb'))def repay(card_id):     # 还款    num = raw_input('请输入金额: ')    if len(num) != 0 and num.isdigit() and float(num) > 0:        re_num = getBalance(card_id) + float(num)        insertBal(card_id, float(re_num))        log = 'repay %s' % num        print log        nowtime = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time()))        genLog(card_id, 'repay', nowtime, getBalance(card_id), log)    else:        print u'请输入正确的金额'def cash(card_id):      # 取现    bal = getBalance(card_id)    money = raw_input('请输入金额: ')    if money.isdigit() and len(money) != 0 and float(money) + float(money) * 0.05 < bal:        num = bal - (float(money) + float(money) * 0.05)        insertBal(card_id, num)        log = 'cash:%s,FEE:%s' % (money, float(money) * 0.05)        print log        nowtime = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time()))        genLog(card_id, 'cash', nowtime, str(num), log)    else:        print u'余额不足或输入错误!'def genLog(card_id, action, date, balance, description):    # 生成日志文件,用了picke    msg = [date, action, description, str(balance)]    try:        log_info = pickle.load(open('logInfo', 'rb'))    except Exception:        log_info = {card_id: [msg]}        pickle.dump(log_info, open('logInfo', 'wb'))    else:        if card_id in log_info:            log_info[card_id].append(msg)            pickle.dump(log_info, open('logInfo', 'wb'))        else:            log_info[card_id] = [msg]            pickle.dump(log_info, open('logInfo', 'wb'))def showLog(card_id):       # 输入日志    try:        log_info = pickle.load(open('logInfo', 'rb'))    except:        print u'之前没有过任何操作'    else:        x = PrettyTable(['datetime', 'action', 'log', 'balance'])        for i in log_info[card_id]:            x.add_row(i)        print xdef transfer(card_id):      # 转账功能    des_id = raw_input('请输入对方账号: ')    if len(des_id) == 4 and des_id.isdigit() and des_id in userDict and des_id != card_id:        money = raw_input('请输入金额: ')        sor_bal = getBalance(card_id)        if len(money) != 0 and money.isdigit() and 0 < float(money) < sor_bal:            nowtime = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time()))            des_bal = getBalance(des_id)            des_bal += float(money)            log = '%s transfer to you %s RMB.' % (card_id, money)            genLog(des_id, 'transfer', nowtime, des_bal, log)            insertBal(des_id, des_bal)            sor_bal -= float(money)            insertBal(card_id, sor_bal)            log = 'transfer to %s, %sRMB' % (des_id, money)            genLog(card_id, 'tramsfer', nowtime, sor_bal, log)            print log        else:            print u'输入错误或余额不足'    else:        print u'输入错误或账号不存在'def shopping(card_id):          # 购物功能    tag = True    while tag:        print u'''        商品列表:            1. iphone       标价: 5000RMB            2. MacBook      标价: 12000RMB            3. iMac         标价: 20000RMB            4. car          标价: 50000RMB            5. 退出购物车        '''        choose = raw_input('输入商品标签: ')        bal = getBalance(card_id)        nowtime = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time()))        if choose == '1':            if bal > 5000:                log = 'Spend: 5000RMB'                print log                bal -= 5000                insertBal(card_id, float(bal))                genLog(card_id, 'shopping', nowtime, getBalance(card_id), log)            else:                print u'余额不足'        elif choose == '2':            if bal > 12000:                log = 'Spend: 12000RMB'                print log                bal -= 12000                insertBal(card_id, float(bal))                genLog(card_id, 'shopping', nowtime, getBalance(card_id), log)            else:                print u'余额不足'        elif choose == '3':            if bal > 20000:                log = 'Spend: 20000RMB'                print log                bal -= 20000                insertBal(card_id, float(bal))                genLog(card_id, 'shopping', nowtime, getBalance(card_id), log)            else:                print u'余额不足'        elif choose == '4':            if bal > 50000:                log = 'Spend: 50000RMB'                print log                bal -= 50000                insertBal(card_id, float(bal))                genLog(card_id, 'shopping', nowtime, getBalance(card_id), log)            else:                print u'余额不足'        elif choose == '5':            tag = False            print u'欢迎下次光临!'        else:            print u'请输入正确的选项'if __name__ == '__main__':    userDict = userInfo()    cardId = raw_input('plz enter your card id: ').strip()    if login(cardId):        operat(cardId)