baseimage-tutorial-framework/tfw/main/tfw_connector.py

from tfw.internals.networking import ServerConnector, ServerUplinkConnector
from tfw.config import TFWENV


class ConnAddrMixin:
    @property
    def uplink_conn_addr(self):
        return f'tcp://localhost:{TFWENV.PULL_PORT}'

    @property
    def downlink_conn_addr(self):
        return f'tcp://localhost:{TFWENV.PUB_PORT}'


class TFWUplinkConnector(ServerUplinkConnector, ConnAddrMixin):
    def __init__(self):
        super().__init__(self.uplink_conn_addr)


class TFWConnector(ServerConnector, ConnAddrMixin):
    def __init__(self):
        super().__init__(
            self.downlink_conn_addr,
            self.uplink_conn_addr
        )