from tfw.internals.networking import ZMQConnector, ZMQUplinkConnector
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(ZMQUplinkConnector, ConnAddrMixin):
    def __init__(self):
        super().__init__(self.uplink_conn_addr)


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