django 装饰器

  装饰器能够给一个函数,方法或类进行加工,添加额外的功能。python

  在这篇中使用装饰器给页面添加session而不让直接访问index,和show。在views.py中session

def index(request):
        return HttpResponse('index')

def show(request):
        return HttpResponse('show')

  这样能够直接访问index和show,若是只容许登录过的用户访问index和show,那么就需修改代码app

def index(request):
        if request.session.get('username'):
            return HttpResponse('index')
        else:
            return  HttpResponse('login')
def show(request): if request.session.get('username'): return HttpResponse('show') else: return HttpResponse('login')

  这样能够实现限制登录过的用户访问功能,可是代码中也出现了许多的相同部分,因而能够把这些相同的部分写入一个函数中,用这样一个函数装饰index和show。这样的函数就是装饰器。函数

def decorator(main_func):
    def wrapper(request):               #index,show中是一个参数,因此在wrapper中也是一个参数
        if request.session.get('username'):
            return main_func(request)
        else:
            return HttpResponse('login')
    return wrapper

@decorator
def index(request):
    return HttpResponse('index')
def show(request):
    return HttpResponse('show')

  这样在视图函数中只要是一个参数就能够经过decorator函数装饰,若是有两个参数就须要修改装饰器blog

def decorator(main_func):
    def wrapper(request):               
        if request.session.get('username'):
            return main_func(request)
        else:
            return HttpResponse('login')
    return wrapper

def decorator1(main_func):
    def wrapper(request,page):               
        if request.session.get('username'):
            return main_func(request,page)
        else:
            return HttpResponse('login')
    return wrapper

@decorator
def index(request):
    return HttpResponse('index')

@decorator1
def show(request,page):
    return HttpResponse('show')

  这个若是有一个参数就经过decorator来修饰,若是有两个参数就经过decorator1来修饰。因而能够经过动态参数的方式来结合decorator和decorator1,能够同时修饰index和show。get

def decorator3(main_func):
        def wrapper(request,*args,**kwargs):
                if not request.session.get('username'):
                        return main_func(request,*args,**kwargs)
                else:
                        return HttpResponse('login')
        return wrapper


@decorator3
def index(request,*args,**kwargs):
        return HttpResponse('index')
@decorator3
def show(request,*args,**kwargs):
        return HttpResponse('show')
相关文章
相关标签/搜索