Source code for darwinpush.messages.StationMessage

from darwinpush.messages.BaseMessage import BaseMessage

from darwinpush.xb.sm import MsgCategoryType, MsgSeverityType
import darwinpush.xb.sm as sm

from pyxb.binding.basis import ElementContent, NonElementContent

import enum

class UnknownStationMessageCategoryError(Exception):
    pass


class UnknownStationMessageSeverityError(Exception):
    pass


class StationMessageCategory(enum.Enum):
    Train = 0
    Station = 1
    Connection = 2
    System = 3
    Misc = 4
    PriorTrain = 5
    PriorOther = 6

    @staticmethod
    def from_xb(x):
        if x == MsgCategoryType.Train:
            return StationMessageCategory.Train
        if x == MsgCategoryType.Station:
            return StationMessageCategory.Station
        if x == MsgCategoryType.Connections:
            return StationMessageCategory.Connection
        if x == MsgCatgoryType.System:
            return StationMessageCategory.System
        if x == MsgCategoryType.Misc:
            return StationMessageCategory.Misc
        if x == MsgCategoryType.PriorTrains:
            return StationMessageCategory.PriorTrain
        if x == MsgCategoryType.PriorOther:
            return StationMessageType.PriorOther

        raise UnknownStationMessageCategoryError("Unknown Station Message Category '{}' received in Station Message.".format(x))


class StationMessageSeverity(enum.Enum):
    LevelZero = 0
    LevelOne = 1
    LevelTwo = 2
    LevelThree = 3

    @staticmethod
    def from_xb(x):
        if x == MsgSeverityType.n0:
            return StationMessageSeverity.LevelZero
        if x == MsgSeverityType.n1:
            return StationMessageSeverity.LevelOne
        if x == MsgSeverityType.n2:
            return StationMessageSeverity.LevelTwo
        if x == MsgSeverityType.n3:
            return StationMessageSeverity.LevelThree

        raise UnknownStationMessageSeverityError("Unknown Station Message Severity '{}' received in Station Message.".format(x))


class UnknownXbOrderedContentItemError(Exception):
    pass


class StationMessageContent:

    def __init__(self, raw):
        self.raw = raw

    def __str__(self):
        s = ""
        for i in self.raw.orderedContent():
            if type(i) == NonElementContent:
                s += i.value
            elif type(i) == ElementContent:
                if type(i.value) == sm.CTD_ANON_2:
                    s += "<a href=\""+i.value.href+"\">"+i.value.value()+"</a>"
                elif type(i.value) == sm.CTD_ANON_:
                    s += "<p>"
                    for j in i.value.orderedContent():
                        if type(j) == NonElementContent:
                            s += j.value
                        elif type(j) == ElementContent:
                            if type(j.value) == sm.CTD_ANON_2:
                                s += "<a href=\""+j.value.href+"\">"+j.value.value()+"</a>"
                            else:
                                raise UnknownXbOrderedContentItemError("Unrecognised ElementType in StationMessage.Msg.P.OrderedContent context: {}".format(j.value))
                        else:
                            raise UnknwonXbOrderedContentItemError("Unknown pyxb orderedContent item {} in msg.P.".format(j))
                    s += "</p>"
                else:
                    raise UnknownXbOrderedContentItemError("Unrecognised ElementType in StationMessage.Msg.OrderedContent context: {}.".format(i.value))
            else:
                raise UnknownXbOrderedContentItemError("Unknown pyxb orderedContent item {}.".format(i))
        return s

    def __repr__(self):
       return self.__str__()


class StationMessage(BaseMessage):

    def __init__(self, message, containing_message, xml):
        super().__init__(message, containing_message, xml)
        self._build_stations_list()
        self._build_message()

    def _build_stations_list(self):
        self._stations = []
        for s in self.raw.Station:
            self._stations.append(s.crs)

    def _build_message(self):
        self._message = StationMessageContent(self.raw.Msg)

    @property
    def stations(self):
        return self._stations

    @property
    def message(self):
        return self._message

    @property
    def smid(self):
        return self.raw.id

    @property
    def category(self):
        return StationMessageCategory.from_xb(self.raw.cat)

    @property
    def severity(self):
        return StationMessageSeverity.from_xb(self.raw.sev)