alex的ATM学习笔记
这是理解和学习ALex的简单ATM https://github.com/triaquae/py3_training/tree/master/atm
day5-atm/
├── 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 #用于从文件里加载和存储账户数据
│ │ ├── auth.py #用户认证模块
│ │ ├── db_handler.py #数据库连接引擎
│ │ ├── logger.py #日志记录模块
│ │ ├── main.py #主逻辑交互程序
│ │ └── transaction.py #记账\还钱\取钱等所有的与账户金额相关的操作都 在这
│ ├── db #用户数据存储的地方
│ │ ├── __init__.py
│ │ ├── account_sample.py #生成一个初始的账户数据 ,把这个数据 存成一个 以这个账户id为文件名的文件,放在accounts目录 就行了,程序自己去会这里找
│ │ └── accounts #存各个用户的账户数据 ,一个用户一个文件
│ │ └── 1234.json #一个用户账户示例文件
│ └── log #日志目录
│ ├── __init__.py
│ ├── access.log #用户访问和操作的相关日志
│ └── transactions.log #所有的交易日志
└── shopping_mall #电子商城程序,需单独实现
└── __init__.py
#这是atm.py文件
import os import sys # base_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) 一个上上级的绝对路径并赋给变量 # # sys.path.append(base_dir) 把路径添加到调用模块时的查找路径 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) #这是上面两句和在一起 from core import main #通过添加路径，我们可以调用ATM目录下的core包的main模块 if __name__ == '__main__': # __name__ 这个函数保证只有在当前文件运行下面的内容 main.run() # 调用main模块下的run（）函数
#这是main.py文件
from core import auth
from core import accounts
from core import logger
from core import accounts
from core import transaction
from core import db_handler
from core.auth import login_required
import time
import json
#transaction logger
# trans_logger = logger.logger('transaction') #一个打印transaction的日志生成器
# #access logger
# access_logger = logger.logger('access') #一个打印access的日志生成器
trc_logger = logger.trc_logger
acc_logger = logger.acc_logger
#temp account data ,only saves the data in memory
user_data = {
'account_id':None,
'is_authenticated':False,
'account_data':None
}
def account_info(acc_data):
print(user_data)
@login_required
def repay(acc_data): #acc_data是一个字典
'''
print current balance and let user repay the bill
:return:
'''
account_data = accounts.load_current_balance(acc_data['account_id']) #获得存储的用户信息文档
account_data = db_handler.write(account_data) #转换为字典
#for k,v in account_data.items():
# print(k,v )
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[33;1mInput repay amount:\033[0m").strip() #输入还款金额
if len(repay_amount) >0 and repay_amount.isdigit():
print('还款')
new_balance = transaction.make_transaction(trc_logger,account_data,'repay', repay_amount) #
if new_balance:
print('''\033[42;1mNew Balance:%s\033[0m''' %(new_balance['balance']))
else:
print('\033[31;1m[%s] is not a valid amount, only accept integer!\033[0m' % repay_amount)
if repay_amount == 'b':
back_flag = True
def withdraw(acc_data):
'''
print current balance and let user do the withdraw action
:param acc_data:
:return:
'''
account_data = accounts.load_current_balance(acc_data['account_id'])
account_data = db_handler.write(account_data)
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 amount:\033[0m").strip()
if len(withdraw_amount) >0 and withdraw_amount.isdigit():
new_balance = transaction.make_transaction(trc_logger,account_data,'withdraw', withdraw_amount)
if new_balance:
print('''\033[42;1mNew Balance:%s\033[0m''' %(new_balance['balance']))
else:
print('\033[31;1m[%s] is not a valid amount, only accept integer!\033[0m' % withdraw_amount)
if withdraw_amount == 'b':
back_flag = True
def transfer(acc_data):
pass
def pay_check(acc_data):
pass
def logout(acc_data):
pass
def interactive(acc_data):
'''
interact with user
:return:
'''
menu = u'''
------- Oldboy Bank ---------
\033[32;1m1. 账户信息
2. 还款(功能已实现)
3. 取款(功能已实现)
4. 转账
5. 账单
6. 退出
\033[0m'''
menu_dic = {
'1': account_info,
'2': repay,
'3': withdraw,
'4': transfer,
'5': pay_check,
'6': logout,
}
exit_flag = False
while not exit_flag:
print(menu)
user_option = input(">>:").strip()
if user_option in menu_dic:
print('accdata',acc_data)
#acc_data['is_authenticated'] = False
menu_dic[user_option](acc_data)
else:
print("\033[31;1mOption does not exist!\033[0m")
def run():
'''
this function will be called right a way when the program started, here handles the user interaction stuff
:return:
'''
acc_data = auth.acc_login(user_data,acc_logger) #赋值并执行auth模块下的认证函数，把用户数据字典(空)和日志器传到这个函数,执行成功会返回一个储存的用户字典
if user_data['is_authenticated']: #如果认证成功（开始为FALSE，认证成功后为TRUE）
user_data['account_data'] = acc_data #ud字典的ad键内容就嵌套一个用户字典
interactive(user_data) #操作信用卡的函数
#这是auth.py文件
import sys
import os
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
from core import db_handler
from conf import settings
from core import logger
import json
import time
def login_required(func):
"验证用户是否登录"
def wrapper(*args,**kwargs):
#print('--wrapper--->',args,kwargs)
if args[0].get('is_authenticated'):
return func(*args,**kwargs)
else:
exit("User is not authenticated.")
return wrapper
def acc_auth(account,password):
'''
account auth func
:param account: credit account number
:param password: credit card password
:return: if passed the authentication , retun the account object, otherwise ,return None
'''
db_path = db_handler.db_handler(settings.DATABASE) #拿到一个存储账户信息文件的路径，使用函数是为了扩展可能的方法（从数据库）
account_file = "%s/%s.json" %(db_path,account) #从上面的路径获取存储某用户信息的文件
# print(account_file)
if os.path.isfile(account_file): #os.path.isfile方法判断是否存括号内的文件
with open(account_file,'r') as f: #确定存在后，以读模式打开
account_data = json.load(f) #用json模块的load方法读取内容
if account_data['password'] == password: #如果密码正确
exp_time_stamp = time.mktime(time.strptime(account_data['expire_date'], "%Y-%m-%d")) #从用户信息字典找到过期时间转换成秒
if time.time() >exp_time_stamp: #当前的秒如果大于过期秒
print("\033[31;1mAccount [%s] has expired,please contact the back to get a new card!\033[0m" % account) #提示过期
else: #passed the authentication
return account_data #没过期就返回用户信息字典
else:
return False
print("\033[31;1mAccount ID or password is incorrect!\033[0m") #如果密码不正确的提示
else:
return False
print("\033[31;1mAccount [%s] does not exist! FLAG\033[0m" % account) #如果不存在文件的提示
# logger = logger.acc_logger
if __name__ == '__main__': logger.debug('welcome') def acc_auth2(account,password): ''' 优化版认证接口 :param account: credit account number :param password: credit card password :return: if passed the authentication , retun the account object, otherwise ,return None ''' db_api = db_handler.db_handler() data = db_api("select * from accounts where account=%s" % account) if data['password'] == password: exp_time_stamp = time.mktime(time.strptime(data['expire_date'], "%Y-%m-%d")) if time.time() > exp_time_stamp: print("\033[31;1mAccount [%s] has expired,please contact the back to get a new card!\033[0m" % account) else: # passed the authentication return data else: print("\033[31;1mAccount ID or password is incorrect!\033[0m") def acc_login(user_data,log_obj): ''' account login func :user_data: user info data , only saves in memory :return: ''' retry_count = 0 #初始次数 while user_data['is_authenticated'] is not True and retry_count < 3 : #如果用户认证属性为Fales且初始次数小于3 account = input("\033[32;1maccount:\033[0m").strip() #输入账号 password = input("\033[32;1mpassword:\033[0m").strip() #输入密码 auth = acc_auth(account, password) #把输入的账号密码传给认证函数去确认是否正确 正确就返回了一个用户字典 if auth: #not None means passed the authentication #认证成功 user_data['is_authenticated'] = True #修改认证属性为True user_data['account_id'] = account #修改用户账号为认证成功的账号 logger.acc_logger.debug('welcome') return auth #返回这个字典 retry_count +=1 else: log_obj.error("account [%s] too many login attempts" % account) exit()
#这是db_handler.py文件
''' handle all the database interactions ''' import json,time ,os from conf import settings def file_db_handle(conn_params): ''' parse the db file path :param conn_params: the db connection params set in settings :return: ''' # print('file db:',conn_params) #打印DATABASE字典，无用 db_path ='%s/%s' %(conn_params['path'],conn_params['name']) #一个ATM/db/accounts/的路径，即储存账户信息的路径 return db_path #返回这个路径 def db_handler(conn_params): ''' connect to db :param conn_parms: the db connection params set in settings :return:a ''' # conn_params = settings.DATABASE if conn_params['engine'] == 'file_storage': #意思是如果储存方式为文件储存， return file_db_handle(conn_params) #那么返回一个从文件读取数据的函数 # elif conn_params['engine'] == 'mysql': # 这是扩展为数据库的方法....提供思路 无用 # # pass #todo def write(file): #这是我添加的函数，因为会报错 f = open(file,'r') account_data = json.load(f) return account_data def file_execute(sql,**kwargs): conn_params = settings.DATABASE db_path = '%s/%s' % (conn_params['path'], conn_params['name']) print(sql,db_path) sql_list = sql.split("where") print(sql_list) if sql_list[0].startswith("select") and len(sql_list)> 1:#has where clause column,val = sql_list[1].strip().split("=") if column == 'account': account_file = "%s/%s.json" % (db_path, val) print(account_file) if os.path.isfile(account_file): with open(account_file, 'r') as f: account_data = json.load(f) return account_data else: exit("\033[31;1mAccount [%s] does not exist!\033[0m" % val ) elif sql_list[0].startswith("update") and len(sql_list)> 1:#has where clause column, val = sql_list[1].strip().split("=") if column == 'account': account_file = "%s/%s.json" % (db_path, val) #print(account_file) if os.path.isfile(account_file): account_data = kwargs.get("account_data") with open(account_file, 'w') as f: acc_data = json.dump(account_data, f) return True
#这是accounts.py文件
import json import time from core import db_handler from conf import settings def load_current_balance(account_id): ''' return account balance and other basic info :param account_id: :return: ''' db_path = db_handler.db_handler(settings.DATABASE) #获得存储用户信息文件夹的路径 account_file = "%s/%s.json" %(db_path,account_id) #获取存储用户信息的文件 # # db_api = db_handler.db_handler() # data = db_api("select * from accounts where account=%s" % account_id) return account_file # with open(account_file) as f: # acc_data = json.load(f) # return acc_data def dump_account(account_data): ''' after updated transaction or account data , dump it back to file db :param account_data: :return: ''' # db_api = db_handler.db_handler() # data = db_api("update accounts where account=%s" % account_data['id'],account_data=account_data) db_path = db_handler.db_handler(settings.DATABASE) account_file = "%s/%s.json" %(db_path,account_data['id']) with open(account_file, 'w') as f: acc_data = json.dump(account_data,f) return True
#这是transaction.py文件
def make_transaction(log_obj,account_data,tran_type,amount,**others): #日志，账户字典，操作类型，金额，其他 ''' deal all the user transactions :param account_data: user account data :param tran_type: transaction type :param amount: transaction amount :param others: mainly for logging usage :return: ''' amount = float(amount) if tran_type in settings.TRANSACTION_TYPE: interest = amount * settings.TRANSACTION_TYPE[tran_type]['interest'] #计算利息 old_balance = account_data['balance'] #之前的现金 if settings.TRANSACTION_TYPE[tran_type]['action'] == 'plus': # new_balance = old_balance + amount + interest # elif settings.TRANSACTION_TYPE[tran_type]['action'] == 'minus': new_balance = old_balance - amount - interest #check credit if new_balance <0: print('''\033[31;1mYour credit [%s] is not enough for this transaction [-%s], your current balance is [%s]''' %(account_data['credit'],(amount + interest), old_balance )) return account_data['balance'] = new_balance accounts.dump_account(account_data) #save the new balance back to file 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)
#这是logger.py文件
import sys
import os
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
import logging
from conf import settings
def logger(log_type):
#create logger
logger = logging.getLogger(log_type)
logger.setLevel(settings.LOG_LEVEL)
# create console handler and set level to debug
ch = logging.StreamHandler()
ch.setLevel(settings.LOG_LEVEL)
# create file handler and set level to warning
log_file = "%s/log/%s" %(settings.BASE_DIR, settings.LOG_TYPES[log_type])
fh = logging.FileHandler(log_file)
fh.setLevel(settings.LOG_LEVEL)
# create formatter
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
# add formatter to ch and fh
ch.setFormatter(formatter)
fh.setFormatter(formatter)
# add ch and fh to logger
logger.addHandler(ch)
logger.addHandler(fh)
return logger
# 'application' code
'''logger.debug('debug message')
logger.info('info message')
logger.warn('warn message')
logger.error('error message')
logger.critical('critical message')'''
acc_logger = logger('access')
trc_logger = logger('transaction')
# acc_logger.debug('123')
