装饰器多层语法糖嵌套
小练习：
# # 先验证登录
# # 再验证 输入的金额 --- 符合数字 / 余额充足
#
# # 取款函数里面
# def get_balance():
# # 校验登录
# # 校验金额 符合数字 / 余额充足 --- 把金额通过装饰器 返回来
# # 拿着你的金额进行提款
use_data = {'username': 'dream', 'password': '521'}
bank_data = {'dream': {'pay_pwd': '521', 'balance': 1000}}
def login_auth(func):
def inner(*args,**kwargs):
if use_data['username']!='dream' and use_data['password']!='521':
return False,f'用户名或密码错误!'
else:
return func(*args,**kwargs)
return inner
def get_balance_auth(func):
def inner(*args,**kwargs):
if bank_data['dream']['pay_pwd'] != '521':
return False,f'取款密码错误！'
get_balance_input=input('请输入取款金额：').strip()
if get_balance_input.isdigit():
get_balance_input=int(get_balance_input)
if 0<get_balance_input<=bank_data['dream']['balance']:
return func(balance=get_balance_input,*args,**kwargs)
else:
return False,f'余额不足！'
return inner
@login_auth
@get_balance_auth
def get_balance(*args,**kwargs):
# print(kwargs)
bank_data['dream']['balance']-=kwargs.get('balance')
bank_data['dream']['balance']=bank_data['dream']['balance']
# print(bank_data)
return f"取款金额为{kwargs.get('balance')},还剩下{bank_data['dream']['balance']}"
print(get_balance())
有参装饰器模板
"""有参装饰器模板"""
def wrapper_func(tag):
def wrapper(func):
def inner(*args,**kwargs):
#tag 表明当前需要处理的使用函数的逻辑
#存钱--不需要校验提款密码
#取钱--需要校验提款密码
if tag==None:
...
#正常函数内需要的逻辑
return func(*args,**kwargs)
return inner
return wrapper
wrapper_func(tag=None)
def balance():
...
练习
- use_data = {'username': 'dream', 'password': '521'}
bank_data = {'dream': {'pay_pwd': '521', 'balance': 1000}}
- 取款-----校验登录-----校验取款密码
- 存款------校验登录-----不需要校验密码
use_data = {'username': 'dream', 'password': '521'}
bank_data = {'dream': {'pay_pwd': '521', 'balance': 1000}}
def check_balance(tag_fun=None,tag=None):
if tag_fun=='login':
def login_auth(func):
def inner(*args,**kwargs):
if use_data['username']!='dream' and use_data['password']!='521':
return False,f"用户名或密码错误!"
else:
return func(*args,**kwargs)
return inner
return login_auth
elif tag_fun=='balance':
def balance_auth(func):
def inner(*args,**kwargs):
#存款
balance_input=input('请输入金额：')
if not balance_input.isdigit():
return False, f'输入有误！'
balance_input=int(balance_input)
if tag!='add_balance':
pay_password_input=input('请输入取款密码：').strip()
if pay_password_input!=bank_data['dream']['pay_pwd']:
return False,f'取款密码有误！'
if balance_input>bank_data['dream']['balance']:
return False,f'余额不足！'
return func(balance=balance_input,*args,**kwargs)
return inner
return balance_auth
@check_balance(tag_fun='login')
@check_balance(tag_fun='balance',tag='get_balance')
#取款
def get_balance(*args,**kwargs):
bank_data['dream']['balance']-=kwargs.get('balance')
bank_data['dream']['balance']=bank_data['dream']['balance']
return f"取款{kwargs.get('balance')}剩余{bank_data['dream']['balance']}"
print(get_balance())
#存款
@check_balance(tag_fun='login')
@check_balance(tag_fun='balance',tag='add_balance')
def add_balance(*args,**kwargs):
bank_data['dream']['balance'] += kwargs.get('balance')
bank_data['dream']['balance'] = bank_data['dream']['balance']
return f"村款{kwargs.get('balance')}剩余{bank_data['dream']['balance']}"
# print(add_balance())