ATM+购物车系统

ATM目录

start.py

import os,sys
from core import src
base_path=os.path.dirname(os.path.dirname(__file__))
sys.path.append(base_path)
if __name__=='__main__':
    src.run()

配置文件setting.py

import os
base_path=os.path.dirname(os.path.dirname(__file__))
DB=os.path.join(base_path,'db')
BASE_LOG_LOCAL=os.path.join(base_path,'log')
LOG_PATH=r'%s\log\user.log'%base_path
ADMIN_LOG_PATH=r'%s\log\admin.log'%base_path


# 定义三种日志输出格式
standard_format = '[%(asctime)s][%(threadName)s:%(thread)d][task_id:%(name)s][%(filename)s:%(lineno)d]' \
                  '[%(levelname)s][%(message)s]' #其中name为getlogger指定的名字

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

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

# log配置字典
LOGGING_DIC = {
    'version': 1,
    'disable_existing_loggers': False,
    'formatters': {
        'standard': {
            'format': standard_format
        },
        'simple': {
            'format': simple_format
        },
        'id_simple':{
            'format':id_simple_format
        }
    },
    'filters': {},

    'handlers': {
        'stream': {
            'level': 'DEBUG',
            'class': 'logging.StreamHandler',
            'formatter': 'simple'
        },
        'user': {
            'level': 'INFO',
            'class': 'logging.handlers.RotatingFileHandler',
            'formatter': 'standard',
            'filename': LOG_PATH,
            'maxBytes': 1024*1024*5,
            'backupCount': 5,
            'encoding': 'utf-8',
        },
        'admin': {
                    'level': 'DEBUG',
                    'class': 'logging.handlers.RotatingFileHandler',
                    'formatter': 'id_simple',
                    'filename': ADMIN_LOG_PATH,
                    'maxBytes': 1024*1024*5,
                    'backupCount': 5,
                    'encoding': 'utf-8',
                },
    },

    'loggers': {
        '': {
            'handlers': ['user', 'stream','user','admin'],
            'level': 'DEBUG',
            'propagate': False,
        },
    },
}

主体函数src.py、admin.py、user.py

src.py

from core import user,admin
user_dic={
    '1':admin.run,
    '2':user.run,
}
def run():
    while True:
        print('''
1、管理员
2、用户

        ''')
        choice=input('>>>>:').strip()
        if choice=='8':break
        if choice not in user_dic:continue
        user_dic[choice]()

admin.py

from lib import common
data={'name':None}
def login():
    if data['name']:
        print('已经登录')
        return
    while True:
        name=input('账号').strip()
        pwd=input('密码').strip()
        flog,msg=admin_interface.login(name,pwd)
        if flog:
            print(msg)
            data['name']=name
            break
        else:
            print(msg)
#解冻信用卡
@common.login_auth(auth_name='admin')
def thaw():
    while True:
        name=input('请输入要解冻用户的名字或输入q退出').strip()
        if name=='q':break
        user_dic=user_interface.select(name)
        if not user_dic:
            print('用户不存在')
            continue
        if user_dic['card_lick']:
            print('用户未被冻结')
            continue
        admin_interface.card_thaw(user_dic)
        print('信用卡解冻成功')
#冻结信用卡
@common.login_auth(auth_name='admin')
def card_lick():
    while True:
        name = input('请输入要冻结用户的名字或输入q退出').strip()
        if name == 'q': break
        user_dic=user_interface.select(name)
        if not user_dic:
            print('用户不存在')
            continue
        if not user_dic['card_lick']:
            print('信用卡已被冻结')
            continue
        admin_interface.card_lick(user_dic)
        print('冻结成功')
#解除用户锁定
@common.login_auth(auth_name='admin')
def thaw_user():
    while True:
        name = input('请输入要锁定用户的名字或输入q退出').strip()
        if name == 'q': break
        user_dic=user_interface.select(name)
        if not user_dic:
            print('用户不存在')
            continue
        if  user_dic['user_lick']:
            print('用户未被锁定')
            continue
        admin_interface.user_thaw(user_dic)
        print('解除锁定成功')
#锁定用户
@common.login_auth(auth_name='admin')
def user_lick():
    while True:
        name = input('请输入要解锁用户的名字或输入q退出').strip()
        if name == 'q': break
        user_dic=user_interface.select(name)
        if not user_dic:
            print('用户不存在')
            continue
        if not user_dic['card_lick']:
            print('用户已被锁定')
            continue
        admin_interface.user_lick(user_dic)
        print('锁定成功')



