#这是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 :    %sBalance:    %s''' %(account_data['credit'],account_data['balance'])  #打印出信用卡额度和现金print(current_balance)back_flag = Falsewhile 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 :    %sBalance:    %s''' %(account_data['credit'],account_data['balance'])print(current_balance)back_flag = Falsewhile 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 = Truedef 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 = Falsewhile not exit_flag:print(menu)user_option = input(">>:").strip()if user_option in menu_dic:print('accdata',acc_data)#acc_data['is_authenticated'] = Falsemenu_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 ossys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))from core import db_handlerfrom conf import settingsfrom core import loggerimport jsonimport 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 authenticationreturn dataelse: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且初始次数小于3account = 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                        #修改认证属性为Trueuser_data['account_id'] = account                           #修改用户账号为认证成功的账号logger.acc_logger.debug('welcome')return auth                                       #返回这个字典retry_count +=1else: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.DATABASEif conn_params['engine'] == 'file_storage':  #意思是如果储存方式为文件储存，return file_db_handle(conn_params)      #那么返回一个从文件读取数据的函数# elif conn_params['engine'] == 'mysql':        # 这是扩展为数据库的方法....提供思路 无用#     #     pass #tododef write(file):        #这是我添加的函数，因为会报错f = open(file,'r')account_data = json.load(f)return account_datadef file_execute(sql,**kwargs):conn_params = settings.DATABASEdb_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 clausecolumn,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_dataelse:exit("\033[31;1mAccount [%s] does not exist!\033[0m" % val )elif sql_list[0].startswith("update") and len(sql_list)> 1:#has where clausecolumn, 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 settingsdef 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 creditif  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 ))returnaccount_data['balance'] = new_balanceaccounts.dump_account(account_data) #save the new balance back to filelog_obj.info("account:%s   action:%s    amount:%s   interest:%s" %(account_data['id'], tran_type, amount,interest) )return account_dataelse:print("\033[31;1mTransaction type [%s] is not exist!\033[0m" % tran_type)

#这是logger.py文件
import sysimport os
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
import logging
from conf import settingsdef logger(log_type):#create loggerlogger = logging.getLogger(log_type)logger.setLevel(settings.LOG_LEVEL)# create console handler and set level to debugch = logging.StreamHandler()ch.setLevel(settings.LOG_LEVEL)# create file handler and set level to warninglog_file = "%s/log/%s" %(settings.BASE_DIR, settings.LOG_TYPES[log_type])fh = logging.FileHandler(log_file)fh.setLevel(settings.LOG_LEVEL)# create formatterformatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')# add formatter to ch and fhch.setFormatter(formatter)fh.setFormatter(formatter)# add ch and fh to loggerlogger.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')