python项目练习

程序框图 （消费模块暂未写入）

bin:程序执行

---

 import os
 import sys
 base_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
 print(base_dir)
 sys.path.append(base_dir)

 from core import main

 if __name__ == '__main__':   #当作为脚本直接运行的时候，此时__name__等于__main__，当作为模块导入的时候，__name__为文件名但不带.py，故不运行if后语句。
     main.run()

atm.py

config：配置文件

---

 import os
 import sys
 import logging
 BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))

 DATABASE = {
     'engine': 'file_storage', #support mysql,postgresql in the future
     'name':'accounts',
     'path': "%s/db" % BASE_DIR
 }

 LOG_LEVEL = logging.INFO
 LOG_TYPES = {
     'transaction': 'transactions.log',
     'access': 'access.log',
     '':'11111.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},
 }

settings

core:程序主要代码

---

 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)
     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_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

accounts

 import os
 from core import db_handler
 from conf import settings
 from core import logger
 import json
 import time

 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)  #base_dir + accounts + account.json
     if os.path.isfile(account_file):   #判断文件名是否存在，存在执行下面语句
         with open(account_file,'r') as f:
             account_data = json.load(f)
             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:
                 print("\033[31;1mAccount ID or password is incorrect!\033[0m")
     else:
         print("\033[31;1mAccount [%s] does not exist!\033[0m" % account)

 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 :
         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
             user_data['account_id'] = account
             #print("welcome")
             return auth
         retry_count +=1
     else:
         log_obj.error("account [%s] too many login attempts" % account)
         exit()

登陆认证

 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)
     db_path ='%s/%s' %(conn_params['path'],conn_params['name'])
     return db_path

 def mysql_db_handle(conn_parms):
     pass
 def db_handler(conn_parms):
     '''
     connect to db
     :param conn_parms: the db connection params set in settings
     :return:a
     '''

     if conn_parms['engine'] == 'file_storage':
         return file_db_handle(conn_parms)

     if conn_parms['engine'] == 'mysql':
         return mysql_db_handle(conn_parms)

数据存储路径

 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

日志

 from core import auth
 from core import accounts
 from core import logger
 from core import accounts
 from core import transaction
 import time

 #transaction logger
 trans_logger = logger.logger('transaction')
 #access logger
 access_logger = logger.logger('access')

 #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)
 def repay(acc_data):
     '''
     print current balance and let user repay the bill
     :return:
     '''
     account_data = accounts.load_current_balance(acc_data['account_id'])  #获取用户id，就是要用实时的最新数据，为了安全
     #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('ddd 00')
             new_balance = transaction.make_transaction(trans_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'])
     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(trans_logger,account_data,'withdraw', withdraw_amount)  # new_balance就是 函数返回值 acount_data
             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 = {
         '': account_info,
         '': repay,
         '': withdraw,
         '': transfer,
         '': pay_check,
         '': logout,
     }
     exit_flag = False
     while not exit_flag:
         print(menu)
         user_option = input(">>:").strip()
         if user_option in menu_dic:
             menu_dic[user_option](acc_data)  #比如选择了2 ，则运行 repay(acc_data)，调用repay函数

         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,access_logger)   #userdata作为条件，access_logger作为日志信息传入
     if user_data['is_authenticated']:
         user_data['account_data'] = acc_data  #acc_data 即是用户信息 1234.json
         interactive(user_data) #交互

主程序

 from conf import settings
 from core import accounts
 from core import logger
 #transaction logger

 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)

交易种类

db:用户信息存储

---

 {"id": "gkx", "password": "", "credit": 15000, "balance": 15000, "enroll_date": "2016-01-02", "expire_date": "2021-01-01", "pay_day": 22, "status": 0}