user_dic={
    '1':login,
    '2':thaw,
    '3':card_lick,
    '4':thaw_user,
    '5':user_lick

}
def run():
    while True:
        print('''
1、登录
2、解冻信用卡
3、冻结信用卡
4、解除用户锁定
5、锁定用户
6、退出
        ''')
        choice=input('>>>>:').strip()
        if choice=='6':break
        if choice not in user_dic:continue
        user_dic[choice]()

user.py

from interface import bank_interface,user_interface
from lib import common
data={'name':None}
shops=[{'name':'至尊宝','price':5000},
       {'name':'韦小宝','price':200},
       {'name':'宋小宝','price':100},
       {'name':'唐玉小宝','price':1000}]
def register():
    while True:
        name=input('输入姓名或输入q退出：').strip()
        if name=='q':break
        user=user_interface.select(name)
        if not user:
            pwd=input('密码').strip()
            pwd2=input('再次输入密码').strip()
            if pwd==pwd2:
                user_interface.update(name,pwd)
                print('注册成功')
                break
            else:
                print('两次密码不一样')
        else:
            print('用户已存在')
# register()
def login():
    if data['name']:
        print('你已经登陆过')
        return
    count=1
    while True:
        name=input('输入姓名或输入q退出：').strip()
        if name == 'q': break
        if not name:continue
        user_dic=user_interface.select(name)
        if not user_dic:
            print('用户名不存在')
            continue
        if not user_dic['user_lick']:
            print('被锁定,请联系管理员')
            return
        pwd=input('密码').strip()
        if pwd==user_dic['pwd']:
            print('登陆成功')
            data['name']=name
            break
        else:
            if count<3:
                print('密码错误,请重新输入')
                count+=1
                continue
            else:
                print('密码错误次数超过三次，被锁定')
                user_interface.login(user_dic)
                break
# login()
#还款
@common.login_auth(auth_name='user')
def repayment():
    user_dic=user_interface.select(data['name'])
    if user_dic['bal'] >= 15000:print('无欠款')
    else:
        print('你需要还%s'%(15000-user_dic['bal']))
    while True:
        money=input('请输入还款金额或输入q退出：')
        if money=='q':break
        if money.isdigit():
            bank_interface.repayment(user_dic,int(money))
            print('还款%s￥'%money)
        else:
            print('金额必须是数字')
# repayment()
#流水
@common.login_auth(auth_name='user')
def detailed_list():
    user_dic=user_interface.select(data['name'])
    bank_interface.detailed_list(user_dic)

#取现
@common.login_auth(auth_name='user')
def loan():
    user_dic=user_interface.select(data['name'])
    if not user_dic['card_lick']:
        print('你的信用卡账户被冻结,无法取现')
        return
    while True:
        money=input('输入取现金额或输入q退出').strip()
        if money=='q':break
        if not money.isdigit():
            print('金额必须是数字')
            continue
        else:
            flog,msg=bank_interface.loan(user_dic,int(money))
            if flog:
                print(msg)
            else:
                print(msg)
#购物
@common.login_auth(auth_name='user')
def shopping():
    while True:
        user_dic = user_interface.select(data['name'])
        print('''
        编号    商品名      价格
        1       至尊宝      5000
        2       韦小宝       200
        3       宋小宝       100
        4       唐玉小宝    1000
        ''')
        choice=input('输入编号或输入q退出').strip()
        if choice=='q':break
        choice=int(choice)-1
        if choice<=len(shops):
            if not user_dic['card_lick']:
                print('你的信用卡账户被冻结,无法购买')
                return
            flog,msg=bank_interface.shopping(user_dic,shops[choice])
            if flog:
                print(msg)
            else:
                print(msg)

#转账
@common.login_auth(auth_name='user')
def transfer_accounts():
    while True:
        user_dic=user_interface.select(data['name'])
        if not user_dic['card_lick']:
            print('你的信用卡账户被冻结,无法转账')
            return
        to_user_name=input('输入对方的名字或q退出').strip()
        if to_user_name=='q':break
        if to_user_name==data['name']:
            print('不能转给自己')
            continue
        to_user_dic=user_interface.select(to_user_name)
        if not to_user_dic:
            print('该用户不存在')
            continue
        money=input('转账金额').strip()
        if not money.isdigit():
            print('金额必须是数字')
            continue
        flog,msg=bank_interface.transfer_accounts(user_dic,to_user_dic,int(money))
        if flog:
            print(msg)
        else:
            print(msg)




