ATM+购物车

ATM+购物车

目录

• ATM+购物车
• 1 需求
• 2 功能
• 3 目录结构
• 4 思路及步骤
  • 4.1 环境bin/start.py
    • 4.1.1 将解释器添加到环境变量中
    • 4.1.2 启动主程序
  • 4.2 配置conf/settings.py
    • 4.2.1 获取项目根目录路径
    • 4.2.2 获取user_data目录路径
    • 4.2.3 获取log目录路径
  • 4.3 数据处理db/db_handle.py
    • 4.3.1 保存用户信息save
    • 4.3.2 返回用户信息select
  • 4.4 视图层view/src
    • 4.4.1 主启动函数run()
    • 4.4.2 功能实现
      • 4.4.2.1 注册功能register()
      • 4.4.2.2 登录功能login()
      • 4.4.2.3 查看余额功能check_balance()
      • 4.4.2.4 提现功能withdrawal()
      • 4.4.2.5 还款功能repay()
      • 4.4.2.6 转账功能transfer()
      • 4.4.2.7 查看流水功能check_flow()
      • 4.4.2.8 购物功能shop()
      • 4.4.2.9 查看购物车功能check_shopping_cart()
      • 4.4.2.10 管理员功能admin()
      • 4.4.2.11 注销功能logout()
  • 4.5 公共lib/common
    • 4.5.1 登陆认证装饰器
    • 4.5.2 日志
  • 4.6 接口interface
    • 4.6.1 银行相关bank.py
      • 4.6.1.1 查看余额接口
      • 4.6.1.2 提现接口
      • 4.6.1.3 查询欠款接口
      • 4.6.1.4 还款接口
      • 4.6.1.5 转账接口
      • 4.6.1.6 查看流水接口
      • 4.6.1.7 消费接口
    • 4.6.2 购物相关shopping.py
      • 4.6.2.1 购买商品接口
      • 4.6.2.2 查看购物车接口
    • 4.6.3 用户相关user.py
      • 4.6.3.1 注册接口
      • 4.6.3.2 提取用户信息接口
      • 4.6.3.3 锁定用户接口
      • 4.6.3.4 解锁用户接口

1 需求

1.额度 15000或自定义 --> 注册功能
2.实现购物商城，买东西加入 购物车，调用信用卡接口结账 --> 购物功能、支付功能
3.可以提现，手续费5% --> 提现功能
4.支持多账户登录 --> 登录功能
5.支持账户间转账 --> 转账功能
6.记录每月日常消费流水 --> 记录流水功能
7.提供还款接口 --> 还款功能
8.ATM记录操作日志 --> 记录日志功能
9.提供管理接口，包括添加账户、用户额度，冻结账户等 -->管理员功能
10.用户认证用装饰器 --> 登录认证装饰器

2 功能

1.注册功能
2.登录功能
3.查看余额
4.提现功能
5.还款功能
6.转账功能
7.查看流水
8.购物功能
9.查看购物车
10.管理员功能

3 目录结构

ATM
-bin
  -start.py
-conf
  -settings.py
-db
  -user_data
    -ccc.py
    ...
  -db_handler.py
-interface
  -bank.py
  -user.py
  -shopping.py
-lib
  -common.py
-log
  -atm.log
-view
  -src.py

4 思路及步骤

4.1 环境bin/start.py

程序的入口

4.1.1 将解释器添加到环境变量中

import os
import sys
sys.path.append(os.path.dirname(os.path.dirname(__file__)))

4.1.2 启动主程序

from view import src
if __name__ == '__main__':
    src.run()
# 此时应该在view/src.py中定义主启动函数run()

4.2 配置conf/settings.py

存放配置信息

4.2.1 获取项目根目录路径

import os
BASE_PATH = os.path.dirname(os.path.dirname(__file__))

4.2.2 获取user_data目录路径

import os
USER_DATA_PATH = os.path.join(BASE_PATH, 'db', 'user_data')

