python购物车升级版

各文件内容

前言

功能架构等请参考前一篇博客，此篇博客为进阶版的存代码展示。

详细文件内容

启动文件

starts.py启动文件

import os
import sys

BASE_DIR = os.path.dirname(os.path.dirname(__file__))
sys.path.append(BASE_DIR)


from core import src
if __name__ == '__main__':
    src.run()

配置文件

logging_dict.py日志配置字典

import os
# ************************************   日志配置字典   **************************************************
'''
    --------------------------------------------
    ------------* 需要自定义的配置 *-------------
    --------------------------------------------
'''
# 定义三种日志输出格式 开始
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'
# 定义日志输出格式 结束


# 定义日志文件的存放目录与文件名（一般视情况选择下面的两个即可）
logfile_dir = os.path.dirname(os.path.dirname(__file__))  # log文件的目录 （执行文件starts.py 在bin目录下）
# logfile_dir = os.path.abspath(__file__)  # log文件的目录 （执行文件starts.py 在项目根目录下）
# 拼上 log 文件夹
logfile_dir = os.path.join(logfile_dir, 'log')
logfile_name = 'ATM_Shop_Cart.log'  # log文件名

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

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

# 像日志文件的最大限度、个数限制，日志过滤等级等也可以在下面的对应地方限制

'''
    --------------------------------------------
    --------------* log配置字典 *---------------
    --------------------------------------------
'''
# 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，超过5M就会换一个日志文件继续记录
            'backupCount': 5,  # 日志文件的数量上限 5个，超出会把最早的删除掉再新建记录日志
            'encoding': 'utf-8',  # 日志文件的编码，再也不用担心中文log乱码了
        },
    },
    'loggers': {
        # logging.getLogger(__name__)拿到的logger配置
        '': {
            'handlers': ['default', 'console'],  # 这里把上面定义的两个handler都加上，即log数据既写入文件又打印到屏幕
            # 'handlers': ['default'],  # ------------- 正式上线的时候改成这个，取消控制台打印，节省资源  -------------
            'level': 'DEBUG',
            'propagate': True,  # 向上（更高level的logger）传递
        },
    },
}

settings.py配置文件

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

入口文件

src.py入口文件

from interface import user_interface
from lib import common
from interface import bank_interface
from interface import shop_interface


def register():
    while True:
        username = input("请输入用户名>>>:").strip()
        if username == 'q':
            return False
        if user_interface.check_user_exits(username):
            print("该用户已存在，请换一个试试！")
            continue
        pwd = input("请输入密码>>>:").strip()
        if pwd == 'q':
            return False
        repwd = input("请再次输入密码>>>:").strip()
        if pwd == 'q':
            return False
        if repwd != pwd:
            print("两次密码不一致，请您重新输入！")
            continue
        flag = user_interface.register(username, pwd)
        if not flag:
            print("注册失败，请重试！")
            continue
        else:
            print(f"{username}注册成功~")
            return True


def login():
    while True:
        username = input("请输入用户名>>>:").strip()
        if username == 'q':
            return False
        if not user_interface.check_user_exits(username):
            print("该用户不存在，请检查输入！")
            continue
        pwd = input("请输入密码>>>:").strip()
        if pwd == 'q':
            return False
        flag = user_interface.login(username, pwd)
        if not flag:
            print("您的密码有误或账户已被冻结，请重新登录！")
            continue
        else:
            print(f"登陆成功，{username}您好~")
            return True


@common.login_auth
def logout():
    user_interface.logout()
    print("欢迎下次登录，祝您生活愉快~")


@common.login_auth
def check_balance():
    username = current_user.get("username")
    balance = user_interface.check_balance(username)
    print(f"{username}您好，您的账户余额为{balance}。")


@common.login_auth
def check_flow():
    username = current_user.get('username')
    flow = user_interface.check_flow(username)
    if flow:
        print(f"{username}您好，您的流水如下：")
        for line in flow:
            print(line)
    else:
        print(f"{username}您好，您的账户暂无流水记录~")


# ---------------------
# 银行接口
@common.login_auth
def repay():
    username = current_user.get('username')
    while True:
        money = input('请输入您的还款金额>>>:').strip()
        if money == 'q':
            return False
        elif not money.isdigit():
            print("请正确输入金额")
            continue
        else:
            money = int(money)
            bank_interface.repay(username, money)
            print("还款成功！")
            return True


