Python NetworkUtil.create_node_network方法代码示例

# 需要导入模块: from ion.agents.platform.util.network_util import NetworkUtil [as 别名]
# 或者: from ion.agents.platform.util.network_util.NetworkUtil import create_node_network [as 别名]
    def test_create_node_network(self):

        # small valid map:
        plat_map = [('R', ''), ('a', 'R'), ]
        pnodes = NetworkUtil.create_node_network(plat_map)
        for p, q in plat_map: self.assertTrue(p in pnodes and q in pnodes)

        # duplicate 'a' but valid (same parent)
        plat_map = [('R', ''), ('a', 'R'), ('a', 'R')]
        NetworkUtil.create_node_network(plat_map)
        for p, q in plat_map: self.assertTrue(p in pnodes and q in pnodes)

        with self.assertRaises(PlatformDefinitionException):
            # invalid empty map
            plat_map = []
            NetworkUtil.create_node_network(plat_map)

        with self.assertRaises(PlatformDefinitionException):
            # no dummy root (id = '')
            plat_map = [('R', 'x')]
            NetworkUtil.create_node_network(plat_map)

        with self.assertRaises(PlatformDefinitionException):
            # multiple regular roots
            plat_map = [('R1', ''), ('R2', ''), ]
            NetworkUtil.create_node_network(plat_map)

        with self.assertRaises(PlatformDefinitionException):
            # duplicate 'a' but invalid (diff parents)
            plat_map = [('R', ''), ('a', 'R'), ('a', 'x')]
            NetworkUtil.create_node_network(plat_map)

# 需要导入模块: from ion.agents.platform.util.network_util import NetworkUtil [as 别名]
# 或者: from ion.agents.platform.util.network_util.NetworkUtil import create_node_network [as 别名]
    def build_network_definition(rsn_oms):
        """
        Creates and returns a NetworkDefinition object reflecting the platform
        network definition reported by the RSN OMS Client object.
        The returned object will have as root the PlatformNode corresponding to the
        actual root of the whole newtork. You can use the `pnodes` property to
        access any node.

        @param rsn_oms RSN OMS Client object.
        @return NetworkDefinition object
        """
        if log.isEnabledFor(logging.DEBUG):
            log.debug("build_network_definition. rsn_oms class: %s",
                      rsn_oms.__class__.__name__)

        # platform types:
        platform_types = rsn_oms.config.get_platform_types()
        if log.isEnabledFor(logging.DEBUG):
            log.debug("got platform_types %s", str(platform_types))

        # platform map:
        map = rsn_oms.config.get_platform_map()
        if log.isEnabledFor(logging.DEBUG):
            log.debug("got platform map %s", str(map))

        # build topology:
        pnodes = NetworkUtil.create_node_network(map)
        dummy_root = pnodes['']
        root_pnode = pnodes[dummy_root.subplatforms.keys()[0]]
        if log.isEnabledFor(logging.DEBUG):
            log.debug("topology's root platform_id=%r", root_pnode.platform_id)

        # now, populate the attributes and ports for the platforms

        def build_attributes_and_ports(pnode):
            """
            Recursive routine to call set_attributes and set_ports on each pnode.
            """
            set_attributes(pnode)
            set_ports(pnode)

            for sub_platform_id, sub_pnode in pnode.subplatforms.iteritems():
                build_attributes_and_ports(sub_pnode)

        def set_attributes(pnode):
            platform_id = pnode.platform_id
            attr_infos = rsn_oms.attr.get_platform_attributes(platform_id)
            if not isinstance(attr_infos, dict):
                raise PlatformDriverException(
                    "%r: get_platform_attributes returned: %s" % (
                    platform_id, attr_infos))

            if log.isEnabledFor(logging.TRACE):
                log.trace("%r: attr_infos: %s", platform_id, attr_infos)

            if not platform_id in attr_infos:
                raise PlatformDriverException(
                    "%r: get_platform_attributes response does not "
                    "include entry for platform_id: %s" %(
                    platform_id, attr_infos))

            ret_infos = attr_infos[platform_id]
            for attrName, attr_defn in ret_infos.iteritems():
                attr = AttrNode(attrName, attr_defn)
                pnode.add_attribute(attr)

        def set_ports(pnode):
            platform_id = pnode.platform_id
            port_infos = rsn_oms.port.get_platform_ports(platform_id)
            if not isinstance(port_infos, dict):
                raise PlatformDriverException(
                    "%r: get_platform_ports response is not a dict: %s" % (
                    platform_id, port_infos))

            if log.isEnabledFor(logging.TRACE):
                log.trace("%r: port_infos: %s", platform_id, port_infos)

            if not platform_id in port_infos:
                raise PlatformDriverException(
                    "%r: get_platform_ports response does not include "
                    "platform_id: %s" % (platform_id, port_infos))

            ports = port_infos[platform_id]

            if not isinstance(ports, dict):
                raise PlatformDriverException(
                    "%r: get_platform_ports: entry for platform_id is "
                    "not a dict: %s" % (platform_id, ports))

            for port_id, dic in ports.iteritems():
                port = PortNode(port_id, dic['network'])
                port.set_state(dic['state'])
                pnode.add_port(port)

                # add connected instruments:
                instrs_res = rsn_oms.instr.get_connected_instruments(platform_id, port_id)
                if not isinstance(instrs_res, dict):
                    log.warn("%r: port_id=%r: get_connected_instruments "
                             "response is not a dict: %s" % (platform_id, port_id, instrs_res))
                    continue
#.........这里部分代码省略.........

# 需要导入模块: from ion.agents.platform.util.network_util import NetworkUtil [as 别名]
# 或者: from ion.agents.platform.util.network_util.NetworkUtil import create_node_network [as 别名]
            subprocess.call(dot_cmd.split())
            subprocess.call(open_cmd.split())
        except Exception, e:
            print "error generating diagrams: %s" % str(e)

    uri = os.getenv('OMS', "http://alice:[email protected]:9021/")

    #####
    import re
    if re.search('simulator', uri):
        from ion.agents.platform.rsn.oms_client_factory import CIOMSClientFactory
        oms = CIOMSClientFactory.create_instance(uri)
        print "ping() = %s"  % oms.ping()
        map = oms.get_platform_map()
        print "get_platform_map() = %s" % map
        pnodes = NetworkUtil.create_node_network(map)
        if not '' in pnodes:
            print "platform map does not include '' to indicate root platforms."
        else:
            dummy_root = pnodes['']
            gen_diagrams('sim_topology', dummy_root)
        sys.exit()
    #####


    proxy = xmlrpclib.ServerProxy(uri)

    retval = proxy.hello.ping()
    print "ping() = %s"  % retval

    retval = proxy.config.get_platform_map()