mockDisplayModule.py

#!/usr/bin/env python3

import os
import robomodules as rm
from messages import message_buffers, MsgType

ADDRESS = os.environ.get("BIND_ADDRESS","localhost")
PORT = os.environ.get("BIND_PORT", 11297)

FREQUENCY = 10

class MockGuiModule(rm.ProtoModule):
    def __init__(self, addr, port):
        self.subscriptions = [MsgType.MOCK_MSG]
        super().__init__(addr, port, message_buffers, MsgType, FREQUENCY, self.subscriptions)
        self.value = -1
        self.sub_ticks = 0
        self.subbed = True

    def msg_received(self, msg, msg_type):
        # This gets called whenever any message is received
        if msg_type == MsgType.MOCK_MSG:
            self.value = msg.mockValue

    def tick(self):
        # this function will get called in a loop with FREQUENCY frequency
        # for this mock module we will print out the current value
        print('Current value: {}'.format(self.value))

        # to demonstrate subscription and unsubscription,
        # we will periodically unsubscribe and resubscribe
        if self.sub_ticks > 100:
            if self.subbed:
                print('Unsubscribed!')
                self.unsubscribe([MsgType.MOCK_MSG])
            else:
                print('Subscribed!')
                self.subscribe([MsgType.MOCK_MSG])
            self.subbed = not self.subbed
            self.sub_ticks = 0
        self.sub_ticks += 1


def main():
    module = MockGuiModule(ADDRESS, PORT)
    module.run()

if __name__ == "__main__":
    main()