wally/discover/node.py

import urlparse


class Node(object):

    def __init__(self, conn_url, roles):
        self.roles = roles
        self.conn_url = conn_url
        self.connection = None
        self.monitor_url = None

    def get_ip(self):
        if self.conn_url == 'local':
            return '127.0.0.1'
        return urlparse.urlparse(self.conn_url).hostname

    def get_conn_id(self):
        if self.conn_url == 'local':
            return '127.0.0.1'

        host = urlparse.urlparse(self.conn_url).hostname
        port = urlparse.urlparse(self.conn_url).port

        if port is None:
            port = 22

        return host + ":" + str(port)

    def __str__(self):
        templ = "<Node: url={conn_url!r} roles={roles}" + \
                " connected={is_connected}>"
        return templ.format(conn_url=self.conn_url,
                            roles=", ".join(self.roles),
                            is_connected=self.connection is not None)

    def __repr__(self):
        return str(self)