import sys
import dnslib
import dnstype
import dnsopcode
import dnsclass
import socket
import select

def main():
    server = 'cnri.reston.va.us'	# How?
    port = 53
    opcode = dnsopcode.QUERY
    rd = 0
    qtype = dnstype.MX
    qname = sys.argv[1:] and sys.argv[1] or 'www.python.org'
    m = dnslib.Mpacker()
    m.addHeader(0,
		0, opcode, 0, 0, rd, 0, 0, 0,
		1, 0, 0, 0)
    m.addQuestion(qname, qtype, dnsclass.IN)
    request = m.getbuf()
    s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    s.connect((server, port))
    s.send(request)
    while 1:
	r, w, x = [s], [], []
	r, w, x = select.select(r, w, x, 0.333)
	print r, w, x
	if r:
	    reply = s.recv(1024)
	    u = dnslib.Munpacker(reply)
	    dnslib.dumpM(u)
	    break

main()