Skip to content

Instantly share code, notes, and snippets.

@aldoah0a

aldoah0a/redis_session_backend.py

Forked from mikeyk/redis_session_backend.py
Created January 18, 2014 23:34
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save aldoah0a/8498274 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save aldoah0a/8498274 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. Mike Krieger revised this gist Apr 8, 2011. 1 changed file with 1 addition and 1 deletion.
    2 changes: 1 addition & 1 deletion redis_session_backend.py
    View file
    Original file line number Diff line number Diff line change
    @@ -4,7 +4,7 @@
    import redis

    class SessionStore(SessionBase):
    """ Write-through Redis store for sessions"""
    """ Redis store for sessions"""

    def __init__(self, session_key=None):
    self.redis = redis.Redis(
  2. Mike Krieger renamed this gist Apr 8, 2011. 1 changed file with 0 additions and 0 deletions.
    0 Redis Session Backend for Django 1.3 → redis_session_backend.py
    View file
    File renamed without changes.
  3. Mike Krieger created this gist Apr 8, 2011.
    69 changes: 69 additions & 0 deletions Redis Session Backend for Django 1.3
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,69 @@
    from django.contrib.sessions.backends.base import SessionBase, CreateError
    from django.conf import settings
    from django.utils.encoding import force_unicode
    import redis

    class SessionStore(SessionBase):
    """ Write-through Redis store for sessions"""

    def __init__(self, session_key=None):
    self.redis = redis.Redis(
    host = settings.SESSION_REDIS_HOST,
    port = settings.SESSION_REDIS_PORT,
    socket_timeout = settings.REDIS_SOCKET_TIMEOUT
    )
    super(SessionStore, self).__init__(session_key)

    @classmethod
    def _get_key_for_redis(self, session_key):
    return 'sessionstore:%s' % session_key

    def _redis_key(self):
    return self.__class__._get_key_for_redis(self.session_key)

    def load(self):
    session_data = self.redis.get(self._redis_key())
    if session_data is not None:
    return self.decode(force_unicode(session_data))
    else:
    self.create()
    return {}

    def create(self):
    max_attempts = 10
    attempt = 0
    while True:
    self.session_key = self._get_new_session_key()
    session_dict = self._get_session(no_load = True)
    session_data = self.encode(session_dict)

    was_created = self.redis.setnx(self._redis_key(),
    session_data)
    if was_created:
    self.redis.expire(self._redis_key(),
    settings.SESSION_COOKIE_AGE)
    self.modified = True
    self._session_cache = session_dict
    return
    else:
    # extremely unlikely
    if attempt == max_attempts:
    raise CreateError
    else:
    attempt += 1
    continue

    def save(self, must_create=False):
    if must_create:
    self.create()
    else:
    encoded_data = self.encode(self._session)
    self.redis.setex(self._redis_key(), encoded_data, settings.SESSION_COOKIE_AGE)

    def exists(self, session_key):
    return self.redis.exists(self.__class__._get_key_for_redis(session_key))

    def delete(self, session_key=None):
    to_delete = (session_key or self.session_key)
    redis_key = self.__class__._get_key_for_redis(to_delete)
    self.redis.delete(redis_key)