@common.login_auth
def withdraw():
    username = current_user.get('username')
    while True:
        money = input('请输入您的提现金额>>>:').strip()
        if money == 'q':
            return False
        elif not money.isdigit():
            print("请正确输入金额")
            continue
        else:
            money = int(money)
            flag = bank_interface.withdraw(username, money)
            if flag:
                print("提现成功！")
                return True
            else:
                print("您的余额不足提现金额（提现需包含0.05%的手续费）。")


@common.login_auth
def transfer():
    username = current_user.get('username')
    while True:
        target_user = input("请输入您的转账目标用户>>>:").strip()
        if user_interface.check_user_exits(target_user):
            money = input('请输入您的转账金额>>>:').strip()
            if money == 'q':
                return False
            elif not money.isdigit():
                print("请正确输入金额")
                continue
            else:
                money = int(money)
                flag = bank_interface.transfer(username, target_user, money)
                if not flag:
                    print("转账失败，请确认您的余额是否充足！")
                    continue
                print("转账成功！")
                return True
        print(f"{target_user}用户不存在，请您重新输入！")


# ---------------------
# 购物功能
@common.login_auth
def shop():
    username = current_user['username']
    while True:
        print("------------- 商品列表 -------------")
        for key in shopping_list:
            print(f"{key}.{shopping_list.get(key)[0]} {shopping_list.get(key)[1]}")
        choice = input("请输入商品编号>>>:").strip()
        if choice == 'q':
            return False
        elif choice in shopping_list:
            # 开始购物
            is_buy = input("请选择购买或加入购物车(y购买/n加入购物车)(q退出)").strip()
            if is_buy == 'q':
                return False
            good = shopping_list[choice][0]
            price = shopping_list[choice][1]

            if is_buy == 'y':
                flag = shop_interface.pay_good(username, good, price)
                if not flag:
                    print(f"您的余额不足以购买该商品，请换一个试试")
                else:
                    print(f"{good}商品购买成功！")
            elif is_buy == 'n':
                # 加入购物车
                flag = shop_interface.put_shop_cart(username, good, price)
                if not flag:
                    print(f"{good}该商品已在购物车中啦~")
                else:
                    print(f"{good}成功放入购物车")
                pass
            else:
                print("请正确输入！")
                continue
        else:
            print("请输入正确的商品编号!")
            continue


@common.login_auth
def check_shop_cart():
    username = current_user['username']
    shop_cart = shop_interface.get_shop_cart(username)
    if shop_cart:
        print(f"{username}您好，您的购物车商品列表如下")
        for shop, price in shop_cart.items():
            print(shop, price)
    else:
        print(f"您的购物车里还没有东西哦~")


# ---------------------
# 管理员功能
@common.login_auth
@common.admin_auth
def lock_user():
    username = current_user['username']
    while True:
        target_user = input("请输入您要冻结的目标用户>>>:").strip()
        if user_interface.check_user_exits(target_user):
            user_interface.lock_user(username, target_user)
            print(f"{target_user}用户已冻结。")
            return True
        print(f"{target_user}用户不存在，请您重新输入！")


@common.login_auth
@common.admin_auth
def unlock_user():
    username = current_user['username']
    while True:
        target_user = input("请输入您要解冻的目标用户>>>:").strip()
        if user_interface.check_user_exits(target_user):
            user_interface.unlock_user(username, target_user)
            print(f"{target_user}用户已成功解冻。")
            return True
        print(f"{target_user}用户不存在，请您重新输入！")


@common.login_auth
@common.admin_auth
def change_balance():
    username = current_user['username']
    while True:
        target_user = input("请输入您要更改的目标用户>>>:").strip()
        if user_interface.check_user_exits(target_user):
            money = input('请输入金额>>>:').strip()
            if money == 'q':
                return False
            elif not money.isdigit():
                print("请正确输入金额")
                continue
            else:
                money = int(money)
                flag = user_interface.change_balance(username, target_user, money)
                if not flag:
                    print("不能设置自己的余额，请换个用户试试。")
                    continue
                print(f"{target_user}用户余额成功修改为{money}元。")
                return True
        print(f"{target_user}用户不存在，请您重新输入！")


func_list = {
    '1': register,
    '2': login,
    '3': logout,
    '4': check_balance,
    '5': check_flow,
    '6': repay,
    '7': withdraw,
    '8': transfer,
    '9': shop,
    '10': check_shop_cart,
    '11': lock_user,
    '12': unlock_user,
    '13': change_balance,
}


