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')
相關文章
相關標籤/搜索