conf存放配置文件
import os
BASE_PATH = os.path.dirname(os.path.dirname(__file__))
# 拼接db目录路径
DB_PATH = os.path.join(BASE_PATH, 'db')
"""
logging配置
"""
# 定义三种日志输出格式 开始
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 = os.path.join(BASE_PATH, 'log') # log文件的目录
logfile_name = 'atm_shop_log.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)传递
},
},
}
core核心业务
import os
BASE_PATH = os.path.dirname(os.path.dirname(__file__))
# 拼接db目录路径
DB_PATH = os.path.join(BASE_PATH, 'db')
"""
logging配置
"""
# 定义三种日志输出格式 开始
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 = os.path.join(BASE_PATH, 'log') # log文件的目录
logfile_name = 'atm_shop_log.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)传递
},
},
}
db数据处理
import os
import json
from conf import settings
# 保存数据
def save(user_dic):
# 拼接用户保存文件
user_path = f'{settings.DB_PATH}/{user_dic["user"]}.json'
# 把用户信息写入文件中
with open(user_path, 'w', encoding='utf-8') as f:
json.dump(user_dic, f, ensure_ascii=False)
f.flush()
# 查看用户数据
def select(user):
# base_path = os.path.dirname(os.path.dirname(__file__))
# db_path = os.path.join(base_path, 'db')
user_path = f'{settings.DB_PATH}/{user}.json'
# 判断用户文件是否存在
if os.path.exists(user_path):
# 若存在, 读取用户信息
with open(user_path, 'r', encoding='utf-8') as f:
user_dic = json.load(f)
return user_dic
interface接口逻辑处理
import os
import json
from conf import settings
# 保存数据
def save(user_dic):
# 拼接用户保存文件
user_path = f'{settings.DB_PATH}/{user_dic["user"]}.json'
# 把用户信息写入文件中
with open(user_path, 'w', encoding='utf-8') as f:
json.dump(user_dic, f, ensure_ascii=False)
f.flush()
# 查看用户数据
def select(user):
# base_path = os.path.dirname(os.path.dirname(__file__))
# db_path = os.path.join(base_path, 'db')
user_path = f'{settings.DB_PATH}/{user}.json'
# 判断用户文件是否存在
if os.path.exists(user_path):
# 若存在, 读取用户信息
with open(user_path, 'r', encoding='utf-8') as f:
user_dic = json.load(f)
return user_dic
管理员接口
from db import db_handler
from lib import common
admin_logger = common.get_logger('admin')
# 冻结用户接口
def lock_interface(user):
user_dic = db_handler.select(user)
user_dic['lock'] = True
db_handler.save(user_dic)
return f'{user}用户冻结成功!'
def unlock_interface(user):
user_dic = db_handler.select(user)
user_dic['lock'] = False
db_handler.save(user_dic)
return f'{user}用户解冻成功!'
# 修改额度接口
def change_limit_interface(user, limit):
user_dic = db_handler.select(user)
user_dic['balance'] = limit
db_handler.save(user_dic)
return f'修改用户[{user}]额度成功!'
银行接口
from db import db_handler
from lib import common
bank_logger = common.get_logger('bank')
# 提现接口
def withdraw_interface(user, money):
user_dic = db_handler.select(user)
if user_dic['balance'] >= money * 1.05:
user_dic['balance'] -= money * 1.05
msg = f'用户[{user}] 提现[{money}]元成功!'
user_dic['flow'].append(msg)
db_handler.save(user_dic)
return True, msg
return False, '余额不足!'
# 还款接口
def repay_interface(user, money):
user_dic = db_handler.select(user)
user_dic['balance'] += money
msg = f'用户{user}, 还款{money}元成功!'
user_dic['flow'].append(msg)
db_handler.save(user_dic)
return True, msg
# 转账接口
def transfer_interface(to_user, from_user, money):
to_user_dic = db_handler.select(to_user)
from_user_dic = db_handler.select(from_user)
if from_user_dic['balance'] >= money:
# 加减钱操作
from_user_dic['balance'] -= money
to_user_dic['balance'] += money
# 拼接流水信息
to_user_flow = f'{to_user}用户接收到用户{from_user}转账{money}元'
from_user_flow = f'{from_user}用户给到用户{to_user}转账{money}元'
# 记录流水
from_user_dic['flow'].append(from_user_flow)
to_user_dic['flow'].append(to_user_flow)
# 保存用户信息
db_handler.save(from_user_dic)
db_handler.save(to_user_dic)
return True, from_user_flow
return False, '余额不足,请充值!'
# 查看流水接口
def check_flow_interface(user):
user_dic = db_handler.select(user)
return user_dic['flow']