current_user = {
    'username': '',
    'pwd': None,
    'balance': 0,
    'lock': False,
    'flow': [],
    'shop_cart': {},
    'is_admin': False
}


shopping_list = {
    "1": ['广东凤爪', 10],
    "2": ['面疙瘩', 15],
    "3": ['mac', 12888],
    "4": ['tesla', 1222222]
}


def run():
    while True:
        print("""------------------------------------------------------------------
        1.注册        2.登录        3.注销        4.查看余额
        5.查看流水    6.还款        7.提现        8.转账        
        9.购物        10.查看购物车  11.锁定用户   12.解锁用户   
        13.更改用户余额
              11-13为管理员功能哦，只有管理员才能操作！
        ------------------------------------------------------------------""")
        choice = input("请选择功能编号>>>:").strip()
        if choice == 'q':
            print("感谢您的使用，欢迎下次再来~")
            return False
        elif choice in func_list:
            func_list[choice]()
        else:
            print("您的输入有误，请重新输入")

数据处理层

db_handler.py数据处理文件

import os
import json
BASE_DIR = os.path.dirname(os.path.dirname(__file__))
DB_DIR = os.path.join(BASE_DIR, 'db')


def select(username: str):
    user_dir = os.path.join(DB_DIR, f"{username}.json")
    if not os.path.exists(user_dir):
        return False
    else:
        with open(user_dir, mode='r', encoding='utf-8') as f:
            user_dict = json.load(f)
            return user_dict


def save(user_dict: dict):
    user_dir = os.path.join(DB_DIR, f"{user_dict['username']}.json")
    with open(user_dir, mode='w', encoding='utf-8') as f:
        json.dump(user_dict, f, ensure_ascii=False)
    return True

用户数据信息

db/swb.json用户信息文件1

{"username": "swb", "pwd": "59f471f262ab586fb959ded0e2c7b94f", "balance": 10, "lock": false, "flow": ["swb购买了广东凤爪，花费10元，现余额为14990元", "swb购买了面疙瘩，花费15元，现余额为14975元", "管理员tank将您的账户余额重置为10元。"], "shop_cart": {"广东凤爪": 10, "面疙瘩": 15, "mac": 12888, "tesla": 1222222}, "is_admin": false}

db/tank.json用户信息文件2

{"username": "tank", "pwd": "59f471f262ab586fb959ded0e2c7b94f", "balance": 14975, "lock": false, "flow": ["tank购买了广东凤爪，花费10元，现余额为14990元", "tank购买了面疙瘩，花费15元，现余额为14975元"], "shop_cart": {"tesla": 1222222}, "is_admin": true}

接口层

interface/bank_interface.py银行功能接口文件

from db import db_handler
from lib import common
logger = common.get_logger('bank_interface')


def repay(username: str, money: float):
    user_dict = db_handler.select(username)
    user_dict['balance'] += money
    user_dict['flow'].append(f"{username}还款{money}元，现余额为{user_dict['balance']}。")
    db_handler.save(user_dict)
    logger.info(f"{username}还款{money}元，现余额为{user_dict['balance']}。")
    return True


def withdraw(username: str, money: float):
    user_dict = db_handler.select(username)
    if money*1.05 > user_dict['balance']:
        return False
    else:
        user_dict['balance'] -= money*1.05
    user_dict['flow'].append(f"{username}提现{money}元，手续费{money*0.05}元，现余额为{user_dict['balance']}元")
    db_handler.save(user_dict)
    logger.info(f"{username}提现{money}元，手续费{money*0.05}元，现余额为{user_dict['balance']}元")
    return True


def transfer(username: str, target_username: str, money: float):
    user_dict = db_handler.select(username)
    if user_dict['balance'] < money:
        return False
    user_dict['balance'] -= money
    user_dict['flow'].append(f"{username}向{target_username}转账{money}元。")
    db_handler.save(user_dict)

    target_user_dict = db_handler.select(target_username)
    target_user_dict['balance'] += money
    target_user_dict['flow'].append(f"{target_username}收到{username}转账{money}元。")
    db_handler.save(target_user_dict)

    logger.info(f"{username}向{target_username}转账{money}元。")
    return True

interface/shop_interface.py购物车功能接口文件

from db import db_handler
from lib import common
logger = common.get_logger('shop')


