code cleanup

This commit is contained in:
Babak Farrokhi 2016-04-10 19:23:20 +04:30
parent b91962ab72
commit add9b1f782
Signed by: farrokhi
GPG Key ID: 6B267AD85D632E9A

View File

@ -28,7 +28,6 @@
import getopt
import ipaddress
import os
import re
import signal
import socket
import sys
@ -115,9 +114,9 @@ def main():
# check if we have a valid dns server address
try:
ipaddress.ip_address(dnsserver)
except ValueError: # so it is not a valid IPv4 or IPv6 address, so try to resolve host name
except ValueError: # so it is not a valid IPv4 or IPv6 address, so try to resolve host name
try:
dnsserver = socket.getaddrinfo(dnsserver, port = None)[1][4][0]
dnsserver = socket.getaddrinfo(dnsserver, port=None)[1][4][0]
except OSError:
print('Error: cannot resolve hostname:', dnsserver)
exit(1)