"""
Library for the robot based system test tool of the OpenDaylight project.
Authors: Baohua Yang@IBM, Denghui Huang@IBM
Updated: 2013-11-10
"""
from robot.libraries.BuiltIn import BuiltIn


class Topology(object):
    '''
    Topology class provide topology database and provide many method to get property of topology.
    '''
    topo_nodes_db = [
        [],
        [{u'type': u'OF', u'id': u'00:00:00:00:00:00:00:01'}],
        [{u'type': u'OF', u'id': u'00:00:00:00:00:00:00:01'},
         {u'type': u'OF', u'id': u'00:00:00:00:00:00:00:02'},
         {u'type': u'OF', u'id': u'00:00:00:00:00:00:00:03'}]]

    def __init__(self):
        self.builtin = BuiltIn()

    def get_nodes_from_topology(self, topo_level):
        '''
        get nodes from topology database by topology tree level
        '''
        if isinstance(topo_level, str) or isinstance(topo_level, unicode):
            if topo_level.isdigit():
                topo_level = int(topo_level)
                if topo_level <= 0:
                    return None
                return self.topo_nodes_db[topo_level]
            else:
                return None
        elif isinstance(topo_level, int):
            if topo_level <= 0:
                return None
            return self.topo_nodes_db[topo_level]
        else:
            return None

if __name__ == '__main__':
    topology = Topology()
    print topology.get_nodes_from_topology(2)
    print topology.get_nodes_from_topology('2')