def get_shop_cart(username: str):
    user_dict = db_handler.select(username)
    return user_dict['shop_cart']


def put_shop_cart(username: str, good: str, price: float):
    user_dict = db_handler.select(username)
    shop_cart = user_dict['shop_cart']
    if good in shop_cart:
        return False
    else:
        shop_cart[good] = price
    user_dict['shop_cart'] = shop_cart
    db_handler.save(user_dict)
    logger.info(f"商品{good},单价{price}，已加入{username}的购物车~")
    return True


def pay_good(username, good, price):
    user_dict = db_handler.select(username)
    if user_dict['balance'] <= price:
        return False
    else:
        user_dict['balance'] -= price
        user_dict['flow'].append(f"{username}购买了{good}，花费{price}元，现余额为{user_dict['balance']}元")
        db_handler.save(user_dict)
        logger.info(f"{username}购买了{good}，花费{price}元，现余额为{user_dict['balance']}元")
        return True

interface/user_interface.py用户功能接口文件

from db import db_handler
from lib import common
logger = common.get_logger('user_interface')


def check_user_exits(username: str):
    flag = db_handler.select(username)
    if not flag:
        return False
    return True


def register(username: str, pwd: str, balance=15000):
    pwd = common.get_md5(pwd)
    user_dict = {
        'username': username,
        'pwd': pwd,
        'balance': balance,
        'lock': False,
        'flow': [],
        'shop_cart': {},
        'is_admin': False
    }
    db_handler.save(user_dict)
    logger.info(f"{username}用户注册成功，初始余额为{balance}元。")
    return True


def login(username: str, pwd: str):
    pwd = common.get_md5(pwd)
    user_dict = db_handler.select(username)
    if user_dict['pwd'] == pwd and not user_dict['lock']:
        from core import src
        src.current_user = user_dict
        return True
    else:
        return False


def logout():
    from core import src
    logger.info(f"{src.current_user['username']}用户注销登录。")
    src.current_user = {
        'username': None,
        'pwd': None,
        'balance': None,
        'lock': False,
        'flow': [],
        'shop_cart': {},
        'is_admin': False
    }
    return True


def check_balance(username: str):
    user_dict = db_handler.select(username)
    return user_dict['balance']


def check_flow(username: str):
    user_dict = db_handler.select(username)
    return user_dict['flow']


def lock_user(username: str, target_user: str):
    target_user_dict = db_handler.select(target_user)
    target_user_dict['lock'] = True
    db_handler.save(target_user_dict)
    logger.info(f"{username}将{target_user}账户冻结。")
    return True


def unlock_user(username: str, target_user: str):
    target_user_dict = db_handler.select(target_user)
    target_user_dict['lock'] = False
    db_handler.save(target_user_dict)
    logger.info(f"{username}将{target_user}账户解冻。")
    return True


def change_balance(username: str, target_user: str, money: float):
    if username == target_user:
        return False
    target_user_dict = db_handler.select(target_user)
    target_user_dict['balance'] = money
    target_user_dict['flow'].append(f"管理员{username}将您的账户余额重置为{money}元。")
    db_handler.save(target_user_dict)
    logger.info(f"{username}将{target_user}的账户余额改为{money}元。")
    return True
    # 这些管理员方法用装饰器判断他是不是管理员

通用方法

lib/common.py通用方法文件

import logging.config
from conf import logging_dict
import hashlib


# 日志字典使用
def get_logger(type_name: str):
    logging.config.dictConfig(logging_dict.LOGGING_DIC)
    logger = logging.getLogger(type_name)
    return logger


# 登录装饰器
def login_auth(func):
    def inner(*args, **kwargs):
        from core import src
        if src.current_user.get('username'):
            # 登录验证
            res = func(*args, **kwargs)
            return res
        else:
            print("请登录后再操作！")
            src.login()
    return inner


# 管理员验证
def admin_auth(func):
    def inner(*args, **kwargs):
        from core import src
        if src.current_user.get('is_admin'):
            # 判断是不是管理员
            res = func(*args, **kwargs)
            return res
        else:
            print("您不是管理员，不能使用此功能哦~")
    return inner


# md5值获取
def get_md5(string: str) -> str:
    md5 = hashlib.md5()
    md5.update('md5_salt'.encode('utf-8'))
    md5.update(string.encode('utf-8'))
    return md5.hexdigest()


if __name__ == '__main__':
    logger = get_logger('common')
    logger.info('info测试数据啦~')
    logger.debug('debug测试数据啦~')

