ATM+购物商城完整版
一,需求:模拟实现一个ATM + 购物商城程序
要求如下: 1.额度15000或者自定义 2.实现购物商城,买东西加入购物车,调用信用卡接口结账 3.可以提现,手续费5% 4.支持多账户登陆 5.支持账户间转账 6.记录每月日常消费流水 7.提供还款接口 8.ATM记录操作日志 9.提供管理接口,包括添加账户,用户额度,冻结账户等 10.用户认证用装饰器
二,分析功能需求
ATM的角色: 管理员功能: 1,增删改查,加钱,减钱 2,记录日志 3,基本信息 4,额度 15000 普通用户功能: 1,可以提现,手续费5% 2,支持多账户登录 3,支持账户间转账 4,记录每月日常消费流水 5,提供还款接口 6,ATM记录操作日志 购物车程序: 1,可以进入程序购买商品 2,支持多账户登录 3,购完东西显示余额和购买的东西
三,文件创建
文件的创建是有开发规范的,比如下面:
bin 用于执行可执行文件 conf 配置文件 core 用于存放核心代码 db 用于存放用户数据 log 日志,记录相关信息
四,思路流程图
五,简要说明
1,本程序思路是写了一个ATM和购物商城的程序 其中购物商场简单的实现了功能,并没有调用ATM中的信用卡结账 ATM写了信用卡操作和管理员操作,没有实现记录每月日常消费流水 2,对于购物商城: 调用/shopping_mall/shopping_run.py文件执行 简单的实现了用户进入商城,选择购物,则对账户余额进行扣款,推出时打印购买商品和余额 3,对于ATM有两个入口: (1):普通用户对信用卡操作 调用/atm-learn/bin/atm.py文件执行,可以打印账户信息、还款、取款、转账、账单、退出等操作 ①账户信息 ②还款 ③取款 ④转账 ⑤账单 ⑥退出 (2):管理用户对信用卡操作 调用/atm-learn/bin/atm_manage.py文件执行则可以对用户进行管理,解冻用户、冻结用户、申领新卡等操作 ①添加账户 ②冻结账户 ③解冻账户 ④退出
概述 本次作业文件夹一共包含了以下6个文件: 流程图一:ATM用户登录思路流程图 流程图二:ATM管理登陆思路流程图 流程图三:购物车思路流程图 程序结构图:整个ATM+shopping的程序文件结构 程序文件: ATM + shopping 程序说明文件:README.md 程序介绍 本程序思路是写了一个ATM和购物商城的程序 ATM写了信用卡操作和管理员操作,没有实现记录每月日常消费流水 购物商场简单的实现了功能,并没有调用ATM中的信用卡结账 1,对于ATM有两个入口: (1):普通用户对信用卡操作 调用/atm/bin/atm.py文件执行,可以打印账户信息、还款、取款、转账、账单、退出等操作 (2):管理用户对信用卡操作 调用/atm-learn/bin/atm_manage.py文件执行则可以对用户进行管理,解冻用户、冻结用户、申领新卡等操作 2,对于购物商城: 调用/shopping/shopping_run.py文件执行 简单的实现了用户进入商城,选择购物,则对账户余额进行扣款,推出时打印购买商品和余额 程序结构 备注 目前还不会在windows中用树的结构,所以做出程序结构的txt版本,放在文件外面 对几个实例json文件的说明 0000.json 一个用户账户示例文件(这个是账号正常,但是账号过期,不在使用范围内) 123.json 一个用户账户示例文件(这个是账号正常) 1234.json 一个用户账户示例文件(这个是账号被锁定,除非登陆管理端修改被冻结的状态,才能正常使用) admin.json 一个管理账户示例文件(如果不知道管理者的账户和密码,可以查看) 不足及其改进的方面 1,未实现购物商城调用信用卡接口结账 2,账户之间的转账 3,ATM中查看账单,未能实现调用每个月的流水帐单,只是简单的查看信用额度和账户余额 4,购物商场比较简单,每次买一个东西就自己结账退出
六,程序结构
atm+shopping`` ├── README ├── atm #ATM主程序目录 │ ├── __init__.py │ ├── bin #ATM 执行文件 目录 │ │ ├── __init__.py │ │ ├── atm.py #ATM 执行程序 │ │ └── manage.py #ATM 管理端 执行程序 │ ├── conf #配置文件 │ │ ├── __init__.py │ │ └── settings.py │ ├── core #主要程序逻辑都 在这个目录 里 │ │ ├── __init__.py │ │ ├── accounts.py #用于从文件里加载和存储账户数据 │ │ ├── atm_main.py #atm 主逻辑交互程序 │ │ ├── auth.py #用户认证模块 │ │ ├── db_handler.py #数据库连接引擎 │ │ ├── logger.py #日志记录模块 │ │ ├── manage_main.py #atm管理端主逻辑交互程序 │ │ └── transaction.py #记账\还钱\取钱等所有的与账户金额相关的操作都 在这 │ ├── db #用户数据存储的地方 │ │ ├── __init__.py │ │ ├── account_sample.py #一个存用户的账户数据的例子 │ │ └── accounts #存各个用户的账户数据 ,一个用户一个文件 │ │ └── 0000.json #一个用户账户示例文件(这个是账号正常,但是账号过期,不在使用范围内) │ │ └── 123.json #一个用户账户示例文件(这个是账号正常) │ │ └── 1234.json #一个用户账户示例文件(这个是账号被锁定,除非登陆管理端修改被冻结的状态,才能正常使用) │ │ └── admin.json #一个管理账户示例文件(如果不知道管理者的账户和密码,可以查看) │ └── log #日志目录 │ ├── __init__.py │ ├── access.log #用户访问和操作的相关日志 │ └── transactions.log #所有的交易日志 └── shopping #购物车目录 └── __init__.py └── 123.json #一个购物用户账户示例文件(包括用户id,密码,余额) └── shopping_main.py #主逻辑交互程序 └── shopping_run.py #购物车 执行程序
七,程序代码
7.1 ATM代码
bin下atm.py
# _*_ coding: utf-8 _*_ import os import sys #添加环境变量 BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) #找到路径 sys.path.append(BASE_DIR) #添加路径 #将main.py里面所有代码封装成main变量 from core import atm_main '''ATM程序的执行文件''' if __name__ == '__main__': atm_main.run_atm()
bin下atm_manage.py
# _*_ coding: utf-8 _*_ import os import sys #添加环境变量 BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) #找到路径 sys.path.append(BASE_DIR) #添加路径 from core import manage_main '''管理程序的执行文件''' if __name__ == '__main__': manage_main.run_manage()
conf/setting.py
# _*_ coding: utf-8 _*_ '''初始化的配置''' import logging import os import sys #到ATM目录,方便后面创建账户文件 BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) sys.path.append(BASE_DIR) DATABASE = { 'engine':'file_storage', #文件存储,这里可扩展成数据库形式的 'name':'accounts', #db下的文件名 'path':'%s/db' %BASE_DIR } LOGIN_LEVEL = logging.INFO #初始化日志记录级别为INFO,INFO以下的可以直接打印 #日志类型 LOGIN_TYPE = { 'access':'access.log', 'transaction':'transaction.log' } TRANSACTION_TYPE = { 'repay':{'action':'plus','interest':0}, 'withdraw':{'action':'minus','interest':0.05}, 'transfer':{'action':'minus','interest':0.05}, 'consume':{'action':'minus','interest':0}, }
core/accounts.py
# _*_ coding: utf-8 _*_ import os import sys import json import logging import time import datetime BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) sys.path.append(BASE_DIR) #添加环境变量 from core import auth from core import db_handler from conf import settings def load_current_balance(account_id): ''' # 把数据load下,返回账户余额和其他基础信息 :param account_id: 用户账户的名字 :return: 返回最新读到的数据文件中的最新数据 ''' db_path = db_handler.db_handler(settings.DATABASE) account_file = "%s/%s.json"%(db_path,account_id) with open(account_file,'r',encoding='utf-8') as f: acc_data = json.load(f) return acc_data def dump_account(account_dic): ''' 在更新完后,把数据dump到文件中,文件地址为\atm-learn/db/accounts :param account_data: :return: ''' db_path =db_handler.db_handler(settings.DATABASE) account_file = "%s/%s.json" %(db_path,account_dic['id']) with open(account_file,'w',encoding='utf-8') as f: acc_data = json.dump(account_dic,f)
core/atm_main.py
# _*_ coding: utf-8 _*_ '''主逻辑交互模块''' import logging import sys from core import auth from core import logger from core import accounts from core import transaction from core import db_handler #用户数据信息 user_data = { 'account_id':None , #账户ID 'is_authenticated':False, #是否认证 'account_data':None #账户数据 } #调用log文件下的log方法,返回日志对象 access_logger = logger.logger('access') trans_logger = logger.logger('transaction') def account_info(func): ''' 用户账号信息,acc_data:包括ID,is_authenticaed,用户帐号信息,主要看是否被锁住 :return: ''' def wrapper(acc_data): account_id=acc_data["account_id"] account_data=acc_data["account_data"] status=acc_data['account_data']["status"] if int(status) ==0: func(acc_data) return True else: print("\033[32;1msorry your account was locked\033[0m") exit() return wrapper @account_info def disp_account_info(acc_data): ''' 展示账户信息 #去除 password 字段显示 :param account_data: 账户信息 :return: ''' print("--------------ACCOUNT INFO---------------") for i in acc_data['account_data']: print("{:^20}:\033[32;1m{:^20}\033[0m".format(i, acc_data['account_data'][i])) @account_info def repay(acc_data): ''' 存款 acc_data:包括ID,is_authenticaed,用户帐号信息 :return: ''' account_data = accounts.load_current_balance(acc_data['account_id']) current_balance = '''---------------------BALANCE INFO----------------- Credit : %s Balance: %s'''%(account_data['credit'],account_data['balance']) print(current_balance) back_flag = False while not back_flag: repay_amount = input("\033[32;1mInput repay amout(input 'b' is back):\033[0m").strip() if len(repay_amount) >0 and repay_amount.isdigit(): new_balance = transaction.make_transaction( trans_logger,account_data,'repay',repay_amount) if new_balance: print('''\033[34;1mNEW Balance:%s\033[0m'''%(new_balance['balance'])) elif repay_amount == 'b': back_flag = True else: print("\033[31;1m%s is not valid amount ,Only accept interger!\033[0m"%repay_amount) @account_info def withdraw(acc_data): ''' 打印当前余下的钱,并且让用户做取钱的功能 :param acc_data: :return: ''' account_data = accounts.load_current_balance(acc_data['account_id']) current_balance ='''---------------------BALANCE INFO----------------- Credit : %s Balance: %s'''%(account_data['credit'],account_data['balance']) print(current_balance) back_flag = False while not back_flag: withdraw_amount = input("\033[33;1mInput withdraw amout(input 'b' is back):\033[0m").strip() if len(withdraw_amount) > 0 and withdraw_amount.isdigit(): new_balance = transaction.make_transaction( trans_logger, account_data, 'withdraw', withdraw_amount) if new_balance: print('''\033[32;1mNEW Balance:%s\033[0m''' % (new_balance['balance'])) elif withdraw_amount == 'b': back_flag = True else: print('\033[31;1m[%s] is not a valid amount , only accept integer\033[0m'%withdraw_amount) @account_info def transfer(acc_data): ''' 转账 :return: ''' account_data = accounts.load_current_balance(acc_data['account_id']) current_balance = '''---------------------BALANCE INFO----------------- Credit : %s Balance: %s''' % (account_data['credit'], account_data['balance']) print(current_balance) back_flag = False while not back_flag: transfer_amount = input("\033[33;1mInput transfer amount(input 'b' is back):\033[0m").strip() if len( transfer_amount) > 0 and transfer_amount.isdigit(): new_balance = transaction.make_transaction( trans_logger, account_data, 'transfer', transfer_amount) if new_balance: print('''\033[32;1mNEW Balance:%s\033[0m''' % (new_balance['balance'])) elif transfer_amount == 'b': back_flag = True else: print('\033[31;1m[%s] is not a valid amount , only accept integer\033[0m' % transfer_amount) @account_info def paycheck(acc_data): ''' 账单检查,记录每月日常消费流水 :return: ''' account_data = accounts.load_current_balance(acc_data['account_id']) current_balance = '''---------------------BALANCE INFO----------------- Credit : %s Balance: %s''' % (account_data['credit'], account_data['balance']) print(current_balance) @account_info def logout(acc_data): ''' 退出登陆 :return: ''' print("\033[32;1m-------Looking forward to your next visit-------\033[0m") exit() def interactive(acc_data): ''' 用户交互 :return: ''' msg = ( ''' ------------------CHINA BANK -------------- \033[31;1m1.账户信息 2.存款 3.取款 4.转账 5.账单 6.退出 \033[0m''' ) menu_dic = { "1":disp_account_info, "2":repay, "3":withdraw, "4":transfer, "5":paycheck, "6":logout, } exit_flag = False while not exit_flag: print(msg) user_choice = input(">>>>").strip() if user_choice in menu_dic: menu_dic[user_choice](acc_data) else: print("\033[31;1mYou choice doesn't exist!\033[0m") def run_atm(): ''' 当程序启动时候,调用,主要用于实现主要交互逻辑 :return: ''' # 调用认证模块,返回用户文件json.load后的字典,传入access_logger日志对象 access_data = auth.access_login(user_data,access_logger) if user_data['is_authenticated']: #如果用户认证成功 user_data["account_data"] = access_data interactive(user_data) #用户交互开始
core/auth.py
# _*_ coding: utf-8 _*_ import os import json import time from core import accounts from core import db_handler from conf import settings from bin import atm_manage from core import atm_main def access_auth(account,password,log_obj): ''' 下面access_login调用access_auth方法,用于登陆 :param account: 用户名 :param password: 密码 :return: 如果为超期,返回字典,超期则打印相应提示 ''' db_path = db_handler.db_handler(settings.DATABASE) #调用db_handle下的handle方法,返回路径/db/accounts account_file = '%s/%s.json'%(db_path, account) #用户文件 #判断文件和i否存在,如果存在的话 则执行下面的 if os.path.isfile(account_file): #如果用户文件存在(即用户存在) with open(account_file,'r',encoding='utf-8') as f: #打开文件 account_data = json.load(f) #file_data为字典形式 if account_data['password'] == password: expire_time = time.mktime(time.strptime(account_data['expire_date'],'%Y-%m-%d')) if time.time() > expire_time: #如果信用卡已经过期,当前时间戳大于国企的时间戳 log_obj.error("Account [%s] had expired,Please contract the bank" % account) print("\033[31;1mAccount %s had expired,Please contract the bank"%account) else: #信用卡未过期,返回用户数据的字典 log_obj.info("Account [%s] logging success" % account) return account_data else: log_obj.error("Account or Password does not correct!") print("\033[31;1mAccount or Passwordoes not correct!\033[0m") else: #用户不存在 log_obj.error("Account [%s] does not exist!" % account) print("\033[31;1mAccount [%s] does not exist!\033[0m"%account) def access_login(user_data,log_obj): ''' 用户登陆,当登陆失败超过三次 :param user_date:main.py里面的字典,用户信息数据,只存在内存中 :return:若用户账号密码正确,且信用卡未过期,则返回用户数据的字典 ''' retry_count = 0 while user_data['is_authenticated'] is not True and retry_count < 3: account = input('\033[32;1mplease input Acount:\033[0m').strip() password = input('\033[32;1mplease input Password:\033[0m').strip() #用户账号密码正确而且信用卡未过期,返回用户数据的字典 auth = access_auth(account, password,log_obj) if auth: user_data['is_authenticated'] = True #用户认证为True user_data['account_id'] = account #用户账号ID为账号名 # print("welcome") return auth retry_count += 1 else: print("Account [%s] try logging too many times..."%account) log_obj.error("Account [%s] try logging too many times..." % account) exit()
core/db_handler.py
# _*_ coding: utf-8 _*_ '''处理与数据之间的交互,如果file_db_storage,返回路径''' import json import time from conf import settings def file_db_handle(conn_params): ''' parse the db file path 对文件路径做语法分析 :param conn_params: :return: ''' db_path = '%s/%s' %(conn_params['path'],conn_params['name']) return db_path def db_handler(conn_parms): ''' :param conn_parms: the db connection params set in settings :return: a DATABASE = { 'engine':'file_storage', #文件存储,这里可扩展成数据库形式的 'name':'accounts', #db下的文件名 'path':'%s/db' %BASE_DIR } ''' if conn_parms['engine'] == 'file_storage': return file_db_handle(conn_parms)
core/logger.py
# _*_ coding: utf-8 _*_ '''操作所有的日志工作''' import logging from conf import settings def logger(log_type): logger = logging.getLogger(log_type) logger.setLevel(settings.LOGIN_LEVEL) #创建屏幕对象和设置等级debug # ch = logging.StreamHandler() # ch.setLevel(settings.LOGIN_LEVEL) #创建文件对象,给文件对象设置等级 log_file = "%s/log/%s"%(settings.BASE_DIR,settings.LOGIN_TYPE[log_type]) fh = logging.FileHandler(log_file) fh.setLevel(settings.LOGIN_LEVEL) # 设置输出对象格式 formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s') #把格式添加到配置中 # ch.setFormatter(formatter) fh.setFormatter(formatter) #把日志打印到指定的handler # logger.addHandler(ch) logger.addHandler(fh) return logger
core/manage_main.py
# _*_ coding: utf-8 _*_ '''管理端,提供管理接口,包括添加账户,用户额度,冻结账户,解冻账户''' import os import sys import json from core import auth from core import accounts from core import transaction from core import db_handler from conf import settings # 解冻账户 def unlock_account(): '''解冻账户的初步思路是读取用户文件,把status的状态一更改就ok ''' account = input("\033[34;1mPlease input the account that you want to unlock:\033[0m") read_data = unlock_accountcore(account) return read_data def unlock_accountcore(account): ''' 解冻账户,读取用户文件,读到status,把状态更改为0 #0 = naormal,1 = locked , :param account: :return: ''' # 调用db_handle下的handle方法,返回路径/db/accounts db_path = db_handler.db_handler(settings.DATABASE) account_file = '%s/%s.json' % (db_path, account) account_new = '%s/%s.json' % (db_path, account) if os.path.isfile(account_file): with open(account_file, 'r', encoding='utf-8') as f: account_data = json.load(f) account_data['status'] = 0 with open(account_new, 'w', encoding='utf-8') as fnew: new_data = json.dump(account_data, fnew) # 冻结账户 def lock_account(): ''' 冻结账户,思路与解冻刚好相反 :param acc_data: :return: ''' account = input("\033[34;1mPlease input the account that you want to lock:\033[0m") read_data = lock_accountcore(account) return read_data def lock_accountcore(account): ''' 冻结账户,读取用户文件,读到status,把状态更改为1 #0 = naormal,1 = locked , :param account: :return: ''' # 调用db_handle下的handle方法,返回路径/db/accounts db_path = db_handler.db_handler(settings.DATABASE) account_file = '%s/%s.json' % (db_path, account) account_new = '%s/%s.json' % (db_path, account) if os.path.isfile(account_file): with open(account_file, 'r', encoding='utf-8') as f: account_data = json.load(f) account_data['status'] = 1 with open(account_new, 'w', encoding='utf-8') as fnew: new_data = json.dump(account_data, fnew) # 添加账户 def add_account(): ''' 添加账户,是admin添加的用户,下次就可以登陆添加的账户了 :return: ''' acc_dic = { 'id':None, 'balance':None, 'password': None, 'credit': None, 'enroll_date': None, 'expire_date': None, 'status': None # 0 = naormal,1 = locked , 2 = disabled } menu = { 0:"请输入要添加的账户:", 1:"请输入要添加的余额:", 2:"请输入要添加的密码:", 3:"请输入要添加的信用额度(only more than 0)", 4:"请输入要添加的办卡日期(such as 2018-8-8)", 5:"请输入要添加的卡到期时间(such as 2018-8-8)", 6:"请输入是否锁定添加账号的状态(only input 0 or 1)", } menu_user = { 0: "id", 1: "balance", 2: "password", 3: "credit", 4: "enroll_date", 5: "expire_date", 6: "status", } print("\033[31;1m\t\twelcome to add account\033[0m") print('*'.center(40,'*')) for i in range(7): data = input('%s'%menu[i]).strip() acc_dic["%s" % menu_user[i]] = data accounts.dump_account(acc_dic) print("\033[32;1mcongratulations you account was created successfully\033[0m") return True # 退出程序 def logout(): ''' 退出登陆 :return: ''' print("\033[32;1m-------Looking forward to your next visit-------\033[0m") exit() def auth_login(): ''' 登陆管理员密码账号 :return: ''' print("\033[34;1m-------Welcome into the management interface--------\033[0m") managename = input("\033[34;1mplease input Username:\033[0m") password = input("\033[34;1mplease input Password:\033[0m") account = account_auth(managename,password) return account def account_auth(managename,password): ''' 管理员认证信息 {"id": admin,"password": "root" } :return: ''' db_path = db_handler.db_handler(settings.DATABASE) # 调用db_handle下的handle方法,返回路径/db/accounts managename_file = '%s/%s.json'%(db_path,managename) if os.path.isfile(managename_file): with open(managename_file,'r',encoding='utf-8') as f: manage_data = json.load(f) # print(manage_data) if manage_data['password'] == password: print("\033[31;1m-------Welcome to the administrator--------\033[0m") return manage_interactive(managename) else: print("\033[31;1mAccount or Passwordoes not correct!\033[0m") # 管理界面主程序 def manage_interactive(managename): menu = ''' \033[31;1m-----------management console----------- 1,add_account 2,lock_account 3,unblock_account 4, exit\033[0m''' menu_dic = { '1':add_account, '2':lock_account, '3':unlock_account, '4': logout } exit_flag = False while not exit_flag: print(menu) user_option = input('please input your choice>>>').strip() if user_option in menu_dic: print(menu_dic[user_option]()) else: print("\033[31;1mYou choice doesn't exist!\033[0m") def run_manage(): ''' 当程序启动的时候调用,主要用于实现主要交互逻辑,客户认证登陆 :return: ''' auth_login()
core/tranction.py
# _*_ coding: utf-8 _*_ import json from conf import settings from core import accounts def make_transaction(log_obj,account_data,tran_type,amount,**others): ''' 处理所有的用户的交易 :param log_obj: :param amount_data: user account data :param tran_type: transaction type :param amount: transaction amount :param other: mainly for logging usage :return: ''' # 将字符串类型转换为float类型 amount = float(amount) # tran_type 交易类型 if tran_type in settings.TRANSACTION_TYPE: # 利息金额 interest =amount * settings.TRANSACTION_TYPE[tran_type]['interest'] old_balance = account_data['balance'] if tran_type in settings.TRANSACTION_TYPE: # 利息金额 interest = amount * settings.TRANSACTION_TYPE[tran_type]["interest"] # 用户原金额 old_balace = account_data["balance"] if settings.TRANSACTION_TYPE[tran_type]['action'] == 'plus': new_balance = float(old_balance) + amount + float(interest) elif settings.TRANSACTION_TYPE[tran_type]['action'] == 'minus': new_balance = float(old_balance) - amount - float(interest) # 做一个判断小于0的操作,减钱时对帐户金额进行检查,防止超额 if new_balance <0: print('\033[31;1mYour credit [%s] is not enough for this transaction' '[%s]'%(account_data['credit'],(amount + interest),old_balance)) return account_data['balance'] = new_balance # 保存新的余额返回到文件中 accounts.dump_account(account_data) log_obj.info("account:%s action:%s amount:%s interest:%s " %(account_data['id'],tran_type,amount,interest)) return account_data else: print("\033[31;1mTransaction type [%s] is not exist\033[0m"%tran_type)
db/account_sample/py(此处只举一个例子)
# _*_ coding: utf-8 _*_ import json acc_dic = { 'id':1234, 'password':'abc', 'credit':15000, 'balance':15000, 'enroll_date':'2018-01-01', 'expire_date':'2023-01-01', # 'pay_day':22, #支付日期(但是现在没有要求,可以不考虑) 'status':0 #0 = naormal,1 = locked , 2 = disabled } print(acc_dic,type(acc_dic)) a = json.dumps(acc_dic) print(a,type(a))
7.2 shopping代码
shopping_main.py
# _*_ coding: utf-8 _*_ import datetime import os import json def auth_login(): ''' 登陆密码账号 :return: ''' print("-------welcome to shopping_mall--------") username = input("\033[32;1mplease input Username:\033[0m") password = input("\033[32;1mplease input Password:\033[0m") account = account_auth(username,password) return account def account_auth(username,password): ''' 用户认证账户信息格式化读取模块,并验证账号密码是否正确 :return: ''' username_file = '%s.json'%username if os.path.isfile(username_file): with open(username_file,'r',encoding='utf-8') as f: username_data = json.load(f) if username_data['password'] == password: balance = username_data['balance'] print("\033[32;1myou balance is %s\033[0m"% balance) return account_shopping(username,balance) else: print("\033[31;1mAccount or Passwordoes not correct!\033[0m") def account_shopping(username,balance): '''用户购物操作 这次只能买一件东西,不能重复买 ''' goods = [ {"name": "电脑", "price": 6999}, {"name": "鼠标", "price": 300}, {"name": "游艇", "price": 2000}, {"name": "美女", "price": 9980}, ] print("*".center(40, '*')) print("goods".center(40, '-')) for index,item in enumerate(goods): print(index,item) print('end'.center(40,'-')) while True: choice = input("请输入要想购买的商品编号(或者按q直接退出):") if choice.isdigit(): #判断是否为数字 choice = int(choice) if choice>=0 and choice<len(goods): #判断是否在商品范围里面 choice_goods = goods[choice].get('name') print("you will buy \033[32;1m%s\033[0m"%choice_goods) if balance > goods[choice].get('price'): new_balance = balance - goods[choice].get('price') print("now your balance is \033[32;1m%s\033[0m"% new_balance) else: print("sorry your balance is \033[32;1m%s\033[0m,unable to purchase" %balance) continue else: print("\033[31;1mplease input the correct goods number\033[0m") elif choice == 'q': exit() return write_data(username,new_balance) def write_data(username, new_balance): ''' 买了东西之后,余额信息更新一下 :param username: 用户姓名 :param new_balance: 余额信息 :return: ''' username_file = '%s.json' % username usernew_file = '%s.json' % username f = open(username_file, 'r', encoding='utf-8') username_data = json.load(f) username_data['balance'] = new_balance f.close() fnew = open(usernew_file, 'w', encoding='utf-8') new_data = json.dump(username_data,fnew) fnew.close() def run_shopping(): ''' 当程序启动的时候,调用,主要用于实现主要交互逻辑,客户认证登陆函数 :return: ''' userdata = auth_login() def account_recharge(balance): '''用户充值操作,此思路是接ATM''' recharge_money = input("please input recharge money: ") recharge_money =int(recharge_money) balance = balance + recharge_money print("Congratulations. Recharge success. balance is \033[32;1m%s\033[0m" %balance) return balance def legout(): ''' 退出程序 :return: ''' print("\033[32;1m-------Looking forward to your next visit-------\033[0m") exit() def interactive(acc_data): ''' 用户交互 :param acc_data: :return: ''' msg = ( ''' ------------------SHOPPING INFO -------------- \033[31;1m1.购物 2.充值 3.退回 \033[0m''' ) menu_dic = { "1": account_shopping, "2": account_recharge, "3": legout, } exit_flag = False while not exit_flag: print(msg) user_choice = input(">>>>").strip() if user_choice in menu_dic: menu_dic[user_choice]() else: print("\033[31;1mYou choice doesn't exist!\033[0m") def manage_accountauth(): ''' 管理员账户认证函数, :return: ''' manage_account = [] with open('manageinfo.txt', 'r', encoding='utf-8') as f: for i in f.readlines(): i_space = i.replace('\n', '') manage_account.append(i_space) return manage_account def account_balance(): ''' 账户余额信息格式化读取模块 :return: ''' balance = [] with open('account_balance.txt', 'r', encoding='utf-8') as f: for i in f.readlines(): i_space = i.replace('\n', '') balance.append(i_space) return balance def account_save(account,cash): ''' 把账户及其余额信息持久化到本地,以消除文件内容,然后以读写的方式打开文件 :param account: 用户信息 :return: ''' save_info = open('account_balance.txt','w+',encoding='utf-8') save_info.write(account) save_info.write('\n') save_info.write(str(cash)) save_info.close() def account_auth_changshi(username,password): ''' 用户认证账户信息格式化读取模块,尝试登陆三次, {"id": 123,"password": "123", "balance": 20000 } :return: ''' retry_count = 0 while username_data['password'] is not True and retry_count <3: username_file = '%s.json'%username if os.path.isfile(username_file): with open(username_file,'r',encoding='utf-8') as f: username_data = json.load(f) print(username_data['password']) print(password) if username_data['password'] == password: return account_shopping else: print("\033[31;1mAccount or Passwordoes not correct!\033[0m") retry_count += 1 else: print("Account [%s] try logging too many times..." % username) exit()
shopping_run.py
# _*_ coding: utf-8 _*_ import os import sys #添加环境变量 BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) #找到路径 sys.path.append(BASE_DIR) #添加路径 from shopping import shopping_main '''购物车程序的执行文件''' if __name__ == '__main__': shopping_main.run_shopping()