import FrameWork import EasyDialogs import Res import Dlg import sys import socket import string # # Definitions for our resources ID_MAIN=512 ID_ABOUT=513 ITEM_LOOKUP_ENTRY=1 ITEM_RESULT=2 ITEM_LOOKUP_BUTTON=3 def main(): try: dummy = Res.GetResource('DLOG', ID_MAIN) except Res.Error: try: Res.FSpOpenResFile("dnslookup-2.rsrc", 1) except Res.Error: EasyDialogs.Message("Cannot open dnslookup-2.rsrc") sys.exit(1) DNSLookup() class DNSLookup(FrameWork.Application): "Application class for DNS Lookup" def __init__(self): # First init menus, etc. FrameWork.Application.__init__(self) # Next create our dialog self.main_dialog = MyDialog(self) # Now open the dialog self.main_dialog.open(ID_MAIN) # Finally, go into the event loop self.mainloop() def makeusermenus(self): self.filemenu = m = FrameWork.Menu(self.menubar, "File") self.quititem = FrameWork.MenuItem(m, "Quit", "Q", self.quit) def quit(self, *args): self._quit() def do_about(self, *args): f = Dlg.GetNewDialog(ID_ABOUT, -1) while 1: n = Dlg.ModalDialog(None) if n == 1: return class MyDialog(FrameWork.DialogWindow): "Main dialog window for DNSLookup" def __init__(self, parent): FrameWork.DialogWindow.__init__(self, parent) self.parent = parent def do_itemhit(self, item, event): if item == ITEM_LOOKUP_BUTTON: self.dolookup() def dolookup(self): """Get text entered in the lookup entry area. Place result of the call to dnslookup in the result entry area.""" tp, h, rect = self.dlg.GetDialogItem(ITEM_LOOKUP_ENTRY) txt = Dlg.GetDialogItemText(h) tp, h, rect = self.dlg.GetDialogItem(ITEM_RESULT) Dlg.SetDialogItemText(h, self.dnslookup(txt)) def dnslookup(self, str): """ Perform DNS lookup on str. If first character of digit is numeric, assume that str contains an IP address. Otherwise, assume that str contains a hostname.""" if str == '': str = ' ' if str[0] in string.digits: try: value = socket.gethostbyaddr(str)[0] except: value = 'Lookup failed' else: try: value = socket.gethostbyname(str) except: value = 'Lookup failed' return value main()