exit the pythonic way

This commit is contained in:
Babak Farrokhi 2016-05-09 13:33:32 +04:30
parent 6715741f66
commit b2dab7affc
Signed by: farrokhi
GPG Key ID: 6B267AD85D632E9A

View File

@ -115,13 +115,13 @@ def usage():
print(' -t --type DNS request record type (default: A)')
print(' -C --color Print colorful output')
print(' ')
exit()
sys.exit()
def signal_handler(sig, frame):
global shutdown
if shutdown: # pressed twice, so exit immediately
exit(0)
sys.exit(0)
shutdown = True # pressed once, exit gracefully
@ -166,11 +166,11 @@ def ping(resolver, hostname, dnsrecord, ttl):
except dns.resolver.NoNameservers as e:
if not quiet:
print("no or bad response:", e)
exit(1)
sys.exit(1)
except dns.resolver.NXDOMAIN as e:
if not quiet:
print("Invalid hostname:", e)
exit(1)
sys.exit(1)
except dns.resolver.Timeout:
pass
except dns.resolver.NoAnswer:
@ -181,7 +181,7 @@ def ping(resolver, hostname, dnsrecord, ttl):
pass
except:
print("unxpected error: ", sys.exc_info()[0])
exit(1)
sys.exit(1)
else:
reached = True
@ -282,7 +282,7 @@ def main():
icmp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM, icmp)
except OSError:
print("Error: Unable to create ICMP socket with unprivileged user. Please run as root.")
exit(1)
sys.exit(1)
icmp_socket.bind(("", dest_port))
icmp_socket.settimeout(timeout)
@ -337,7 +337,7 @@ def main():
as_name = "[%s %s] " % (ASN.asn, ASN.owner)
except AttributeError:
if shutdown:
exit(0)
sys.exit(0)
pass
c = color.N # default