1.权限登录,有些页面前提条件是需要用户名密码登录
username, password = 'lily', '1qaz2wsx'
def auth(func):
def wrapper(*args, **kwargs):
user = input("username:").strip()
passw = input("password:").strip()
if user == username and passw == password:
print("login successfully")
func(*args, **kwargs)
else:
exit("invalid username or password")
return wrapper
def index():
print("welcome to index")
@auth
def home():
print("welcome to home")
@auth
def bbs():
print("welcome to bbs")
index()
home()
bbs()
运行结果
welcome to index
username:lily
password:1qaz2wsx
login successfully
welcome to home
username:li
password:rf
invalid username or password
2.带参数传入,两个账号一个local登录,一个remote登录
在外部再加一层装饰
username, password = 'lily', '1qaz2wsx'
def auth(auth_type):
def outer_wrapper(func):
def wrapper(*args, **kwargs):
if auth_type == 'local':
user = input("username:").strip()
passw = input("password:").strip()
if user == username and passw == password:
print("login successfully")
func(*args, **kwargs)
else:
exit("invalid username or password")
elif auth_type == 'remote':
print("login by other way 这里只是作为示例")
return wrapper
return outer_wrapper
def index():
print("welcome to index")
@auth(auth_type = 'local')
def home():
print("welcome to home")
@auth(auth_type = 'remote')
def bbs():
print("welcome to bbs")
index()
home()
bbs()
3.日志
4.handle处理,例如测试中的黑名单(叉掉弹框)