senaite.core.astm.consumer

Complexity

Total Complexity

12

Size/Duplication

Total Lines	79
Duplicated Lines	0 %

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	PushConsumer.__init__()	0	3	1
A	PushConsumer.get_sender_name()	0	8	3
B	PushConsumer.process()	0	47	8

# -*- coding: utf-8 -*-

from bika.lims import api
from senaite.core import logger
from senaite.core.interfaces import IASTMImporter
from senaite.jsonapi.interfaces import IPushConsumer
from zope.component import adapter
from zope.component import getMultiAdapter
from zope.component import queryMultiAdapter
from zope.interface import Interface
from zope.interface import implementer


@adapter(Interface)
@implementer(IPushConsumer)
class PushConsumer(object):
    """Adapter that handles push requests for name "enaite.core.lis2a.import"
    """
    def __init__(self, data):
        self.data = data
        self.request = api.get_request()

    def get_sender_name(self, json_data, default=""):
        """Parse the sender name from the header record
        """
        header = json_data.get("H")
        if not isinstance(header, list) and len(header) != 1:
            return default
        sender = header[0].get("sender", {})
        return sender.get("name", default)

    def process(self):
        """Processes the LIS2-A compliant message.
        """
        # Extract LIS2-A messages from the data
        messages = self.data.get("messages")
        if not messages:
            raise ValueError("No messages found: {}".format(repr(self.data)))

        # Just in case we got a message instead of a list of messages
        if api.is_string(messages):
            messages = (messages,)

        logger.info("Received {} messages".format(len(messages)))

        # process messages
        for message in messages:

            json_data = api.parse_json(message, None)
            if json_data is None:
                logger.error("Invalid JSON message: %r" % message)
                continue

            sender_name = self.get_sender_name(json_data)

            # ASTM Data Importer Adapter
            importer = queryMultiAdapter(
                (json_data, message, self.request), IASTMImporter,
                name=sender_name)
            if importer is None:
                importer = getMultiAdapter(
                    (json_data, message, self.request), IASTMImporter)

            # import the data
            try:
                result = importer.import_data()
            except Exception as exc:
                message = "An error occured in '{}.import_data()'".format(
                    importer.__class__.__name__)
                logger.error(exc)
                continue

            if api.is_string(result):
                logger.error("Failed to import ASTM data with importer '{}'"
                             .format(importer.__class__.__name__))
                logger.error("Reason: {}".format(result))

        return True