GAE/Pでセッションエンジンのバックエンドをmemcacheにするサンプル

Djangoのセッションエンジンのバックエンドを、Google App Engineのmemcacheにするサンプルを以下に示します。※ 追記: 「Google App Engine/Pythonでセッションを使う方法」という使い方を載せた記事を投稿しました。

from django.contrib.sessions.backends.base import SessionBase, CreateError
from google.appengine.api import memcache


class SessionStore(SessionBase):
    def __init__(self, session_key=None):
        self._session_namespace = "__SESSION_NAMESPACE__"
        super(SessionStore, self).__init__(session_key)

    def load(self):
        data = memcache.get(self.session_key,
                namespace=self._session_namespace)
        if data:
            return data
        else:
            self.create()
            return {}

    def exists(self, session_key):
        data = memcache.get(session_key,
                namespace=self._session_namespace)
        if data:
            return True
        else:
            return False

    def create(self):
        while True:
            self.session_key = self._get_new_session_key()
            try:
                self.save(must_create=True)
            except CreateError:
                continue
            self.modified = True
            self._session_cache = {}
            return

    def save(self, must_create=False):
        if must_create:
            is_success = memcache.add(
                    self.session_key,
                    self._get_session(no_load=must_create),
                    time=self.get_expiry_age(),
                    namespace=self._session_namespace)
        else:
            is_success = memcache.set(
                    self.session_key,
                    self._get_session(no_load=must_create),
                    time=self.get_expiry_age(),
                    namespace=self._session_namespace)
        if not is_success:
            raise CreateError

    def delete(self, session_key=None):
        if session_key is None:
            if self._session_key is None:
                return
            session_key = self._session_key
        memecache.delete(session_key,
                seconds=300,
                namespace=self._session_namespace)

この記事が役に立った場合、シェアしていただけると励みになります!!