Django中如何自定義裝飾器

Django中如何自定義裝飾器，相信很多沒有經驗的人對此束手無策，為此本文總結了問題出現的原因和解決方法，通過這篇文章希望你能解決這個問題。

裝飾器作用

decorator是當今最流行的設計模式之一，很多使用它的人并不知道它是一種設計模式。這種模式有什么特別之處? 有興趣可以看看Python Wiki上例子，使用它可以很方便地修改對象行為，通過使用類似例中的接口將修改動作封裝在裝飾對象中。

decorator 可以動態地修改函數、方法或類的功能，而無需創建子類或修改類的源代碼。正因為如此，裝飾器可以讓代碼將變得更干凈、更可讀、更可維護(這很重要!)，并且減少了許多冗余但又不得不寫的代碼，使我們可以使用單個方法向多個類添加功能。

對于裝飾器的重用性和易用性，Django里面的@login_required就是一個很好的例子。使用它只用一句代碼就可以檢查用戶是否通過身份驗證，并將未登錄用戶重定向到登錄url。

該裝飾器的使用方法如下:

from django.contrib.auth.decorators import login_required

@login_required(login_url='/accounts/login/')
def my_view(request):
  ...

每次用戶試圖訪問 my_view 時，都會進入 login_required 中的代碼。

Django裝飾器

下面介紹一些個人認為比較有用的，或者是之前使用過的具有積極效果的裝飾器。事先聲明，如要實現同樣的業務場景，并不是只有本文中的方法。Django可以實現各種各樣的裝飾器，這完全根據您的需要進行定制。

Group Required

有時需要保護一些視圖，只允許某些用戶組訪問。這時就可以使用下面的裝飾器來檢查用戶是否屬于該用戶組。

from django.contrib.auth.decorators import user_passes_test

def group_required(*group_names):
  """Requires user membership in at least one of the groups passed in."""

  def in_groups(u):
    if u.is_authenticated():
      if bool(u.groups.filter(name__in=group_names)) | u.is_superuser:
        return True
    return False
  return user_passes_test(in_groups)


# The way to use this decorator is:
@group_required('admins', 'seller')
def my_view(request, pk):
  ...

有關此裝飾器更多的介紹，可以參考這里。

Anonymous required

這個裝飾器是參考Django自帶的 login_required 裝飾器，但是功能是相反的情況，即用戶必須是未登錄的，否則用戶將被重定向到 settings.py 中定義的地址。當我們想要已登錄的用戶不允許進入某些視圖(比如登錄)時，非常有用。

def anonymous_required(function=None, redirect_url=None):

  if not redirect_url:
    redirect_url = settings.LOGIN_REDIRECT_URL

  actual_decorator = user_passes_test(
    lambda u: u.is_anonymous(),
    login_url=redirect_url
  )

  if function:
    return actual_decorator(function)
  return actual_decorator


# The way to use this decorator is:
@anonymous_required
def my_view(request, pk):
  ...

有關此裝飾器更多的介紹，可以參考這里。

Superuser required

這個裝飾器和上面的 group_required 類似， 但是它只允許超級用戶才能訪問視圖。

from django.core.exceptions import PermissionDenied


def superuser_only(function):
  """Limit view to superusers only."""

  def _inner(request, *args, **kwargs):
    if not request.user.is_superuser:
      raise PermissionDenied
    return function(request, *args, **kwargs)

  return _inner


# The way to use this decorator is:
@superuser_only
def my_view(request):
  ...

有關此裝飾器更多的介紹，可以參考這里。

Ajax required

這個裝飾器用于檢查請求是否是AJAX請求，在使用jQuery等Javascript框架時，這是一個非常有用的裝飾器，也是一種保護應用程序的好方法。

from django.http import HttpResponseBadRequest


def ajax_required(f):
  """
  AJAX request required decorator
  use it in your views:
 
  @ajax_required
  def my_view(request):
    ....
 
  """

  def wrap(request, *args, **kwargs):
    if not request.is_ajax():
      return HttpResponseBadRequest()
    return f(request, *args, **kwargs)

  wrap.__doc__ = f.__doc__
  wrap.__name__ = f.__name__
  return wrap


# The way to use this decorator is:
@ajax_required
def my_view(request):
  ...

有關此裝飾器更多的介紹，可以參考這里。

Time it

如果您需要改進某個視圖的響應時間，或者只想知道運行需要多長時間，那么這個裝飾器非常有用。

def timeit(method):

  def timed(*args, **kw):
    ts = time.time()
    result = method(*args, **kw)
    te = time.time()
    print('%r (%r, %r) %2.2f sec' % (method.__name__, args, kw, te - ts))
    return result

  return timed


# The way to use this decorator is:
@timeit
def my_view(request):
  ...

有關此裝飾器更多的介紹，可以參考這里。

自定義功能

下面這個裝飾器只是一個示例，測試你能夠輕松地檢查某些權限或某些判斷條件，并100%自己定制。

 想象你有一個博客、購物論壇，如果用戶需要有很多積分才能發表評論，這是一個避免垃圾信息的好方法。下面創建一個裝飾器來檢查用戶是否已登錄并擁有超過10個積分，這樣才可以發表評論，否則將拋出一個Forbidden。

from django.http import HttpResponseForbidden

logger = logging.getLogger(__name__)
def user_can_write_a_review(func):
  """View decorator that checks a user is allowed to write a review, in negative case the decorator return Forbidden"""

  @functools.wraps(func)
  def wrapper(request, *args, **kwargs):
    if request.user.is_authenticated() and request.user.points < 10:
      logger.warning('The {} user has tried to write a review, but does not have enough points to do so'.format( request.user.pk))
      return HttpResponseForbidden()

    return func(request, *args, **kwargs)

  return wrapper

看完上述內容，你們掌握Django中如何自定義裝飾器的方法了嗎？如果還想學到更多技能或想了解更多相關內容，歡迎關注億速云行業資訊頻道，感謝各位的閱讀！