需求：
1.额度1500或自定义 2.实现购物商城，买东西加入 购物车，调用信用卡接口结账 3.可以提现、存款、余额查询和转账等 4.打印指定日期间购物清单 5.支持多账户 6.ATM记录操作所有日志 7.提供管理接口，包括查询账户、添加账户、注销账户，冻结解冻账户等 8.日志功能用装饰器
写这个小项目之前先构思了一下，画个流程图，然后设置好代码的目录结构，实现方式等，把框架写好了，接下来就是一个个小功能的实现啦。
流程图如下：
代码目录结构如下：
我先写了一个用户类，专门提供用户管理。主要功能如下：
1、账户信息查询
2、添加新账户
3、删除账户
4、修改密码
5、修改手机号
6、冻结账户
7、解冻账户
8、修改账户信用额度
把一些环境变量信息写在conf目录下的settings.py里
# -*- coding: utf-8 -*- import time import os from core import users as u # 管理员账号密码 adminName = 'admin' adminPassword = 'admin' # 当前时间 timeYear = time.strftime('%Y%m%d', time.localtime(time.time())) timeDay = time.strftime('%H%M%S', time.localtime(time.time())) # 信用卡额度 creditLimit = 15000 # 用户信息文件 userInfoFile = "../db/user-info.json" # 商品信息文件 goodsInfoFile = '../db/goods-info.json' # 用户购物车文件 def ShoppingCarFile(username,year=timeYear,day=timeDay): dirs = '../db/shoppinCar/%s' % username if not os.path.exists(dirs): os.makedirs(dirs) shoppingInfoFile = '%s/shopping-info_%s%s.json' % (dirs,year,day) #print(shoppingInfoFile) return shoppingInfoFile # 日志文件 logFile = '../log/atm.log' # log功能,使用装饰器 def log(func): def wrapper(*args, **kwargs): users = u.Users().getUser() f = open(logFile, "a", encoding="utf-8") if func.__doc__ != '查询余额' and func.__doc__ != '修改密码': f.write('%s\t%s%s%s元\n' % (time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()),users[args[1]]['username'],func.__doc__,args[-1])) else: f.write('%s\t%s%s\n' % (time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()),users[args[1]]['username'],func.__doc__)) f.close() return func(*args, **kwargs) return wrapper
接下来就是开始写core目录下的users.py了，把以上的各个小功能逐一实现。
# -*- coding: utf-8 -*- import os,sys BASE_DIR = os.path.dirname(os.path.dirname(__file__)) sys.path.append(BASE_DIR) from conf import settings as ss import json class Users(object): '''用户类，管理用户接口''' def getUser(self): '''获取账户信息''' try: with open(ss.userInfoFile, "r", encoding="utf-8") as f: users = json.load(f) #print(users) except: print('have no users inof!') users = {} return users def setUser(self,users): '''重新写入账户信息''' with open(ss.userInfoFile, 'w') as f : json.dump(users , f , indent="\t") #print(users) def checkUser(self,username): '''检查用户是否存在''' users = self.getUser() for userID in users.keys(): if username == users[userID]['username']: #print('账户名已存在！') return True return False def getUserID(self,username): '''通过账户名获取账户id值''' users = self.getUser() for userID in users.keys(): if username == users[userID]['username']: return userID print('账户不存在!!!') return self.turnBack() def addUser(self,username,password,phone=None,limit=ss.creditLimit,isLock=False): '''添加账户''' users=self.getUser() # 字典key值自动加1 max_id = "000000" if users != {}: max_id = max(users.keys()) userID = str(int(max_id) + 1).zfill(6) # id自动+1，补足6位 users[userID] = { 'username':username, 'password':password, 'phone':phone, 'limit':limit, 'isLock':isLock } self.setUser(users) print(''.center(50,'#')) print('账户\033[42;1m%s\033[0m添加成功' % users[userID]['username']) print('账户名： %s' % users[userID]['username']) print('手机号： %s' % users[userID]['phone']) print('信用卡额度： %s' % users[userID]['limit']) print(''.center(50,'#')) return True def delUser(self,userID): '''账户注销''' users=self.getUser() try: print('账户\033[41;1m%s\033[0m已删除！' % users[userID]['username']) users.pop(userID) except: print('账户不存在！') return False self.setUser(users) return True def frozenUser(self,userID): '''冻结账户''' users=self.getUser() users[userID]['isLock'] = True self.setUser(users) print('账户%s已冻结！' % users[userID]['username'] ) def thawUser(self,userID): '''解冻账户''' users=self.getUser() users[userID]['isLock'] = False self.setUser(users) print('账户%s已解除冻结！' % users[userID]['username'] ) def userLoggin(self): '''用户登录验证''' for i in range (1,4): username = input('请输入用户名：') if self.checkUser(username): userID = self.getUserID(username) users=self.getUser() if users[userID]['isLock']: print('账户已被冻结，请联系管理员！') exit(-1) password = input('请输入密码：') if self.checkPassword(userID,password): return username,userID else: print('密码错误！') else: print('账户不存在！') print('操作过于频繁！') return False,False def checkPassword(self,userID,password): '''校验密码是否正确''' users = self.getUser() if password == users[userID]['password']: return True else: return False @ss.log def changePasswd(self,userID): '''修改密码''' for i in range(1,4): old_passwd = input('请输入旧密码：') # 校验旧密码是否正确 check = self.checkPassword(userID,old_passwd) if check: for j in range(1,4): if j == 3: print('操作过于频繁') return False new_passwd1 = input('请输入新密码：') if len(new_passwd1) < 8: print('密码不能少于8位！') continue if new_passwd1 == old_passwd: print('新密码不能和旧密码一样！') continue new_passwd2 = input('请再次确认新密码：') if new_passwd1 != new_passwd2: print('两次输入的密码不一致！') continue else: users=self.getUser() users[userID]['password'] = new_passwd1 self.setUser(users) print('密码修改成功！') return True else: print('旧密码输入错误！') print('操作过于频繁。') return False def changePhone(self,userID,phone): '''修改手机号''' users=self.getUser() users[userID]['phone'] = phone self.setUser(users) print('手机号修改完成！') print('账户名： %s' % users[userID]['username']) print('手机号： %s' % users[userID]['phone']) def changeLimit(self,userID,limit): '''修改账户信用额度''' users=self.getUser() users[userID]['limit'] = limit self.setUser(users) print('账户%s的信用额度已更新为%s' % (users[userID]['username'],users[userID]['limit'])) def turnBack(self): '''返回上级菜单''' print(''.center(15,'*')) print( "1、返回上级") print( "2、退出") print(''.center(15,'*')) while True: choise = input("请选择：") if choise == '1': self.userManagement() elif choise == '2': print ("谢谢使用！") exit(-1) else: print ("输入有误，请重新输入！") def userManagement(self): '''账户管理，总入口函数''' print('''请选择账户管理操作： 1、账户信息查询 2、添加新账户 3、删除账户 4、修改密码 5、修改手机号 6、冻结账户 7、解冻账户 8、修改账户信用额度''') while True: choise=input().strip() if choise.isdigit(): if '1' == choise: # 账户信息查询 users = self.getUser() print(''.center(50,'#')) for i in users.keys(): print('账户名： %s' % users[i]['username']) print('手机号： %s' % users[i]['phone']) print('信用卡额度： %s' % users[i]['limit']) print(''.center(50,'#')) self.turnBack() break elif '2' == choise: # 添加新账户 username = input('请输入用户名：') # 校验用户名是否已存在 if self.checkUser(username): print('账户已存在！') self.turnBack() break password = input('请输入密码：') phone = input('请输入手机号：') self.addUser(username,password,phone) self.turnBack() break elif '3' == choise: # 删除账户 username = input('请输入要删除的账号名：') userID = self.getUserID(username) self.delUser(userID) self.turnBack() break elif '4' == choise: # 修改密码 username = input('请输入要修改密码的账号名：') userID = self.getUserID(username) self.changePasswd(userID) self.turnBack() break elif '5' == choise: # 修改手机号 username = input('请输入要修改手机号的账号名：') userID = self.getUserID(username) phone = input('请输入新手机号：') self.changePhone(userID,phone) self.turnBack() break elif '6' == choise: # 冻结账户 username = input('请输入要冻结的账号名：') userID = self.getUserID(username) self.frozenUser(userID) self.turnBack() break elif '7' == choise: # 解冻账户 username = input('请输入要解除冻结的账号名：') userID = self.getUserID(username) self.thawUser(userID) self.turnBack() break elif '8' == choise: # 修改账户信用额度 username = input('请输入要更改信用额度的账号名：') userID = self.getUserID(username) limit = input('请输入要更改的信用额度：') if limit.isdigit(): self.changeLimit(userID,limit) else: print('输入无效,更新失败！') self.turnBack() break else: print('请输入有效数字！') else: print('请输入有效数字！') def run(): users=Users() users.userManagement() if __name__ == '__main__': run()
把用户类实现后，接下来是管理员的商品类啦，可以实现商品的上架、下架、修改商品价格已经商品查询功能等
# -*- coding: utf-8 -*- import os,sys BASE_DIR = os.path.dirname(os.path.dirname(__file__)) sys.path.append(BASE_DIR) from conf import settings as ss import json class Goods(object): '''商品类''' def getGoods(self): '''获取账户信息''' try: with open(ss.goodsInfoFile, "r", encoding="utf-8") as f: goods = json.load(f) #print(goods) except: print('have no goods inof!') goods = [] #print(goods) return goods def setGoods(self,goods): '''写入商品信息''' with open(ss.goodsInfoFile, "w", encoding="utf-8") as f: json.dump(goods, f, ensure_ascii=False, indent="\t") def printGoods(self): '''打印商品列表''' goods = self.getGoods() print(''.center(20,'-')) print('所有商品列表如下：') print(''.center(20,'-')) print('序号\t\t名称\t\t价格') print(''.center(20,'-')) for name,price in goods: print(goods.index([name,price]), '\t', name,"\t",price) print(''.center(20,'-')) def checkPrice(self,price): '''校验输入的价格是否为有效数字''' if price.isdigit(): return int(price.strip()) else: print('请输入有效数字！') return False def checkName(self,name): '''校验商品名称是否存在''' goods = self.getGoods() for goodName,price in goods: if name == goodName: return True return False def goodsShelves(self): '''商品上架''' name = input("请输入要上架的商品名称").strip() if self.checkName(name): print('商品已存在！') return False price=input("请输入该商品的价格") if self.checkPrice(price): goods = self.getGoods() goods.append([name,int(price)]) print('商品\033[32;1m%s\033[0m已上架完成！价格为：\033[32;1m%s\033[0m' % (name,int(price))) self.setGoods(goods) return True def goodDelete(self): '''商品下架''' name = input("请输入要下架的商品名称").strip() if not self.checkName(name): print('商品不存在！') return False else: goods = self.getGoods() for goodName,goodsPrice in goods: if goodName == name: goods.remove([goodName,goodsPrice]) #print(goods) print('商品\033[41;1m%s\033[0m已下架成功！' % name) self.setGoods(goods) return True def goodSearch(self,name): '''商品搜索查询''' if not self.checkName(name): print('商品不存在！') return False else: goods = self.getGoods() for goodName,goodsPrice in goods: if name == goodName: # 获取商品在list中的index值 index = goods.index([goodName,goodsPrice]) #print(index) print('商品价格为：') print(name,'\t',goods[index][1]) return (goods,index,True) def goodModify(self,name): '''修改商品价格''' goods,index,Flag = self.goodSearch(name) #print(Flag,'\n',goods,"\n",index) price=input("请要修改商品的价格") if self.checkPrice(price): goods[index][1] = int(price) self.setGoods(goods) print('商品价格修改完成！') print(goods[index][0],"\t","\033[32;1m%s\033[0m" % goods[index][1] ) return True def turnBack(self): '''返回上级菜单''' print(''.center(15,'*')) print( "1、返回上级") print( "2、退出") print(''.center(15,'*')) while True: choise = input("请选择：") if choise == '1': self.goodManagement() elif choise == '2': print ("谢谢使用！") exit(-1) else: print ("输入有误，请重新输入！") def goodManagement(self): '''用户选择''' print("请选择操作：") print("1、打印商品列表") print("2、商品上架") print("3、商品下架") print("4、修改商品价格") print("5、商品查询") while True: choice = input().strip() if choice.isdigit(): choice = int(choice) if 1 == choice: # 打印商品列表 self.printGoods() self.turnBack() break elif 2 == choice: # 商品上架 self.goodsShelves() self.turnBack() break elif 3 == choice: # 商品下架 self.goodDelete() self.turnBack() break elif 4 == choice: # 修改商品价格 name = input("请要修改商品价格的名称：").strip() self.goodModify(name) self.turnBack() break elif 5 == choice: # 商品查询 name = input("请输入要查询商品价格的名称:").strip() self.goodSearch(name) self.turnBack() break else: print("输入有误，请重新输入") else: print("请输入有效数字！") def run(): goods = Goods() goods.goodManagement() if __name__ == '__main__': run()
emmm,管理员的两大块就完成啦，接下来就是普通用户的一些操作啦，这个开发顺序也是要稍微注意一下的，先实现管理员的功能，就有了普通用户的信息嘛，然后再实现普通用户，这就可以直接用已有的账户信息啦，然后接着实现ATM功能，再实现购物功能，这样购物结算的时候就可以调用AMT结算啦。
废话不多说，看看ATM类的主要功能：
1、存款
2、取款
3、转账
4、查询余额
5、修改密码
# -*- coding: utf-8 -*- import os,sys BASE_DIR = os.path.dirname(os.path.dirname(__file__)) sys.path.append(BASE_DIR) from core import users as u from conf import settings as ss class ATM(object): '''ATM类''' @ss.log def deposit(self,userID,money): '''存款''' # 调用用户类的获取用户信息方法 users = u.Users().getUser() try: users[userID]['money'] = int(users[userID]['money']) + money except: # 第一次存钱 users[userID]['money'] = money # 调用用户类，重新写入用户信息 u.Users().setUser(users) #print('账户%s已存入%s元！' % (users[userID]['username'],money) ) #print('总余额为： %s 元' % users[userID]['money']) return True @ss.log def withdrawals(self,userID,money): '''取款''' users = u.Users().getUser() try: int(users[userID]['money']) if money > users[userID]['money']: print('余额不足！！') return False else: users[userID]['money'] -= money except: print('余额不足！') return False u.Users().setUser(users) #print('账户%s已取出%s元！' % (users[userID]['username'],money) ) #print('总余额为： %s 元' % users[userID]['money']) return True @ss.log def transfer(self,userID1,userID2,money): '''转账''' # 直接调用取款、存款函数，取款成功了再存款。 if self.withdrawals(userID1,money): self.deposit(userID2,money) return True return False @ss.log def queryBalance(self,userID): '''查询余额''' users = u.Users().getUser() try: print('账户%s的余额为：\n%s元' % (users[userID]['username'],users[userID]['money'])) except: print('账户%s的余额为：\n0元' % (users[userID]['username'])) def turnBack(self,userID): '''返回上级菜单''' print(''.center(15,'*')) print( "1、返回上级") print( "2、退出") print(''.center(15,'*')) while True: choise = input("请选择：") if choise == '1': self.atmOperation(userID) elif choise == '2': print ("谢谢使用！") exit(-1) else: print ("输入有误，请重新输入！") def inputMoney(self,userID): '''输入金额''' money = input() if money.isdigit(): return int(money) else: print('输入不是有效数字！') return self.turnBack(userID) def atmOperation(self,userID): '''ATM操作，总入口函数''' print('''请选择： 1、存款 2、取款 3、转账 4、查询余额 5、修改密码''') while True: choise=input().strip() if '1' == choise: # 存款 print('请输入存款金额：') money = self.inputMoney(userID) self.deposit(userID,money) print('存入%s元成功！' % (money)) self.turnBack(userID) break elif '2' == choise: # 取款 print('请输入取款金额！') money = self.inputMoney(userID) if self.withdrawals(userID,money): print('已取出%s元！请收好！' % (money)) self.turnBack(userID) break elif '3' == choise: # 转账 username = input('请输入要转账的账户名：') checkUser = u.Users().checkUser(username) if checkUser: print('请输入要转账金额:') money = self.inputMoney(userID) userID2 = u.Users().getUserID(username) if self.transfer(userID,userID2,money): print('已成功向%s转账%s元！' % (username, money)) else: print('账户不存在！') self.turnBack(userID) break elif '4' == choise: # 查询余额 self.queryBalance(userID) self.turnBack(userID) break elif '5' == choise: # 修改密码 u.Users().changePasswd(userID) self.turnBack(userID) break else: print('输入有误！') def run(): atm = ATM() print('欢迎登录瑞士银行！！') # 登录验证，通过后返回userID值 username,userID = u.Users().userLoggin() if userID: atm.atmOperation(userID) if __name__ == '__main__': run()
接下来就是普通用户的购物，这个比其他功能复杂点的地方就是不同的用户购物信息要分类存储。
# -*- coding: utf-8 -*- import os,sys BASE_DIR = os.path.dirname(os.path.dirname(__file__)) sys.path.append(BASE_DIR) from core import goods as g from core import atm as a from conf import settings as ss from core import users as u import json class ShoppingCar(object): '''用户购物''' def getBuyList(self,username): '''读取用户购物车清单''' try: with open(ss.ShoppingCarFile(username), "r", encoding="utf-8") as f: buyList = json.load(f) #print(buyList) except: print('have no goods inof!') buyList = [] #print(buyList) return buyList def setBuyList(self,username,buyList): '''将用户购物清单写入文件中''' with open(ss.ShoppingCarFile(username), "w", encoding="utf-8") as f: json.dump(buyList, f, ensure_ascii=False, indent="\t") def printBuyList(self,username): '''打印购物车列表''' sum = 0 buyList = self.getBuyList(username) print(''.center(20,'-')) print('购买商品列表如下：') print(''.center(20,'-')) print('名称\t\t价格') print(''.center(20,'-')) for name,price in buyList: print(name,"\t",price) sum = sum + int(price) print(''.center(20,'-')) print('总计：\t %s' % sum) return sum def buyList(self,username,choice): '''请选择购物''' buyList = self.getBuyList(username) goods = g.Goods().getGoods() if choice < len(goods) and choice >=0: print(goods[choice]) buyList.append(goods[choice]) #print(buyList) self.setBuyList(username,buyList) print('加入购物车成功') else: print('商品未上架！') return buyList def shopping(self,username,userID): '''购物''' goods = g.Goods().printGoods() print(goods) while True: print('请选择，停止购物按q') choice = input().strip() if choice != 'q': if choice.isdigit(): choice = int(choice) self.buyList(username,choice) else: print('请输入有效数字！') else: # 获取购物车总价 sum = self.printBuyList(username) print('开始结算！') n = 2 while n >=0: password = input('请输入银行卡密码：') if u.Users().checkPassword(userID,password): # 调用银行卡接口结算 if a.ATM().withdrawals(userID,sum): print('结算成功！') return True else: print('结算失败！') else: if n !=0: print('密码错误！你还有%s次尝试机会！' % n) else: print('尝试次数已用完,结算失败！') n -=1 # 清空本次购物信息 #buyList = ['结算失败！'] #self.setBuyList(username, buyList) os.remove(ss.ShoppingCarFile(username)) return False def run(): s = ShoppingCar() print('欢迎登录XXX网上购物商城！') # 登录验证，通过后返回userID值 username,userID = u.Users().userLoggin() if userID: s.shopping(username,userID) if __name__ == '__main__': run()
最后就是写一个总的入口功能啦，在bin/main.py里去实现调用以上各个功能。
# -*- coding: utf-8 -*- # 2019/7/10 16:57 import os,sys BASE_DIR = os.path.dirname(os.path.dirname(__file__)) sys.path.append(BASE_DIR) from core import goods as g from core import atm as a from conf import settings as ss from core import users as u from core import shopping as s def checkAdmin(): '''校验管理员账号密码''' i = 2 while i >= 0: username = input('请输入管理员账号：') if username == ss.adminName: j = 2 while j >= 0: password = input('请输入管理员密码：') if password == ss.adminPassword: return True else: print('管理员密码输入有误！') if j == 0: print('账号已锁定，请10分钟后重新尝试！') return False else: print('你还有%s次尝试机会！' % j) j -=1 else: print('管理员账号输入有误！') if i != 0: print('你还有%s次尝试机会！' % i) else: print('账号已锁定，请10分钟后重新尝试！') i -=1 return False def averageUser(): '''普通用户操作''' print('''请选择操作： 1、购买商品 2、ATM操作''') while True: choice = input().strip() if '1' == choice: s.run() break elif '2' == choice: a.run() break else: print('输入有误！') def adminManagement(): '''管理员操作''' print('''请选择操作： 1、商品管理 2、用户管理''') while True: choice = input().strip() if '1' == choice: g.run() break elif '2' == choice: u.run() break else: print('输入有误！') def run(): '''程序入口函数''' print('''欢迎进入ATM+购物程序! 请选择操作： 1、普通用户 2、管理员''') while True: choice = input().strip() if '1' == choice: averageUser() break elif '2' == choice: if checkAdmin(): adminManagement() break else: print('输入有误！') if __name__ == '__main__': run()
再给ATM的操作加上日志功能，这里用了装饰器，直接应用到ATM操作的函数上，而不改变原来函数的调用方式。就写了一个log函数，放在settings.py里啦，就不再单独写一个py文件。
源码链接：https://github.com/caiqinxiong/ATM-Shopping.git