4.2.3 获取log目录路径

import os
USER_LOG = os.path.join(BASE_PATH, 'log')

4.3 数据处理db/db_handle.py

数据处理层专门用来处理数据

4.3.1 保存用户信息save

import os
import json
from conf import settings

def save(username, user_dic):
    user_path = os.path.join(settings.USER_DATA_PATH, f'{username}.json')
    with open(user_path, 'wt', encoding='utf-8') as f:
        json.dump(user_dic, f, ensure_ascii=False)
        f.flush()  # 强制刷入硬盘

4.3.2 返回用户信息select

import os
import json
from conf import settings

def select(username):
    user_path = os.path.join(settings.USER_DATA_PATH, f'{username}.json')
    if os.path.exists(user_path):
        with open(user_path, 'rt', encoding='utf-8') as f:
        	user_dic = json.load(f)
        return user_dic

4.4 视图层view/src

用户视图层

4.4.1 主启动函数run()

func_dic = {
    "1": register,
    "2": login,
    "3": check_balance,
    "4": withdrawal,
    "5": repay,
    "6": transfer,
    "7": check_flow,
    "8": shop,
    "9": check_shopping_cart,
    "10": admin,
    "11": logout,
}

def run():
    while True:
        print(
        =======================
             1.注册
			2.登录
			3.查看余额
			4.提现
			5.还款
			6.转账
			7.查看流水
			8.购物
			9.查看购物车
			10.管理员
			11.注销
        =======================
        )
        choice = input("请输入功能编号(按q退出程序):").strip()
        if choice == "q":break
        if choice not in func_dic:
            print("请输入正确的编号!")
            continue
        func_dic.get(choice)()  # 输入1相当于func_dic.get('1')()-->register()

4.4.2 功能实现

4.4.2.1 注册功能register()

from interface import user

user_login = {
    'is_login': False,
    'username': None
}

def register():
    while True:
        if not user_login['is_login']:
            username = input("请输入注册用户名(按q退出):").strip()
            if username == "q":break
            password = input("请输入密码:").strip()
            re_password = input("请再次输入密码:").strip()
            if password == re_password:
                msg = user.register_interface(username, password)
                print(msg)
                break
            else:
                print("两次密码不一致")
		else:
            print("登陆状态不可再注册")
            break

4.4.2.2 登录功能login()

from interface import user

user_login = {
    'is_login': False,
    'username': None
}

def login():
    count = 0
    while True:
        if not user_login['is_login']:
            username = input("请输入登录名(按q退出):").strip()
            if username == "q":break
            user_info = user.get_user_info_interface(username)
            if user_info:
                if not user_info['locked']:
                    if count < 3:
                        password = input("请输入登录密码:").strip()
                        if password == user_info['password']:
                            user_login['is_login'] = True
                            user_login['username'] = username
                            print("登陆成功")
                            break
                        else:
                            count += 1
                            print("密码输入错误")
                    else:
                        user.locked_interface(username)
                        print("该用户密码错误输入次数过多,已被锁定")
                        break
                else:
                    msg = user.unlocked_interface(username)
                    print(msg)
            else:
                print("该用户不存在")
        else:
            print("登录状态下不可再次登录")
            break
            
            
1、判断用户登录状态，非登录状态继续执行代码，否则告诉用户登录情况下不能再次登录
2、用户输入登录名，判断输入是否为q，非q继续执行代码，否则退出
3、判断用户名是否存在，不存在继续执行代码，存在则告诉用户该用户名不存在
4、判断该用户是否被锁，没有则继续执行，被锁则调用解锁接口 等待解锁
5、判断密码输入次数，小于三次则继续执行，否则调用锁定接口 锁定账号
6、用户输入密码，判断密码是否正确，正确则登录，错误则输入次数加1

4.4.2.3 查看余额功能check_balance()

# 首先使用登录认证装饰器判断登陆状态
from lib import common
from interface import bank

user_login = {
    'is_login': False,
    'username': None
}