日志记录

log/ATM_Shop_Cart.log日志记录文件

[2019-07-23 21:55:50,172][MainThread:2688][task_id:user_interface][user_interface.py:25][INFO][swb用户注册成功，初始余额为15000元。]
[2019-07-23 21:56:07,970][MainThread:2688][task_id:shop][shop_interface.py:30][INFO][swb购买了广东凤爪，花费10元，现余额为14990元]
[2019-07-23 21:56:11,611][MainThread:2688][task_id:shop][shop_interface.py:19][INFO][商品面疙瘩,单价15，已加入swb的购物车~]
[2019-07-23 21:56:13,812][MainThread:2688][task_id:shop][shop_interface.py:19][INFO][商品mac,单价12888，已加入swb的购物车~]
[2019-07-23 21:56:54,035][MainThread:764][task_id:shop][shop_interface.py:20][INFO][商品广东凤爪,单价10，已加入swb的购物车~]
[2019-07-23 21:56:55,516][MainThread:764][task_id:shop][shop_interface.py:20][INFO][商品面疙瘩,单价15，已加入swb的购物车~]
[2019-07-23 21:56:56,882][MainThread:764][task_id:shop][shop_interface.py:20][INFO][商品mac,单价12888，已加入swb的购物车~]
[2019-07-23 21:56:58,797][MainThread:764][task_id:shop][shop_interface.py:20][INFO][商品tesla,单价1222222，已加入swb的购物车~]
[2019-07-23 21:57:06,937][MainThread:764][task_id:shop][shop_interface.py:31][INFO][swb购买了广东凤爪，花费10元，现余额为14990元]
[2019-07-23 21:57:09,363][MainThread:764][task_id:shop][shop_interface.py:31][INFO][swb购买了面疙瘩，花费15元，现余额为14985元]
[2019-07-23 21:57:10,666][MainThread:764][task_id:shop][shop_interface.py:31][INFO][swb购买了mac，花费12888元，现余额为2112元]
[2019-07-23 21:59:27,610][MainThread:14372][task_id:shop][shop_interface.py:31][INFO][swb购买了广东凤爪，花费10元，现余额为14990元]
[2019-07-23 22:00:37,898][MainThread:7112][task_id:shop][shop_interface.py:32][INFO][swb购买了广东凤爪，花费10元，现余额为14990元]
[2019-07-23 22:00:39,419][MainThread:7112][task_id:shop][shop_interface.py:32][INFO][swb购买了面疙瘩，花费15元，现余额为14975元]
[2019-07-23 22:01:08,585][MainThread:1056][task_id:user_interface][user_interface.py:25][INFO][tank用户注册成功，初始余额为15000元。]
[2019-07-23 22:01:18,705][MainThread:1056][task_id:shop][shop_interface.py:32][INFO][tank购买了广东凤爪，花费10元，现余额为14990元]
[2019-07-23 22:01:23,370][MainThread:1056][task_id:shop][shop_interface.py:32][INFO][tank购买了面疙瘩，花费15元，现余额为14975元]
[2019-07-23 22:01:29,289][MainThread:1056][task_id:shop][shop_interface.py:20][INFO][商品tesla,单价1222222，已加入tank的购物车~]
[2019-07-23 22:03:05,065][MainThread:1056][task_id:user_interface][user_interface.py:42][INFO][tank用户注销登录。]
[2019-07-23 22:03:18,514][MainThread:1056][task_id:user_interface][user_interface.py:69][INFO][tank将swb账户冻结。]
[2019-07-23 22:03:20,986][MainThread:1056][task_id:user_interface][user_interface.py:42][INFO][tank用户注销登录。]
[2019-07-23 22:03:45,178][MainThread:1056][task_id:user_interface][user_interface.py:77][INFO][tank将swb账户解冻。]
[2019-07-23 22:04:05,704][MainThread:1056][task_id:user_interface][user_interface.py:42][INFO][swb用户注销登录。]
[2019-07-23 22:04:18,321][MainThread:1056][task_id:user_interface][user_interface.py:88][INFO][tank将swb的账户余额改为10元。]
[2019-07-23 22:04:22,425][MainThread:1056][task_id:user_interface][user_interface.py:42][INFO][tank用户注销登录。]

注意： # swb 用户为管理员用户，可以执行管理员操作（swb 以及 tank的密码均为 123），右键starts.py文件运行即可