Produits Entr'ouvert » Authentic 2

from . import middleware, utils

class WrappedHandler(object):

    def __init__(self, real_class, **kwargs):

        if isinstance(real_class, basestring):

            real_class = utils.import_class(real_class)

        self.wrapped = real_class(**kwargs)

    def fill_record(self, record):

        raise NotImplementedError

    def handle(self, record):

        return self.wrapped.handle(record)

    def __getattr__(self, name):

        return getattr(self.wrappped, name)

    def __setattr(self, name, value):

        setattr(self.wrapped, name, value)

class RequestWrappedHandler(WrappedHandler):

    def fill_record(self, record):

        request = middleware.StoreRequestMiddleware.get_request()

        if request is None:

            record.update({

                'host': '-',

                'ip': '-',

                'user': '-',

            })

        record.update({

                'host': request.META.get('HTTP_HOST', '-'),

                'ip': request.META.get('REMOTE_ADDR', '-'),

                'user': request.user.get_username(),

        })