装饰器是一个很著名的设计模式,经常被用于有切面需求的场景,较为经典的有插入日志、性能测试、事务处理等。
装饰器其实就是一个工厂函数,它接受一个函数为参数,然后返回一个新函数,其闭包中包含了原函数
1、简单装饰器:
def deco(func):
def wrapper():
print "start"
func() #调用函数
print "end"
return wrapper
@deco
def myfun():
print "run"
myfun()
由于装饰器函数返回的是原函数的闭包wrapper,实际上被装饰后的函数就是wrapper,其运行方式就和wrapper一样。相当于:
myfun=deco(myfun)
2、装饰任意参数的函数:
def deco(func):
def warpper(*args,**kw):
print "start"
func(*args,**kw)
print "end"
return warpper
@deco
def myfun1(param1):
print "run with param %s"%(param1)
@deco
def myfun2(param1,param2):
print "run with param %s and %s"%(param1,param2)
myfun1("something")
myfun2("something","otherthing")
# start
# run with param something
# end
# start
# run with param something and otherthing
# end
3、带参数的装饰器
装饰器接受一个函数作为参数,这个毋庸置疑。但是有时候我们需要装饰器接受另外的参数。此时需要再加一层函数,实际上是定义了一个生成装饰器的工厂函数,调用它,搭配需要的参数,来返回合适的装饰器
def log(text):
def deco(func):
def wrapper(*args,**kw):
print text
func(*args,**kw)
print text + " again"
return wrapper
return deco
@log("hello")
def myfun(message):
print message
myfun("world")
# hello
# world
# hello again
log=log("hello"),把返回的deco函数赋值给log,此时log相当于其包含text=“hello”的闭包
myfun=log(myfun),相当于把myfun传入了deco函数,并且返回wrapper,并赋值给myfun,此时myfun相当于其装饰后的闭包。
整体来看是myfun=log("hello")(myfun)
4、装饰器带类参数
class locker:
def __init__(self):
print("locker.__init__() should be not called.")
@staticmethod
def acquire():
print("locker.acquire() called.(这是静态方法)")
@staticmethod
def release():
print(" locker.release() called.(不需要对象实例)")
def deco(cls):
'''''cls 必须实现acquire和release静态方法'''
def _deco(func):
def __deco():
print("before %s called [%s]." % (func.__name__, cls))
cls.acquire()
try:
return func()
finally:
cls.release()
return __deco
return _deco
@deco(locker)
def myfunc():
print(" myfunc() called.")
myfunc()
myfunc()
5、django自定义装饰器实现登录验证
def Check_Login(func): #自定义登录验证装饰器
def warpper(request,*args,**kwargs):
is_login = request.session.get('IS_LOGIN', False)
if is_login:
func(request,*args,**kwargs)
else:
return HttpResponseRedirect("/polls/login_user")
return warpper
def login_user(request):
if request.method == 'POST':
form = LoginForm(request.POST)
if form.is_valid():
all_data = form.clean() #获取post数据,例如 {'username': u'yang1', 'password': 111}
exist = User.objects.filter(username = all_data['Form_username'],password = all_data['Form_password']).first()
if exist:
request.session['IS_LOGIN'] = True #设置session的随机字段值
request.session['uname'] = exist.username #设置uname字段为登录用户
return HttpResponseRedirect('/polls/home')
else:
return HttpResponse("账户或密码错误")
else:
form = LoginForm()
return render(request, 'polls/login_user.html', {'form': form})
@Check_Login
def home(request):
username = request.session.get('uname', False) #获取登录用户名
return render(request, 'polls/home.html', {'username': username}) #用户名渲染到前端页面