From 0b068b849b01cda83d3bbd1268daa64829b740ea Mon Sep 17 00:00:00 2001 From: Babak Farrokhi Date: Fri, 8 Apr 2016 12:32:50 +0430 Subject: [PATCH] beautify code --- dnstraceroute.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/dnstraceroute.py b/dnstraceroute.py index af64888..366dab0 100755 --- a/dnstraceroute.py +++ b/dnstraceroute.py @@ -27,8 +27,8 @@ import getopt import signal -import sys import socket +import sys import time import dns.rdatatype @@ -127,9 +127,9 @@ def main(): icmp_socket.bind(("", dnsport)) icmp_socket.settimeout(timeout) - try: # send DNS request + try: # send DNS request stime = time.time() - answers = resolver.query(hostname, dnsrecord, ipttl = ttl) + answers = resolver.query(hostname, dnsrecord, ipttl=ttl) except dns.resolver.NoNameservers as e: if not quiet: print("no or bad response:", e) @@ -153,7 +153,7 @@ def main(): curr_addr = None curr_host = None - try: # expect ICMP response + try: # expect ICMP response _, curr_addr = icmp_socket.recvfrom(512) curr_addr = curr_addr[0] @@ -182,12 +182,13 @@ def main(): if curr_addr: print("%d\t%s (%s) %d ms" % (ttl, curr_name, curr_addr, elapsed)) else: - print("%d\t *" % ttl) + print("%d\t *" % ttl) ttl += 1 hops += 1 if (hops >= count) or (curr_addr == dnsserver) or reached: break + if __name__ == '__main__': main()