python(4) - 装饰器2

接下来修改一下上一篇的login，将用户名传递给验证函数。

def login(func):         #接收一个函数作为参数
def inner(name):
print("用户验证通过。。。。")
return func(name)         #将函数返回
return inner                  #返回inner函数

@login
def video(name):
print("welcome %s to video!" % name)

这样当用户 张三 执行video(“张三”)时，程序怎么运行的呢？

首先会将inner返回给video. 然后video(“张三”)实际上就是执行inner(“张三”)。

也可以将用户名密码同时传进去。

def login(func):
def inner(username, password):
if username == "zhangsan" and password == '1234':
func(username, password)
else:
print("username or password is incorrect")
return inner

@login
def tv(username, password):
print("Welcom %s, your password is %s" % (username, password))

tv('zhangsan', "1234")