"""A general session store."""
try:
from cPickle import load, dump, HIGHEST_PROTOCOL as maxPickleProtocol
except ImportError:
from pickle import load, dump, HIGHEST_PROTOCOL as maxPickleProtocol
from time import time
from MiscUtils import AbstractError
def dumpWithHighestProtocol(obj, f):
"""Same as pickle.dump, but by default with the highest protocol."""
return dump(obj, f, maxPickleProtocol)
class SessionStore(object):
"""A general session store.
SessionStores are dictionary-like objects used by Application to
store session state. This class is abstract and it's up to the
concrete subclass to implement several key methods that determine
how sessions are stored (such as in memory, on disk or in a
database). We assume that session keys are always strings.
Subclasses often encode sessions for storage somewhere. In light
of that, this class also defines methods encoder(), decoder() and
setEncoderDecoder(). The encoder and decoder default to the load()
and dump() functions of the cPickle or pickle module. However,
using the setEncoderDecoder() method, you can use the functions
from marshal (if appropriate) or your own encoding scheme.
Subclasses should use encoder() and decoder() (and not
pickle.load() and pickle.dump()).
Subclasses may rely on the attribute self._app to point to the
application.
Subclasses should be named SessionFooStore since Application
expects "Foo" to appear for the "SessionStore" setting and
automatically prepends Session and appends Store. Currently, you
will also need to add another import statement in Application.py.
Search for SessionStore and you'll find the place.
TO DO
* Should there be a check-in/check-out strategy for sessions to
prevent concurrent requests on the same session? If so, that can
probably be done at this level (as opposed to pushing the burden
on various subclasses).
"""
def __init__(self, app):
"""Initialize the session store.
Subclasses must invoke super.
"""
self._app = app
self._alwaysSave = app._alwaysSaveSessions
self._encoder = dumpWithHighestProtocol
self._decoder = load
def application(self):
"""Return the application owning the session store."""
return self._app
def __len__(self):
"""Return the number of sessions in the store.
Subclasses must implement this method.
"""
raise AbstractError(self.__class__)
def __getitem__(self, key):
"""Get a session item from the store.
Subclasses must implement this method.
"""
raise AbstractError(self.__class__)
def __setitem__(self, key, value):
"""Set a session item, saving it to the store.
Subclasses must implement this method.
"""
raise AbstractError(self.__class__)
def __delitem__(self, key):
"""Delete a session item from the store.
Subclasses are responsible for expiring the session as well.
Something along the lines of:
session = self[key]
if not session.isExpired():
session.expiring()
"""
raise AbstractError(self.__class__)
def __contains__(self, key):
"""Check whether the session store has a given key.
Subclasses must implement this method.
"""
raise AbstractError(self.__class__)
def __iter__(self):
"""Return an iterator over the stored session keys.
Subclasses must implement this method.
"""
raise AbstractError(self.__class__)
def has_key(self, key):
"""Check whether the session store has a given key."""
return key in self
def keys(self):
"""Return a list with the keys of all the stored sessions.
Subclasses must implement this method.
"""
raise AbstractError(self.__class__)
def iterkeys(self):
"""Return an iterator over the stored session keys."""
return iter(self)
def clear(self):
"""Clear the session store, removing all of its items.
Subclasses must implement this method.
"""
raise AbstractError(self.__class__)
def setdefault(self, key, default=None):
"""Return value if key available, else default (also setting it).
Subclasses must implement this method.
"""
raise AbstractError(self.__class__)
def pop(self, key, default=None):
"""Return value if key available, else default (also remove key).
Subclasses must implement this method.
"""
raise AbstractError(self.__class__)
def storeSession(self, session):
"""Save potentially changed session in the store.
Used at the end of transactions.
Subclasses must implement this method.
"""
raise AbstractError(self.__class__)
def storeAllSessions(self):
"""Permanently save all sessions in the store.
Used when the application server is shut down.
Subclasses must implement this method.
"""
raise AbstractError(self.__class__)
def cleanStaleSessions(self, task=None):
"""Clean stale sessions.
Called by the Application to tell this store to clean out all
sessions that have exceeded their lifetime.
"""
curTime = time()
keys = []
for key in self.keys():
try:
session = self[key]
except KeyError:
pass
else:
try:
timeout = session.timeout()
if timeout is not None and (timeout == 0
or curTime >= session.lastAccessTime() + timeout):
keys.append(key)
except AttributeError:
raise ValueError('Not a Session object: %r' % session)
for key in keys:
try:
del self[key]
except KeyError:
pass
def get(self, key, default=None):
"""Return value if key available, else return the default."""
try:
return self[key]
except KeyError:
return default
def items(self):
"""Return a list with the (key, value) pairs for all sessions."""
items = []
for key in self:
try:
items.append((key, self[key]))
except KeyError:
pass
return items
def values(self):
"""Return a list with the values of all stored sessions."""
values = []
for key in self:
try:
values.append(self[key])
except KeyError:
pass
return values
def iteritems(self):
"""Return an iterator over the (key, value) pairs for all sessions."""
for key in self:
try:
yield key, self[key]
except KeyError:
pass
def itervalues(self):
"""Return an iterator over the stored values of all sessions."""
for key in self:
try:
yield self[key]
except KeyError:
pass
def encoder(self):
"""Return the value serializer for the store."""
return self._encoder
def decoder(self):
"""Return the value deserializer for the store."""
return self._decoder
def setEncoderDecoder(self, encoder, decoder):
"""Set the serializer and deserializer for the store."""
self._encoder = encoder
self._decoder = decoder
def __repr__(self):
"""Return string representation of the store like a dictionary."""
return repr(dict(self.items()))