bottle_memcache.py

#!/usr/bin/env python
#
# -*- mode:python; sh-basic-offset:4; indent-tabs-mode:nil; coding:utf-8 -*-
# vim:set tabstop=4 softtabstop=4 expandtab shiftwidth=4 fileencoding=utf-8:
#

import memcache
import inspect
from bottle import PluginError


class MemcachePlugin(object):

    name = 'memcache'

    def __init__(self, servers=['localhost:11211', ], keyword='mc',
                 server_max_value_length=memcache.SERVER_MAX_VALUE_LENGTH,
                 socket_timeout=memcache._SOCKET_TIMEOUT):

        self.servers = servers
        self.keyword = keyword
        self.server_max_value_length = server_max_value_length
        self.socket_timeout = socket_timeout

    def setup(self, app):
        for other in app.plugins:
            if not isinstance(other, MemcachePlugin):
                continue
            if other.keyword == self.keyword:
                raise PluginError("Found another memcache plugin with "\
                        "conflicting settings (non-unique keyword).")

    def apply(self, callback, context):
        conf = context['config'].get('memcache') or {}
        self.servers = conf.get('servers', self.servers)
        self.keyword = conf.get('keyword', self.keyword)
        self.server_max_value_length = conf.get(
            'server_max_value_length', self.server_max_value_length
        )
        self.socket_timeout = conf.get('socket_timeout', self.socket_timeout)

        args = inspect.getargspec(context['callback'])[0]
        if self.keyword not in args:
            return callback

        def wrapper(*args, **kwargs):
            mc = memcache.Client(
                servers=self.servers,
                server_max_value_length=self.server_max_value_length,
                socket_timeout=self.socket_timeout,
                debug=0
            )
            kwargs[self.keyword] = mc
            rv = callback(*args, **kwargs)
            return rv
        return wrapper

Plugin = MemcachePlugin