translation_service_rpc.py

from lofar.messaging import RPCClientContextManagerMixin, RPCClient, DEFAULT_BROKER, DEFAULT_BUSNAME, DEFAULT_RPC_TIMEOUT
from .config import SPECIFICATIONTRANSLATION_SERVICENAME
import logging
logger = logging.getLogger(__file__)


class TranslationRPC(RPCClientContextManagerMixin):

    def __init__(self, busname=DEFAULT_BUSNAME,
                 broker=DEFAULT_BROKER,
                 timeout=DEFAULT_RPC_TIMEOUT):
        super(TranslationRPC, self).__init__()
        self._rpc_client = RPCClient(SPECIFICATIONTRANSLATION_SERVICENAME, busname, broker, timeout=timeout)


    def trigger_to_specification(self, trigger_spec, trigger_id, job_priority):
        logger.info("Requesting validation of trigger XML")
        result = self._rpc_client.execute('trigger_to_specification',
                          trigger_spec=trigger_spec,
                          trigger_id=trigger_id,
                          job_priority=job_priority)
        logger.info("Received validation result -> " +str(result))
        return result


    def specification_to_momspecification(self, spec):
        logger.info("Requesting validation of trigger XML")
        result = self._rpc_client.exchange('specification_to_momspecification', spec_xml=spec)
        logger.info("Received validation result -> " +str(result))
        return result