[GoogleAppEngine] @login_required for Django's Class-based View

Introduction

Google App Engine / Python redirects you to the login screen if the user isn't logged in A decorator called @ login_required.

https://cloud.google.com/appengine/docs/python/tools/webapp/utilmodule#login_required

appengine/ext/webapp/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

(Code quoted from SDK)

It's very convenient, but it's only compatible with webapp I rewrote it for Django. For Class-based View.

code

util.py

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


def login_required(handler_method):
	u"""Django Class-Decorator for 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