user_dic={
    '1':register,
    '2':login,
    '3':repayment,
    '4':detailed_list,
    '5':loan,
    '6':shopping,
    '7':transfer_accounts
}
def run():
    while True:
        print('''
1、注册
2、登录
3、还款
4、信用卡清单
5、取现
6、购物
7、转账
8、退出
        ''')
        choice=input('>>>>:').strip()
        if choice=='8':break
        if choice not in user_dic:continue
        user_dic[choice]()

接口:admin_interface.py,bank_interface.py,user_interface.py

admin_interface.py

from db import db_handler
from lib import common
logger_admin=common.get_logger('admin')
def login(name,pwd):
    admin = db_handler.select(name)
    if admin:
        if pwd==admin['pwd']:
            return True,'登陆成功'

        else:
            return False,'密码错误'
    else:
        return False,'用户不存在'

def card_lick(user_dic):
    user_dic['card_lick']=False
    db_handler.update(user_dic)
    logger_admin.info('管理员冻结了%s的信用卡'%user_dic['name'])
def card_thaw(user_dic):
    user_dic['card_lick']=True
    db_handler.update(user_dic)
    logger_admin.info('管理员解冻了%s的信用卡' % user_dic['name'])
def user_lick(user_dic):
    user_dic['user_lick']=False
    db_handler.update(user_dic)
    logger_admin.info('管理员锁定了%s的账号' % user_dic['name'])
def user_thaw(user_dic):
    user_dic['user_lick']=True
    db_handler.update(user_dic)
    logger_admin.info('管理员解锁了%s的账号' % user_dic['name'])

bank_interface.py

from db import db_handler
from lib import common
from lib import common
logger_user=common.get_logger('admin')
def repayment(user_dic,money):
    user_dic['bal']+=money
    print(user_dic['bal'])
    user_dic['ls'].append('你还款%s'%(money))
    db_handler.update(user_dic)
    logger_user.debug('%s还款%s' % (user_dic['name'],money))

def detailed_list(user_dic):
    if not user_dic['ls'] :print('你还没有消费过')
    for line in user_dic['ls']:print(line)


def loan(user_dic,money):
    if user_dic['bal']<money:
        return False,'你的额度不够'
    user_dic['bal']-=money*1.05
    user_dic['ls'].append('你取现%s' % money)
    db_handler.update(user_dic)
    logger_user.debug('%s取现%s' % (user_dic['name'], money))
    return True,'取现成功'


def transfer_accounts(user_dic,to_user_dic,money):
    if user_dic['bal']<money:
        return False,'你的额度不够'
    user_dic['bal']-=money
    user_dic['ls'].append('你给%s转了%s￥'%(to_user_dic['name'],money))
    db_handler.update(user_dic)
    to_user_dic['bal']+=money
    to_user_dic['ls'].append('你收到%s转账%s￥'%(user_dic['name'],money))
    db_handler.update(to_user_dic)
    logger_user.debug('%s转账给%s%s￥' % (user_dic['name'],to_user_dic['name'], money))
    return True,'转账成功'


def shopping(user_dic,shop):
    if user_dic['bal']<shop['price']:
        return False,'你的额度不够'
    user_dic['bal']-=shop['price']
    user_dic['ls'].append('你购买%s花了%s￥' % (shop['name'],shop['price']))
    db_handler.update(user_dic)
    logger_user.info('%s在商城购买了%s花了%s￥' % (user_dic['name'], shop['name'], shop['price']))
    return True,'购买成功'

user_interface.py

from db import db_handler
def select(name):
    return db_handler.select(name)
def update(name,pwd):
    user_dic={'name':name,'pwd':pwd,'card_lick':True,'user_lick':True,'bal':15000,'imin':15000,'ls':[]}
    db_handler.update(user_dic)

def login(user_dic):
    user_dic['user_lick'] = False
    db_handler.update(user_dic)

装饰器以及常用功能common.py

from conf import setting
import logging.config
def login_auth(auth_name):
    from core import user, admin
    def login(func):
        def wrapper(*args,**kwargs):
            if auth_name=='user':
                if not user.data['name']:
                    print('请先登录')
                    user.login()
                else:
                    return func(*args, **kwargs)
            if auth_name=='admin':
                if not admin.data['name']:
                    print('请先登录')
                    admin.login()
                else:
                    return func(*args, **kwargs)
        return wrapper
    return login

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