Python实现atm机的功能
# cat atm.py#!/usr/bin/env python
# -*- coding: utf-8 -*-
'''
Date:2017-03-23
Author:Bob
'''
import os
import time
import pickle
import readline #解决退格键和上下键引起的乱码,需要安装readline和readline-devel包
#定义账单,商品和购物车
Bill = {}
products = {}
shoplist = []
#define Bill function, used to record billing details(account/time/describe/money).
def Bill(Account,Time,Description,RMB):
Bill = {"Account":Account,"Time":Time,"Description":Description,"RMB":RMB}
#用pickle模块把账单信息存入到bill文件中去
pickle.dump(Bill,open("bill","a"))
#购物功能
def shop():
print '\033[;32mWelcome to shopping!\n\033[0m'
with open('shops.txt') as f:
for line in f.readlines():
print '{}'.format(line.strip())
while 1:
with open('shops.txt') as f:
for line in f.readlines():
line = line.strip()
commodity = line.split()
price = line.split()
products = price
choice = raw_input("\n\033[;36mPlease enter what you want to buy,if you want back,you can enter\033[0m \033[;31mback\033[0m:").strip()
if len(choice) == 0:
continue
elif choice == 'back':
list()
#如果有这个商品,就判断商品价格,如果商品价格大于余额,就提示余额不足
if products.has_key(choice):
#从userinfo文件中读取并反序列化
remaining = pickle.load(open('userinfo','rb'))
if int(products) > remaining:
print 'In your card remaining sum already insufficiency, please prompt sufficient value!'
else:
while 1:
#把购买的商品追加到购物车
shoplist.append(choice)
#计算余额,余额就是总金额减去购买的商品价格
new_remaining = int(remaining) - int(products)
userInfo = int(new_remaining)
#把余额信息序列化并存到userinfo文件中
pickle.dump(userInfo,open("userinfo","wb"))
#把购买的记录和账单写到Bill文件中
Bill(accountAuth,time.strftime("%Y-%m-%d %H:%M:%S"),choice,"-%d" % int(products))
#打印消费的金额和剩余金额
print "\033[;32mConsumption is %r Money is %r\033,new_remaining)
#打印购物车的商品
print "\033[;33mThe shopping list %s \033[0m" % shoplist
break
else:
print 'You choose {} is not in the shoplist!'.format(choice)
shop()
#查询余额功能
def query_money():
userInfo = pickle.load(open('userinfo','rb'))
totalmoney = userInfo
remaining = userInfo
print 'Your total money is {}, remaining money is \033[1;31m{}\033[0m!'.format(totalmoney, remaining)
#存钱功能
def save_money():
while 1:
save_desc = raw_input("Please describe save money the details:").strip()
if len(save_desc) == 0:
continue
try:
save_money = int(raw_input("Please save the money:"))
except ValueError:
print "\033[;31mYou entered must be number.\033[0m"
save_money()
if save_money % 100 != 0:
print 'You must enter an integer of 100!'
continue
userInfo = pickle.load(open('userinfo', 'rb'))
remaining = int(userInfo) + save_money
userInfo = remaining
pickle.dump(userInfo, open('userinfo', 'wb'))
print 'Your total money is %s, your remaining is \033[;31m%s\033, userInfo)
Bill(accountAuth,time.strftime("%Y-%m-%d %H:%M:%S"),save_desc,"+%d" % float(save_money))
next = raw_input("1.continue \n2.return \n3.exit \nPlease select: ").strip()
if next == '1':
continue
elif next == '2':
list()
elif next == '3':
exit()
else:
print 'Please enter the correct content!'
#取钱功能
def draw_money():
while 1:
draw_desc = raw_input("Please describe draw money the details:").strip()
if len(draw_desc) == 0:
continue
try:
draw_money = int(raw_input("Please draw the money:"))
except ValueError:
print "\033[;31mYou entered must be number.\033[0m"
draw_money()
if draw_money % 100 != 0:
print 'You must enter an integer of 100!'
continue
userInfo = pickle.load(open('userinfo', 'rb'))
#There are bugs here!
if draw_money > int(userInfo):
print '\033[;31mYour remaining is insufficient!\033[0m'
list()
userInfo = pickle.load(open('userinfo', 'rb'))
remaining = int(userInfo) - draw_money - draw_money * 0.05
userInfo = remaining
pickle.dump(userInfo, open('userinfo', 'wb'))
print 'Your total money is %s, your remaining is \033[;31m%s\033, userInfo)
Bill(accountAuth,time.strftime("%Y-%m-%d %H:%M:%S"),draw_desc,"+%d" % float(draw_money))
next = raw_input("1.continue \n2.return \n3.exit \nPlease select: ").strip()
if next == '1':
continue
elif next == '2':
list()
elif next == '3':
exit()
else:
print 'Please enter the correct content!'
#转账功能,和上面的逻辑基本一样
def transfer_money():
while 1:
userInfo = pickle.load(open('userinfo', 'rb'))
transfer_desc = raw_input("Please describe transfer money: ").strip()
if len(transfer_desc) == 0:
continue
d_account = raw_input("Please input transfer account: ").strip()
if len(d_account) == 0:
continue
if userInfo.has_key(d_account) is False:
print "\033[;31mThis account does not exist\033[0m"
transfer_money()
d_money = int(raw_input("Please input transfer amount money: "))
if d_money % 100 != 0:
print "\033[;31mDeposit amount must be 100 integer times\033[0m"
continue
if d_money > int(userInfo):
print "\033[;31mYour balance is insufficient\033[0m"
continue
userInfo = int(userInfo) - d_money - d_money * 0.10
userInfo = int(userInfo) + d_money
pickle.dump(userInfo,open('userinfo', 'wb'))
print "\033[;32mYour credit is %r,Your balance is %r\033,userInfo)
Bill(accountAuth,time.strftime("%Y-%m-%d %H:%M:%S"),transfer_desc,"-%d" % (userInfo - d_money - d_money * 0.10))
next = raw_input("1.continue \n2.return \n3.exit \nPlease select: ").strip()
if next == '1':
continue
elif next == '2':
list()
elif next == '3':
exit()
else:
print 'Please enter the correct content!'
#账单功能
def query_bill():
Income = []
Spending = []
num = 0
print "Account\t\tTime\t\tDescription\t\tRMB"
with open('bill', 'rb') as f:
while True:
try:
line = pickle.load(f)
if line["Account"] == accountAuth:
if '+' in line["RMB"]:
print "\033[;33m%r\t%r\t%r\t\t\t%r\033,line["Time"],line["Description"],line["RMB"])
income = line["RMB"].strip("+")
Income.append(income)
else:
print "%r\t%r\t%r\t\t\t%r" % (line["Account"],line["Time"],line["Description"],line["RMB"])
spending = line["RMB"].strip("-")
Spending.append(spending)
except:
break
for i in Income:
num = num + int(i)
income = num
print "Income is %r" % num
for i in Spending:
num = num + int(i)
spending = num
print "Spending is %r" % num
print "Total is %r" % (int(income) + int(spending))
#修改密码功能
def modify_passwd():
userInfo = pickle.load(open('userinfo', 'rb'))
old_passwd = raw_input("Please enter old password:").strip()
while 1:
if old_passwd == userInfo:
new_passwd = raw_input("Please enter new password:").strip()
if len(new_passwd) < 6:
print 'Your password is too simple!'
continue
confirm_new_password = raw_input("Please confirm new password again:").strip()
if new_passwd != confirm_new_password:
print 'Two passwords do not match!'
else:
userInfo = confirm_new_password
pickle.dump(userInfo, open('userinfo', 'wb'))
print '\033[;32mYour password is changed successful!\033[0m'
exit()
else:
print 'Your password is error!'
modify_passwd()
#ATM机所有功能
def list():
print '''\033[;32m
###################################################
# welcome to ATM! #
# #
# 1.shop 2.query money #
# 3.save money 4.draw money #
# 5.transfer money 6.query bill #
# 7.modify password 8.exit #
# #
###################################################
\033[0m'''
while 1:
choice = raw_input("Please choose according to your needs:").strip()
if len(choice) == 0:
continue
elif choice == '1':
shop()
elif choice == '2':
query_money()
elif choice == '3':
save_money()
elif choice == '4':
draw_money()
elif choice == '5':
transfer_money()
elif choice == '6':
query_bill()
elif choice == '7':
modify_passwd()
else:
print "\n\033[;35mYou have been exit the system!\033[0m"
exit()
#用户登录功能
userInfo = pickle.load(open('userinfo', 'rb'))
while 1:
accountAuth = raw_input("Please input user account:").strip()
if len(accountAuth) == 0:
continue
if userInfo.has_key(accountAuth):
if 'lock' in userInfo:
print '%s has been locked!' % accountAuth
exit()
for num in range(3,0,-1):
passwdAuth = raw_input("Please input user password:").strip()
if len(passwdAuth) == 0:
continue
if passwdAuth == userInfo:
list()
else:
print "Wrong password, Can try again \033[;31m%r\033[0m itmes" % num
continue
else:
lockaccount = userInfo
lockaccount.append('lock')
pickle.dump(userInfo,open('userinfo', 'wb'))
print "\033[;31mAccount freeze within 24 hours\033[0m"
exit()
else:
print "\033[;31mWrong account %r,retype\033[0m" % accountAuth
页:
[1]