diff -r 4177eba88c22 -r 44522cd37b07 whois/__init__.py --- a/whois/__init__.py Fri Apr 15 09:21:42 2016 -0600 +++ b/whois/__init__.py Fri Apr 15 18:29:24 2016 -0600 @@ -51,17 +51,18 @@ return socket.gethostbyaddr(url)[0] tlds_path = os.path.join(os.getcwd(), os.path.dirname(__file__), 'data', 'tlds.txt') - suffixes = [ - line.lower().strip().encode('utf-8') - for line in open(tlds_path).readlines() - if not line.startswith('#') - ] + with open(tlds_path) as tlds_fil: + suffixes = [line.lower().encode('utf-8') + for line in (x.strip() for x in tlds_fil) + if not line.startswith('#')] + suff = 'xn--p1ai' if not isinstance(url, str): url = url.decode('utf-8') - url = re.sub(b'^.*://', b'', url.encode('idna')).split(b'/')[0].lower() + url = re.sub('^.*://', '', url) + url = url.split('/')[0].lower().encode('idna') + domain = [] - for section in url.split(b'.'): if section in suffixes: domain.append(section)