root/ossiedev/branches/jsnyder/trunk/tools/cornetApps/DomainMan/models.py @ 11071

from django.db import models
"""Models for the DomainMan application
-Node makes it easy to extract the ip and port fields
"""


"""Creates a node with ip and port fields
"""
class Node(models.Model):
    ip = models.IPAddressField()
    port=models.IntegerField()
    floor=models.IntegerField()
    num=models.IntegerField()

    """
    Constructor defines its internal ip address and port number for the node
    @param floor: The floor for the node
    @param num:  The number of the node 
    """
    def __init__(self, thisFloor, thisNum):
        self.ip = self.getIP(thisFloor, thisNum)
        self.port = self.getPort(thisFloor,thisNum)
        self.floor = thisFloor
        self.num = thisNum
        
    def __unicode__(self):
        return unicode("node_id: ".__add__(self.floor).__add__("-").__add__(self.num))
    
    """Calculates the internal ip address of the node based on the ICTAS floor and number
    
    @param floor: The floor of ICTAS where the node is located
    @param num:  The number of the node on the particular floor
    @return: The port number used in the CORNET server to access the node; exception if params are not in the range
    """
    def getIP(self, floor, num):
        if floor > 0 and floor < 5:
            if num > 0 and num < 13:
                ip_num=floor*12-2+num
                return "192.168.1.".__add__(str(ip_num))
            else:
                raise Exception("Num input must be greater than 0 and less than 13")
        else:
            raise Exception("Floor input must be greater than 0 and less than 5")
            
    """Calculates the port number based on the floor and node number of the floor
    
    @param floor: The floor of ICTAS where the node is located
    @param num:  The number of the node on the particular floor
    @return: The port number used in the CORNET server to access the node; exception if params are not in the range
    """
    def getPort(self, floor, num):
        if floor > 0 and floor < 5:
            if num > 0 and num < 13:
                port_num=(floor-1)*12+num
                return 7000+port_num
            else:
                raise Exception("Num input must be greater than 0 and less than 13")
        else:
            raise Exception("Floor input must be greater than 0 and less than 5")
        
class SSHClientList():
    clientList = {};
    
    def addClient(self, id, client):
        clientList = {id: client};
    
    def getClient(self, id):
        if clientList.__contains__(id):
            return clientList.get(id)
        else:
            raise Exception("no".__add__(id).__add__("found"))