Fork me on GitHub

模拟实现一个ATM + 购物商城程序

           记得上次小编上传了一个购物车程序,这次呢稍微复杂一点,还是那句话,上传在这里不是为了炫耀什么,只是督促小编学习,如果大神有什么意见和建议,欢迎指导。

  于2018.3.11开始动笔,继续完成这个项目。

一,需求:模拟实现一个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()

  

 

 

 

                                                                                                              

 

posted @ 2017-09-10 16:46  战争热诚  阅读(1502)  评论(6编辑  收藏  举报