From 5273e9a48d04e0d36c2a0eb328fbd1104c33cdda Mon Sep 17 00:00:00 2001 From: virusdefender <1670873886@qq.com> Date: Wed, 12 Aug 2015 23:18:22 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=20decorator=20=E5=AF=BC?= =?UTF-8?q?=E8=87=B4=20swagger=20=E6=97=A0=E6=B3=95=E8=AF=86=E5=88=AB?= =?UTF-8?q?=E6=B3=A8=E9=87=8A=E7=9A=84=E9=97=AE=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- account/decorators.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/account/decorators.py b/account/decorators.py index 01cdd68..ff1b5b0 100644 --- a/account/decorators.py +++ b/account/decorators.py @@ -1,4 +1,5 @@ # coding=utf-8 +from functools import wraps from django.http import HttpResponse from django.shortcuts import render @@ -7,6 +8,7 @@ from .models import User def login_required(func): + @wraps(func) def check(*args, **kwargs): # 在class based views 里面,args 有两个元素,一个是self, 第二个才是request, # 在function based views 里面,args 只有request 一个参数 @@ -21,6 +23,7 @@ def login_required(func): def admin_required(func): + @wraps(func) def check(*args, **kwargs): request = args[-1] if request.user.is_authenticated() and request.user.admin_type: