Coverage for ion/services/coi/logger : 80.00%

#!/usr/bin/env python 

""" 

@file ion/services/coi/logger.py 

@author Michael Meisinger 

@brief service backend for logging. Plays nicely with logging package 

""" 

import logging 

from twisted.internet import defer 

from ion.core.process.process import ProcessFactory 

from ion.core.process.service_process import ServiceProcess, ServiceClient 

import ion.util.ionlog 

log = ion.util.ionlog.getLogger(__name__) 

logserv = logging.getLogger('logServer') 

class LoggerService(ServiceProcess): 

    """Logger service interface 

    """ 

    # Declaration of service 

    declare = ServiceProcess.service_declare( 

        name='logger', 

        version='0.1.0', 

        dependencies=[] 

    ) 

    def slc_init(self): 

        log.info("LoggingService initialized") 

    def op_config(self, content, headers, msg): 

        pass 

    @defer.inlineCallbacks 

    def op_logmsg(self, content, headers, msg): 

        level = content.get('level','info') 

        logmsg = content.get('msg','#NO MESSAGE#') 

        # need to do something reasonable with these soon 

        # lfrom = headers.get('sender','') 

        # ltime = content.get('logtime') 

        if level == 'debug': 

            logserv.debug(logmsg) 

        elif level == 'info': 

            logserv.info(logmsg) 

        elif level == 'warn': 

            logserv.warn(logmsg) 

        elif level == 'error': 

            logserv.error(logmsg) 

        elif level == 'critical': 

            logserv.critical(logmsg) 

        else: 

            log.error('Invalid log level: '+str(level)) 

        yield self.reply_ok(msg) 

class LoggerClient(ServiceClient): 

    """ 

    Class for client to sent log message to service 

    """ 

    def __init__(self, proc=None, **kwargs): 

        if not 'targetname' in kwargs: 

            kwargs['targetname'] = "logger" 

        ServiceClient.__init__(self, proc, **kwargs) 

    @defer.inlineCallbacks 

    def logmsg(self, level, msg, sender, logtime): 

        yield self._check_init() 

        # do we or don't we trust/care about client 

        # timestamps?  If not, the logtime isn't needed. 

        cont = { 

            'level':level, 

            'msg':msg, 

            'sender':sender, 

            'logtime':logtime 

        } 

        (content, headers, msg) = yield self.rpc_send('logmsg', cont) 

        log.info('Service reply: '+str(content)) 

        defer.returnValue(str(content)) 

# Spawn of the process using the module name 

factory = ProcessFactory(LoggerService)