#!/usr/local/bin/python import cgi, mSQL, os, string, sys; class Domain: def __init__(self, host, db): self.server = host; self.database = db; def doGet(self): self.printHeader(); form = cgi.FieldStorage(); self.printTitle(); print ""; print "

Enter Host Information

"; print "

"; print "Please enter the desired host name as well as the host type."; print "

"; print "
"; print ""; print ""; print ""; print "
"; self.printFooter(); def doPost(self): try: self.printHeader(); form = cgi.FieldStorage(); self.printTitle(); print ""; if not form.has_key("type"): self.printError("You must choose a host type."); return; else: type = string.strip(form["type"].value); if type == 'Host Type': self.printError("You must choose a host type."); return; if not form.has_key("name"): self.printError("You must choose a name for the host."); return; else: host = string.strip(form["name"].value); if host == '': self.printError("You must choose a name for the host."); return; ip = self.selectIP(type, host); if ip == '': self.printError("No more IP's are available for this type."); return; print "

IP for " + host + "

"; print "

"; print "The IP assigned to " + host + " is: " + ip + ""; print "

"; self.printFooter(); except: print "Error!"; print sys.exc_type, sys.exc_value; def printError(self, msg): print "

Error

"; print "

"; print msg; print "

"; self.printFooter(); def printFooter(self): print ""; print ""; def printHeader(self): print "Content-type: text/html"; print; def printTitle(self): print ""; print ""; print "IP Selector"; print ""; def selectIP(self, type, host): sql = "SELECT ip FROM hosts WHERE type = '" + type + "' and name = ''"; db = mSQL.connect(self.server); db.selectdb(self.database); result = db.query(sql); if len(result) < 1: db.close(); return ""; else: ip = result[0][0]; sql = "UPDATE hosts SET name = '" + host + "' WHERE ip = '" + ip + "'"; db.query(sql); db.close(); return ip; if __name__ == '__main__': domain = Domain('athens.imaginary.com', 'db_test'); if not os.environ.has_key('REQUEST_METHOD'): domain.doGet(); elif os.environ['REQUEST_METHOD'] == 'GET': domain.doGet(); else: domain.doPost();