farrokhi/dnsdiag
1
0
Fork 0
Code Issues Pull Requests Releases Activity

improve cli option parsing

Browse Source
This commit is contained in:
Babak Farrokhi 2016-02-03 16:08:11 +03:30
parent 7172108353
commit f3761993a6
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
dnsping.py
View File

@ -55,10 +55,13 @@ def main():
# signal.signal(signal.SIGTSTP, signal.SIG_IGN) # ignore CTRL+Z # signal.signal(signal.SIGTSTP, signal.SIG_IGN) # ignore CTRL+Z
# signal.signal(signal.SIGINT, signal.SIG_IGN) # ignore CTRL+C # signal.signal(signal.SIGINT, signal.SIG_IGN) # ignore CTRL+C
hostname = 'wikipedia.org' if len(sys.argv) == 1:
usage()
dnsrecord = 'A' dnsrecord = 'A'
count = 10 count = 10
timeout = 5 timeout = 5
quiet = False
verbose = False verbose = False
dnsserver = '8.8.8.8' dnsserver = '8.8.8.8'
@ -71,9 +74,10 @@ def main():
usage() usage()
exit(2) exit(2)
quiet = False if args and len(args) == 1:
hn = [h for h in sys.argv[-1:] if not h.startswith('-')] hostname = args[0]
hostname = hn[0] else:
usage()
for o, a in opts: for o, a in opts:
if o in ("-h", "--help"): if o in ("-h", "--help"):
@ -130,8 +134,8 @@ def main():
response_time.append(elapsed) response_time.append(elapsed)
if not quiet: if not quiet:
print( print(
"%d bytes from %s: seq=%-3d time=%3.3f ms" % ( "%d bytes from %s: seq=%-3d time=%3.3f ms" % (
len(str(answers.rrset)), dnsserver, i, elapsed)) len(str(answers.rrset)), dnsserver, i, elapsed))
if verbose: if verbose:
print(answers.rrset) print(answers.rrset)