bbgo_origin/python/bbgo/handlers/order.py

from ..data import UserDataEvent
from ..enums import ChannelType
from ..enums import EventType
from .handler import Handler


class OrderHandler(Handler):

    def __call__(self, event: UserDataEvent) -> None:
        if event.channel_type != ChannelType.ORDER:
            return

        super(OrderHandler, self).__call__(event)


class OrderSnapshotHandler(OrderHandler):

    def __call__(self, event: UserDataEvent) -> None:
        if event.event_type != EventType.SNAPSHOT:
            return

        super(OrderSnapshotHandler, self).__call__(event)


class OrderUpdateHandler(OrderHandler):

    def __call__(self, event: UserDataEvent) -> None:
        if event.event_type != EventType.UPDATE:
            return

        super(OrderUpdateHandler, self).__call__(event)