Source code for spinnman.connections.udp_packet_connections.ip_address_connection

from .udp_connection import UDPConnection
from spinnman.constants import UDP_BOOT_CONNECTION_DEFAULT_PORT


class IPAddressesConnection(UDPConnection):
    """ A connection that detects any UDP packet that is transmitted by \
        spinnaker boards prior to boot
    """

    def __init__(self, local_host=None,
                 local_port=UDP_BOOT_CONNECTION_DEFAULT_PORT):
        UDPConnection.__init__(self, local_host=local_host,
                               local_port=local_port)

    def supports_sends_message(self, message):  # @UnusedVariable
        return False

    def receive_ip_address(self, timeout=None):
        try:
            (_, (ip_address, port)) = self.receive_with_address(timeout)
            if port == 54321:
                return ip_address
        except Exception:
            pass
        return None

    def __repr__(self):
        return \
            "IPAddressesConnection(local_host={}, local_port={})".format(
                self.local_ip_address, self.local_port)