各文件内容

前言

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

详细文件内容

启动文件

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

bin/starts.py

配置文件

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）传递

        },

    },

}

conf/logging_dict.py

settings.py配置文件

import os

import sys

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

sys.path.append(BASE_DIR)

conf/settings.py

入口文件

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 = {

    '': register,

    '': login,

    '': logout,

    '': check_balance,

    '': check_flow,

    '': repay,

    '': withdraw,

    '': transfer,

    '': shop,

    '': check_shop_cart,

    '': lock_user,

    '': unlock_user,

    '': change_balance,

}

current_user = {

    'username': '',

    'pwd': None,

    'balance': 0,

    'lock': False,

    'flow': [],

    'shop_cart': {},

    'is_admin': False

}

shopping_list = {

    "": ['广东凤爪', 10],

    "": ['面疙瘩', 15],

    "": ['mac', 12888],

    "": ['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("您的输入有误，请重新输入")

core/src.py

数据处理层

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/db_handler.py

用户数据信息

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/swb.json用户信息文件1

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}

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

接口层

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/bank_interface.py

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/shop_interface.py

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

    # 这些管理员方法用装饰器判断他是不是管理员

interface/user_interface.py

通用方法

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测试数据啦~')

lib/common.py

日志记录

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用户注销登录。]

log/ATM_Shop_Cart.log

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

最新文章

  1. js中使用new Date(str)创建时间对象不兼容firefox和ie的解决方式
  2. Queue 的用法
  3. 通过docker-machine和etcd部署docker swarm集群
  4. java语言的认识
  5. 重构前的程序：通过rsync命令抓取日志文件
  6. 树状jquery导航条
  7. ajax动态加载的图标
  8. Vladik and cards
  9. Python configparser 读取指定节点内容失败
  10. ArcGIS API for JavaScript 入门教程[1] 渊源
  11. MongoDB 文章目录
  12. rsa加密算法及js的JSEncrypt实现前端加密
  13. mysql连接失败HikariPool错误
  14. 第三篇、Python函数
  15. 初始MapReduce
  16. os.walk函数
  17. OC与AS3的观察者模式比较以及外部回调
  18. Maven项目的拆分与聚合
  19. zend opcache的最佳设置
  20. Raspberry install wine

热门文章

  1. PHP阿里大鱼短信验证
  2. 移动端数据爬取和Scrapy框架
  3. spark 源码分析之二 -- SparkContext 的初始化过程
  4. 06、MySQL—列类型
  5. C# RESTful API 访问辅助类
  6. Thread中的start()方法和自己定义的run()方法有什么区别
  7. Java 8 新特性-Stream更优雅的处理集合入门
  8. HashMap源码分析(一):JDK源码分析系列
  9. Git使用小技巧之Stash命令藏储零乱分支
  10. C++学习书籍推荐《Effective C++ 第三版（英文)》下载