详解Django中六个常用的自定义装饰器
装饰器作用
decorator是当今最流行的设计模式之一，很多使用它的人并不知道它是一种设计模式。这种模式有什么特别之处?有兴趣可以看看PythonWiki上例子，使用它可以很方便地修改对象行为，通过使用类似例中的接口将修改动作封装在装饰对象中。
decorator可以动态地修改函数、方法或类的功能，而无需创建子类或修改类的源代码。正因为如此，装饰器可以让代码将变得更干净、更可读、更可维护(这很重要!)，并且减少了许多冗余但又不得不写的代码，使我们可以使用单个方法向多个类添加功能。
对于装饰器的重用性和易用性，Django里面的@login_required就是一个很好的例子。使用它只用一句代码就可以检查用户是否通过身份验证，并将未登录用户重定向到登录url。
该装饰器的使用方法如下:
fromdjango.contrib.auth.decoratorsimportlogin_required @login_required(login_url='/accounts/login/') defmy_view(request): ...
每次用户试图访问my_view时，都会进入login_required中的代码。
Django装饰器
下面介绍一些个人认为比较有用的，或者是之前使用过的具有积极效果的装饰器。事先声明，如要实现同样的业务场景，并不是只有本文中的方法。Django可以实现各种各样的装饰器，这完全根据您的需要进行定制。
GroupRequired
有时需要保护一些视图，只允许某些用户组访问。这时就可以使用下面的装饰器来检查用户是否属于该用户组。
fromdjango.contrib.auth.decoratorsimportuser_passes_test
defgroup_required(*group_names):
"""Requiresusermembershipinatleastoneofthegroupspassedin."""
defin_groups(u):
ifu.is_authenticated():
ifbool(u.groups.filter(name__in=group_names))|u.is_superuser:
returnTrue
returnFalse
returnuser_passes_test(in_groups)
#Thewaytousethisdecoratoris:
@group_required('admins','seller')
defmy_view(request,pk):
...
有关此装饰器更多的介绍，可以参考这里。
Anonymousrequired
这个装饰器是参考Django自带的login_required装饰器，但是功能是相反的情况，即用户必须是未登录的，否则用户将被重定向到settings.py中定义的地址。当我们想要已登录的用户不允许进入某些视图(比如登录)时，非常有用。
defanonymous_required(function=None,redirect_url=None): ifnotredirect_url: redirect_url=settings.LOGIN_REDIRECT_URL actual_decorator=user_passes_test( lambdau:u.is_anonymous(), login_url=redirect_url ) iffunction: returnactual_decorator(function) returnactual_decorator #Thewaytousethisdecoratoris: @anonymous_required defmy_view(request,pk): ...
有关此装饰器更多的介绍，可以参考这里。
Superuserrequired
这个装饰器和上面的group_required类似，但是它只允许超级用户才能访问视图。
fromdjango.core.exceptionsimportPermissionDenied defsuperuser_only(function): """Limitviewtosuperusersonly.""" def_inner(request,*args,**kwargs): ifnotrequest.user.is_superuser: raisePermissionDenied returnfunction(request,*args,**kwargs) return_inner #Thewaytousethisdecoratoris: @superuser_only defmy_view(request): ...
有关此装饰器更多的介绍，可以参考这里。
Ajaxrequired
这个装饰器用于检查请求是否是AJAX请求，在使用jQuery等Javascript框架时，这是一个非常有用的装饰器，也是一种保护应用程序的好方法。
fromdjango.httpimportHttpResponseBadRequest defajax_required(f): """ AJAXrequestrequireddecorator useitinyourviews: @ajax_required defmy_view(request): .... """ defwrap(request,*args,**kwargs): ifnotrequest.is_ajax(): returnHttpResponseBadRequest() returnf(request,*args,**kwargs) wrap.__doc__=f.__doc__ wrap.__name__=f.__name__ returnwrap #Thewaytousethisdecoratoris: @ajax_required defmy_view(request): ...
有关此装饰器更多的介绍，可以参考这里。
Timeit
如果您需要改进某个视图的响应时间，或者只想知道运行需要多长时间，那么这个装饰器非常有用。
deftimeit(method):
deftimed(*args,**kw):
ts=time.time()
result=method(*args,**kw)
te=time.time()
print('%r(%r,%r)%2.2fsec'%(method.__name__,args,kw,te-ts))
returnresult
returntimed
#Thewaytousethisdecoratoris:
@timeit
defmy_view(request):
...
有关此装饰器更多的介绍，可以参考这里。
自定义功能
下面这个装饰器只是一个示例，测试你能够轻松地检查某些权限或某些判断条件，并100%自己定制。
想象你有一个博客、购物论坛，如果用户需要有很多积分才能发表评论，这是一个避免垃圾信息的好方法。下面创建一个装饰器来检查用户是否已登录并拥有超过10个积分，这样才可以发表评论，否则将抛出一个Forbidden。
fromdjango.httpimportHttpResponseForbidden
logger=logging.getLogger(__name__)
defuser_can_write_a_review(func):
"""Viewdecoratorthatchecksauserisallowedtowriteareview,innegativecasethedecoratorreturnForbidden"""
@functools.wraps(func)
defwrapper(request,*args,**kwargs):
ifrequest.user.is_authenticated()andrequest.user.points<10:
logger.warning('The{}userhastriedtowriteareview,butdoesnothaveenoughpointstodoso'.format(request.user.pk))
returnHttpResponseForbidden()
returnfunc(request,*args,**kwargs)
returnwrapper
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持毛票票。