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

from django.db import models
from os import environ
from paramiko import SSHClient, AutoAddPolicy
"""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: " + str(self.floor)+ "-" + str(self.num))
    
    def setClient(self, user, password):
        c = SSHClient()
        c.set_missing_host_key_policy(AutoAddPolicy())
        self.client = c.connect(self.ip, self.port, user, password)
    
    """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 ip address used in the CORNET server to access the node; 
            Through an external network, returns address to cornet server;
            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:
                if environ['DJANGO_SETTINGS_MODULE'] == 'cornetApps.settings':
                    return "128.173.221.40"
                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:
                if environ['DJANGO_SETTINGS_MODULE'] == 'cornetApps.settingsWeb':
                    return 22
                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")
        
"""Connects to a server through a given ip, port and associated username and password 
Returns the SSHClient connected to the server
"""
def connect(ip, port, user, password):
    client = SSHClient()
    client.set_missing_host_key_policy(AutoAddPolicy())
    try:
        client.connect(ip, int(port), user, password)
    except:
        raise Http404(u'Incorrect login information')
    return client