Python实现ATM+购物商城

Python实现ATM+购物商城

一、程序介绍

需求：

模拟实现一个ATM + 购物商城程序
    额度 15000或自定义
    实现购物商城，买东西加入 购物车，调用信用卡接口结账
    可以提现，手续费5%
    每月22号出账单，每月10号为还款日，过期未还，按欠款总额 万分之5 每日计息
    支持多账户登录
    支持账户间转账
    记录每月日常消费流水
    提供还款接口
    ATM记录操作日志 
    提供管理接口，包括添加账户、用户额度，冻结账户等。。。
    用户认证用装饰器

实现功能：
    额度 15000或自定义
    实现购物商城，买东西加入 购物车，调用信用卡接口结账
    可以提现，手续费5%
    支持多账户登录
    记录每月日常消费流水
    提供还款接口
    ATM记录操作日志 
    提供管理接口，包括添加账户、用户额度，冻结账户等。。。
    用户认证用装饰器

程序结构:

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    #所有的交易日志

二、流程图

三、代码

bin/atm.py


#!/usr/bin/env python
# -*- coding: utf-8 -*-

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__':
    main.run()

conf/settings.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
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',
}

TRANSACTION_TYPE = {
    'repay':{'action':'plus', 'interest':0.03},
    'withdraw':{'action':'minus', 'interest':0.05},
    'transfer':{'action':'minus', 'interest':0.05},
    'consume':{'action':'minus', 'interest':0},

}

core/main.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

'''
main program handle module , handle all the user interaction stuff

'''

from core import auth
from core import accounts
from core import logger
from core import accounts
from core import transaction
from core.auth import login_required
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):
    account_data = accounts.load_current_balance(acc_data['account_id'])
    data_info = u'''
    33[34;1m 账号ID：%s
    余额：  %s
    信用度：%s
    账号注册时间：%s
    账号过期时间：%s
    工资天数：%s
    33[0m'''%(acc_data['account_id'],
                account_data['balance'],
                account_data['credit'],
                account_data['enroll_date'],
                account_data['expire_date'],
                account_data['pay_day'],)
    print(data_info)


@login_required
def repay(acc_data):
    '''
    print current balance and let user repay the bill
    :return:
    '''
    account_data = accounts.load_current_balance(acc_data['account_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("33[33;1mInput repay amount:33[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('''33[42;1mNew Balance:%s33[0m''' %(new_balance['balance']))

        else:
            print('33[31;1m[%s] is not a valid amount, only accept integer!33[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("33[33;1mInput withdraw amount:33[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('''33[42;1mNew Balance:%s33[0m''' %(new_balance['balance']))

        else:
            print('33[31;1m[%s] is not a valid amount, only accept integer!33[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):
    exit()


def shopping(acc_data):
    '''

    :param acc_data:
    :return:
    '''
    product_list = [
        ['Iphone7 Plus', 6500],
        ['Iphone8 ', 8200],
        ['MacBook Pro', 12000],
        ['Python Book', 99],
        ['Coffee', 33],
        ['Bike', 666],
        ['pen', 2]
    ]
    shopping_cart = []
    count = 0
    salary = acc_data['account_data']['balance']
    while True:
        account_data = accounts.load_current_balance(acc_data['account_id'])
        print(">> 欢迎来到电子商城 您的余额是 %s 元<<" % (salary))
        for index, i in enumerate(product_list):  # 循环商品列表，商品列表索引
            print("%s.	%s	%s" % (index, i[0], i[1]))  # 打印商品列表，显示商品列表索引
        choice = input(">>请输入商品序号或输入 exit 退出商城>>: ").strip()
        if len(choice) == 0:  # 判断输入字符串是否为空和字符串长度
            print('-->您没有选择商品<--')
            continue
        if choice.isdigit():  # 判断输入的choice是不是一个数字
            choice = int(choice)  # 把输入的字符串转成整型
            if choice < len(product_list) and choice >= 0:  # 输入的整数必须小于商品列表的数量
                product_item = product_list[choice]  # 获取商品
                if salary >= product_item[1]:  # 拿现有金额跟商品对比，是否买得起
                    salary -= product_item[1]  # 扣完商品的价格
                    shopping_cart.append(product_item)  # 把选着的商品加入购物车
                    print("添加 33[32;1m%s33[0m 到购物车,您目前的金额是 
    33[31;1m%s33[0m" % (product_item[0], salary))
                else:
                    print("对不起，您的金额不足,还差 33[31;1m%s33[0m" % (product_item[1] - salary,))
            else:
                print("-->没有此商品<--")
        elif choice == "exit":
            total_cost = 0
            print("您的购物车列表:")
            for i in shopping_cart:
                print(i)
                total_cost += i[1]
            print("您的购物车总价是: 33[31;1m%s33[0m" % (total_cost,))
            print("您目前的余额是: 33[31;1m%s33[0m" % (salary,))
            new_balance = transaction.make_transaction(trans_logger, account_data, 'withdraw', total_cost)
            if new_balance:
                print('''33[42;1mNew Balance:%s33[0m''' % (new_balance['balance']))
            break


def interactive(acc_data):
    '''
    interact with user
    :return:
    '''
    menu = u'''
    ------- hehe Bank ---------
    33[32;1m
    1.  账户信息(实现)
    2.  还款(实现)
    3.  取款(实现)
    4.  转账
    5.  账单
    6.  商城(实现)
    7.  退出(实现)
    33[0m'''
    menu_dic = {
        '1': account_info,
        '2': repay,
        '3': withdraw,
        '4': transfer,
        '5': pay_check,
        '6': shopping,
        '7': 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("33[31;1mOption does not exist!33[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)
    if user_data['is_authenticated']:
        user_data['account_data'] = acc_data
        interactive(user_data)

core/transaction.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

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('''33[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("33[31;1mTransaction type [%s] is not exist!33[0m" % tran_type)

core/accounts.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

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 data

    # 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

core/auth.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import os
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):
        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("33[31;1mAccount [%s] has expired,please contact the back to get a new card!33[0m" % account)
                else: #passed the authentication
                    return  account_data
            else:
                print("33[31;1mAccount ID or password is incorrect!33[0m")
    else:
        print("33[31;1mAccount [%s] does not exist!33[0m" % account)


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()  #连接数据库 file_execute内存地址
    data = db_api("select * from accounts where account=%s" % account) #执行sql
    if data['password'] == password:
        exp_time_stamp = time.mktime(time.strptime(data['expire_date'], "%Y-%m-%d"))
        if time.time() > exp_time_stamp:
            print("33[31;1mAccount [%s] has expired,please contact the back to get a new card!33[0m" % account)
        else:  # passed the authentication
            return data
    else:
        print("33[31;1mAccount ID or password is incorrect!33[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 :
        account = input("33[32;1maccount:33[0m").strip()
        password = input("33[32;1mpassword:33[0m").strip()
        auth = acc_auth2(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()

core/db_handler.py

#!_*_coding:utf-8_*_
#__author__:"Alex Li"

'''
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)
    #db_path ='%s/%s' %(conn_params['path'],conn_params['name'])
    return file_execute
def db_handler():
    '''
    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 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("33[31;1mAccount [%s] does not exist!33[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

core/logger.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

'''
handle all the logging works
'''

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

db/account_sample.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-


import json
acc_dic = {
    'id': 1234,
    'password': 'abc',
    'credit': 15000,
    'balance': 15000,
    'enroll_date': '2016-01-02',
    'expire_date': '2021-01-01',
    'pay_day': 22,
    'status': 0 # 0 = normal, 1 = locked, 2 = disabled
}

print(json.dumps(acc_dic))