@common.check_login
def check_balance():
    msg = bank.check_balance_interface(user_login['username'])
    print("%s的余额是%s元" % (user_login['username'], msg))

4.4.2.4 提现功能withdrawal()

from lib import common
from interface import bank

user_login = {
    'is_login': False,
    'username': None
}

@common.check_login
def withdrawal():
    while True:
        print("""
=============================
        0) 退出
        1) 查看余额
        2) 提现
=============================
        """)
        choice = input("请输入指令:").strip()
        if choice == "0":break
        elif choice == "1":
            msg = bank.check_balance_interface(user_login['username'])
            print(msg)
        elif choice == "2":
            money = input("请输入提现金额:").strip()
            if money.isdigit:
                money = int(money)
                msg = bank.withdrawal_interface(user_login['username'], money)
                print(msg)
                break
            else:
                print("请输入正确的金额")
        else:
            print("请输入正确的提现指令")
            
1、装饰器判断登录状态，未登录则执行登陆程序，登陆则继续
2、用户选择功能，0->退出 1->查看余额 2->提现(循环功能)
3、0直接退出 1调用余额接口
4、2让用户输入提现金额，判断是否为数字，不为数字提示重新输入，为数字继续
5、调用提现接口，打印返回值

4.4.2.5 还款功能repay()

from lib import common
from interface import bank

user_login = {
    'is_login': False,
    'username': None
}

@common.check_login
def repay():
    while True:
        choice = input("""
=============================
        0) 退出
        1) 查询欠款金额
        2) 还款
=============================
        请输入选项:""").strip()
        if choice == "0":break
        elif choice == "1":
            msg = bank.check_debt_interface(user_login['username'])
            print(msg)
        elif choice == "2":
            money = input("请输入还款金额:").strip()
            if money.isdigit():
                money = int(money)
                msg = bank.repay_interface(user_login['username'], money)
                print(msg)
                break
            else:
                print("请输入正确的金额")
        else:
            print("请输入正确的选项")
            
1、调用登录认证装饰器
2、用户选择还款功能 0退出 1查询欠款金额 2还款
3、1调用查询欠款接口查询
4、2用户输入还款金额，判断是否为数字
5、调用还款接口

4.4.2.6 转账功能transfer()

from lib import common
from interface import bank

user_login = {
    'is_login': False,
    'username': None
}

@common.check_login
def transfer():
    while True:
        print("""
=============================
        0) 退出
        1) 查看余额
        2) 转账
=============================
        """)
        choice = input("请输入功能选项:").strip()
        if choice == "0":break
        elif choice == "1":
            msg = bank.check_balance_interface(user_login['username'])
            print(msg)
        elif choice == "2":
            to_username = input("请输入被转帐用户名:").strip()
            if to_username == user_login['username']:
                print("不能转账给自己")
                continue
            if user.get_user_info_interface(to_username):
                money = input("请输入转账金额:").strip()
                if money.isdigit():
                    money = int(money)
                    msg = bank.transfer_interface(user_login['username'], to_username, money)
                    print(msg)
                    break
                else:
                    print("请输入正确的金额数")
            else:
                print("被转账用户不存在")
        else:
            print("请输入正确的选项")
            
1、调用登录认证装饰器
2、用户选择功能 0退出 1调用查看余额接口查看余额 2转账
3、2输入被转账用户名，判断是否为自己
4、调用提取信息接口判断用户是否存在，存在则继续，否则提示用户不存在
5、用户输入金额，判断是否为数字，是继续，否则提示输入正确的金额
6、调用转账接口

4.4.2.7 查看流水功能check_flow()

from lib import common
from interface import bank

user_login = {
    'is_login': False,
    'username': None
}

@common.check_login
def check_flow():
    flow_list = bank.check_flow_interface(user_login['username'])
    for line in flow_list:
        print(line)

4.4.2.8 购物功能shop()

from lib import common
from interface import shopping

