diff --git a/from_dict/check_from_dict.py b/from_dict/check_from_dict.py
index e701ab5..0f20f46 100644
--- a/from_dict/check_from_dict.py
+++ b/from_dict/check_from_dict.py
@@ -5,6 +5,9 @@
import urllib
import os
import time
+import socket
+
+socket.setdefaulttimeout(60)
domain_suffix = '.com'
@@ -33,7 +36,11 @@ def get_words_from_file(words_file=words_file, init_word_line=0):
def is_ava(domain):
"""判断该域名是否被注册"""
- data = urllib.urlopen(api % domain).read()
+ try:
+ data = urllib.urlopen(api % domain).read()
+ except:
+ return False
+
ava_pattern = re.compile(r'(.*) : .*')
result = ava_pattern.findall(data)
if '210' in result:
diff --git a/reverse/check_domain.py b/reverse/check_domain.py
index 932dcdb..2b10615 100644
--- a/reverse/check_domain.py
+++ b/reverse/check_domain.py
@@ -9,7 +9,9 @@
import re
import os
import pickle
+import socket
+socket.setdefaulttimeout(60)
api = "http://panda.www.net.cn/cgi-bin/check.cgi?area_domain=%s" # api地址
string = "abcdefghigklmnopqrstuvwxyz" # 所有字母
string_len = len(string) # 长度
@@ -48,7 +50,10 @@ def num_2_string(name, string):
def is_ava(domain):
"""判断该域名是否被注册"""
- data = urllib.urlopen(api % domain).read()
+ try:
+ data = urllib.urlopen(api % domain).read()
+ except:
+ return False
ava_pattern = re.compile(r'(.*) : .*')
perm_pattern = re.compile(r'Forbidden')
result = ava_pattern.findall(data)