Synnefo » ./kamaki

from astakosclient import AstakosClient as SynnefoAstakosClientOrig

from kamaki.clients import (

    Client, ClientError, Logged, RequestManager, recvlog)

class SynnefoAstakosClient(SynnefoAstakosClientOrig, Logged):

    """A synnefo astakosclient.AstakosClient wrapper, that logs"""

    def _dump_response(self, request, status, message, data):

        plog = ''

        if self.LOG_PID:

            recvlog.info('\n%s <-- [req: %s]\n' % (self, request))

            plog = '\t[%s]' % self

        recvlog.info('%d %s%s' % (status, message, plog))

        recvlog.info('data size: %s%s' % (len(data), plog))

        if self.LOG_DATA and data:

            data = '%s%s' % (data, plog)

            if not self.LOG_TOKEN:

                token = request.headers.get('X-Auth-Token', '')

                if token:

                    data = data.replace(token, '...')

                recvlog.info(data)

        recvlog.info('-             -        -     -   -  - -')

    def _call_astakos(self, *args, **kwargs):

        r = super(SynnefoAstakosClient, self)._call_astakos(*args, **kwargs)

        try:

            log_request = getattr(self, 'log_request', None)

            if log_request:

                req = RequestManager(

                    method=log_request['method'],

                    url='%s://%s' % (self.scheme, self.astakos_base_url),

                    path=log_request['path'],

                    data=log_request.get('body', None),

                    headers=log_request.get('headers', dict()))

                req.dump_log()

                log_response = getattr(self, 'log_response', None)

                if log_response:

                    self._dump_response(

                        req,

                        status=log_response['status'],

                        message=log_response['message'],

                        data=log_response.get('data', ''))

        except Exception as e:

            print 'GOT EXCEPTION', e

        finally:

            return r