user_login = {
    'is_login': False,
    'username': None
}

@common.check_login
def shop():
    good_list = [
        ['tesla', 10000],
        ['mac', 8000],
        ['thinkpad', 6000],
        ['火锅', 580],
        ['烧烤', 320],
        ['串串', 280],
        ['包子', 8],
        ['矿泉水',5]
    ]
    good_cart = {}
    balance = bank.check_balance_interface(user_login['username'])
    amount_spend = 0
    while True:
        for i,line in enumerate(good_list):
            print("%s 商品名:%s 单价:%s" % (i, line[0], line[1]))
        choice = input("请输入商品序号(按q退出并购买):").strip()
        if choice.isdigit():
            choice = int(choice)
            if 0 <= choice <= len(good_list)-1:
                good_name = good_list[choice][0]
                good_price = good_list[choice][1]
                if balance >= good_price:
                    if good_name in good_cart:
                        good_cart[good_name]['count'] += 1
                    else:
                        good_cart[good_name] = {'price': good_price, 'count': 1}
                    amount_spend += good_price
                    balance -= good_price
                    print("%s已被加入购物车" % good_name)
                    print("目前购物车有:%s" % good_list)
                    print("余额:%s元" % balance)
                else:
                    print("余额不足,剩余%s" % balance)
            else:
                print("该商品尚未上架")
        elif choice == "q":
            while True:
                choice = input("请输入 y)结算 或 n)退出:").strip()
                if choice == "y":
                    if amount_spend > 0:
                        password = input("请输入密码:").strip()
                        file_password = user.get_user_info_interface(user_login['username'])['password']
                        if password == file_password:
                            msg = shopping.buy_goods_interface(user_login['username'], good_cart, amount_spend)
                            print(msg)
                            break
                        else:
                            print("密码错误")
                    else:
                        print("没买东西结什么账啊")
                        break
                elif choice == "n":
                    print("白加这么多东西在购物车")
                    break
                else:
                    print("请输入正确的选项")
            break
        else:
            print("请输入数字")

4.4.2.9 查看购物车功能check_shopping_cart()

from lib import common
from interface import shopping

user_login = {
    'is_login': False,
    'username': None
}

@common.check_login
def check_shopping_cart():
    msg = shopping.check_shopping_cart_interface(user_login['username'])
    print(msg)

4.4.2.10 管理员功能admin()

def admin():
    pass

4.4.2.11 注销功能logout()

from lib import common

user_login = {
    'is_login': False,
    'username': None
}

@common.check_login
def logout():
    if user_login['is_login']:
        user_login['is_login'] = False
        print("注销成功")
    else:
        print("请先登录")

4.5 公共lib/common

存放公共登陆方法，登录认证装饰器及日志模块

4.5.1 登陆认证装饰器

from view import src

# 登录认证装饰器
def check_login(func):
    def inner(*args, **kwargs):
        if src.user_login['is_login']:
            res = func(*args, **kwargs)
            return res
        else:
            print("暂未登录，请先登录后在使用本功能")
            src.login()
    return inner

4.5.2 日志

import os
import logging.config
from conf import settings

# 日志模块
standard_format = '[%(asctime)s][%(threadName)s:%(thread)d][task_id:%(name)s][%(filename)s:%(lineno)d]' 
                  '[%(levelname)s][%(message)s]'  # 其中name为getlogger指定的名字

simple_format = '[%(levelname)s][%(asctime)s][%(filename)s:%(lineno)d]%(message)s'

id_simple_format = '[%(levelname)s][%(asctime)s] %(message)s'

# 定义日志输出格式 结束

logfile_dir = settings.USER_LOG  # log文件的目录

logfile_name = 'atm.log'  # log文件名

# 如果不存在定义的日志目录就创建一个
if not os.path.isdir(logfile_dir):
    os.mkdir(logfile_dir)

# log文件的全路径
logfile_path = os.path.join(logfile_dir, logfile_name)

