DjangoのClass-based view用@login_required

django_login.py

# -*- coding: utf-8 -*-
from django.http import HttpResponsePermanentRedirect
from google.appengine.api import users


def login_required(handler_method):
  u"""Django の Class-based view 用デコレータ

  Example:
    class MyView(View):

      @login_required
      def get(self, request):
        user = users.get_current_user()
        ...
  """
  def check_login(self, request, *args):
    user = users.get_current_user()

    if not user:
      login_url = users.create_login_url(request.get_full_path()))
      return HttpResponsePermanentRedirect(login_url)
    else:
      return handler_method(self, request, *args) 
    
  return check_login

util.py

def login_required(handler_method):
  """A decorator to require that a user be logged in to access a handler.

  To use it, decorate your get() method like this:

    @login_required
    def get(self):
      user = users.get_current_user(self)
      self.response.out.write('Hello, ' + user.nickname())

  We will redirect to a login page if the user is not logged in. We always
  redirect to the request URI, and Google Accounts only redirects back as a GET
  request, so this should not be used for POSTs.
  """
  def check_login(self, *args):
    if self.request.method != 'GET':
      raise webapp.Error('The check_login decorator can only be used for GET '
                         'requests')
    user = users.get_current_user()
    if not user:
      self.redirect(users.create_login_url(self.request.uri))
      return
    else:
      handler_method(self, *args)
  return check_login

util.py は GAE SDK の appengine/ext/webapp/util.py より引用