# log配置字典
LOGGING_DIC = {
    'version': 1,
    'disable_existing_loggers': False,
    'formatters': {
        'standard': {
            'format': standard_format
        },
        'simple': {
            'format': simple_format
        },
    },
    'filters': {},
    'handlers': {
        # 打印到终端的日志
        'console': {
            'level': 'DEBUG',
            'class': 'logging.StreamHandler',  # 打印到屏幕
            'formatter': 'simple'
        },
        # 打印到文件的日志,收集info及以上的日志
        'default': {
            'level': 'DEBUG',
            'class': 'logging.handlers.RotatingFileHandler',  # 保存到文件
            'formatter': 'standard',
            'filename': logfile_path,  # 日志文件
            'maxBytes': 1024*1024*5,  # 日志大小 5M
            'backupCount': 5,
            'encoding': 'utf-8',  # 日志文件的编码，再也不用担心中文log乱码了
        },
    },
    'loggers': {
        # logging.getLogger(__name__)拿到的logger配置
        '': {
            'handlers': ['default', 'console'],  # 这里把上面定义的两个handler都加上，即log数据既写入文件又打印到屏幕
            'level': 'DEBUG',
            'propagate': True,  # 向上（更高level的logger）传递
        },
    },
}


def make_log(name):
    logging.config.dictConfig(LOGGING_DIC)  # 导入上面定义的logging配置
    logger = logging.getLogger(name)  # 生成一个log实例

    return logger

4.6 接口interface

4.6.1 银行相关bank.py

4.6.1.1 查看余额接口

from db import db_handler
from lib import common
logger_bank = common.make_log('bank')

# 查看余额接口
def check_balance_interface(username):
    user_info = db_handler.select(username)
    res = user_info['balance']
    # 记录日志
    logger_bank.info("%s查看了余额" % username)
    return res

4.6.1.2 提现接口

from db import db_handler
from lib import common
logger_bank = common.make_log('bank')

# 提现接口
def  withdrawal_interface(username, money):
    user_info = db_handler.select(username)
    if user_info['balance'] >= money*1.05:
        user_info['balance'] -= money*1.05
        # 查看流水
        user_info['flow'].append("%s成功提现%s元,扣除手续费%s元" % (username, money, money*0.05))
        db_handler.save(username, user_info)
        # 记录日志
        logger_bank.info("%s成功提现%s元,扣除手续费%s元" % (username, money, money*0.05))
        return "%s成功提现%s元,扣除手续费%s元" % (username, money, money*0.05)
    else:
        return "%s账户余额不足" % username

4.6.1.3 查询欠款接口

from db import db_handler
from lib import common
logger_bank = common.make_log('bank')

# 查询欠款接口
def check_debt_interface(username):
    user_info = db_handler.select(username)
    res = user_info['credit'] - user_info['balance']
    if res > 0:
        # 记录日志
        logger_bank.info("%s查看了欠款" % username)
    	return "%s当前欠款%s元" % (username, res)
    else:
        return "%s当前并未欠款" % username

4.6.1.4 还款接口

from db import db_handler
from lib import common
logger_bank = common.make_log('bank')

# 还款接口
def repay_interface(username, money):
    user_info = db_handler.select(username)
    res = user_info['credit'] - user_info['balance']
    if res >= money:
        user_info['balance'] += money
        # 记录流水
        user_info['flow'].append("%s成功还款%s元" % (username, money))
        db_handler.save(username, user_info)
        # 记录日志
        logger_bank.info("%s成功还款%s元" % (username, money))
        return "%s成功还款%s元" % (username, money)
    else:
        return "还款金额超过欠款额度"

4.6.1.5 转账接口

from db import db_handler
from lib import common
logger_bank = common.make_log('bank')

# 转账功能接口
def transfer_interface(username, to_username, money):
    user_info = db_handler.select(username)
    to_user_info = db_handler.select(to_username)
    if user_info['balance'] >= money:
        user_info['balance'] -= money
        to_user_info['balance'] += money
        # 记录流水
        user_info['flow'].append("您向%s转账%s元" % (to_username, money))
        to_user_info['flow'].append("%s向您转账%s元" % (username, money))
        
        db_handler.save(username, user_info)
        db_handler.save(to_username, to_user_info)
        
        # 记录日志
        logger_bank.info("%s向%s转账%s元" % (username, to_username, money))
        return "%s向%s转账%s元" % (username, to_username, money)
    else:
        return "%s账户余额不足" % username

4.6.1.6 查看流水接口

from db import db_handler
from lib import common
logger_bank = common.make_log('bank')

# 查看流水接口
def check_flow_interface(username):
    user_info = db_handler.select(username)
    # 查看流水
    logger_bank.info("%s查看了流水" % username)
    return user_info['flow']

4.6.1.7 消费接口

from db import db_handle
from lib import common
logger_bank = common.make_log('bank')

def consume_interface(username, money):
    user_info = db_handler.select(username)
    user_info['balance'] -= money
    # 记录流水
    user_info['flow'].append("%s消费了%s元" % (username, money))
    db_handler.save(username, user_info)
	# 记录日志
    logger_bank.info("%s消费了%s元" % (username, money))

4.6.2 购物相关shopping.py

4.6.2.1 购买商品接口

from db import db_handler
from interface import bank
from lib import common
logger_shopping = common.make_log('shopping')

def buy_goods_interface(username, good_cart, money):
    user_info = db_handler.select(username)
    bank.consume_interface(username, money)
    user_info['shopping_cart'] = good_cart
    db_handler.save(username, user_info)
    # 记录日志
    logger_shopping.info("%s成功购买%s" % (username, good_cart))
    return "%s成功购买%s" % (username, good_cart)

4.6.2.2 查看购物车接口

from db import db_handler
from lib import common
logger_shopping = common.make_log('shopping')

def check_shopping_cart_interface(username):
    user_info = db_handler.select(username)
    res = user_info['shopping_cart']
    # 记录日志
    logger_shopping.info("%s查看了购物车" % username)
    return res

4.6.3 用户相关user.py

4.6.3.1 注册接口

from db import db_handler
from lib import common
logger_user = common.make_log('user')

# 注册接口
def register_interface(username, password, balance=15000):
    user = db_handler.select(username)  # 取到user_info = user_dic
    if user:  # 判断该用户名是否存在
        return "该用户名已存在"
    else:
        user_dic = {
            'username': username,
            'password': password,
            'credit': balance,
            'balance': balance,
            'locked': False,
            'time': 0,
            'flow': [],
            'shopping_cart': {}
        }
        db_handler.save(username, user_dic)
        # 记录日志
        logger_user.info("%s注册成功" % username)
        return "%s注册成功" % username

4.6.3.2 提取用户信息接口

from db import db_handler

# 提取用户信息
def get_user_info_interface(username):
    user_info = db_handler.select(username)
    return user_info

4.6.3.3 锁定用户接口

import time
from db import db_handler
from lib import common
logger_user = common.make_log('user')

# 锁定用户
def locked_interface(username):
    user_info = db_handler.select(username)
    user_info['locked'] = True
    user_info['time'] = time.time()+20
    db_handler.save(username, user_info)
    # 记录日志
    logger_user.info("%s被锁定了" % username)

4.6.3.4 解锁用户接口

import time
from db import db_handler
from lib import common
logger_user = common.make_log('user')

# 解锁用户接口
def unlocked_interface(username):
    user_info = db_handler.select(username)
    last_time = user_info['time'] - time.time()
    if user_info['time'] <= time.time():
        user_info['locked'] = False
        db_handler.save(username, user_info)
    	# 记录日志
    	logger_user.info("%s用户已解锁" % username)
        return "%s用户已经解锁,请重新登陆" % username
    else:
        return "%s用户已被锁,剩余锁定时间%s" % (username, last_time)