python - 连接服务器时如何避免程序卡住

Question

我有一个小脚本可以过滤那些尚未注册的域名。我使用 pywhois 模块。问题是它在几次(有时是数百个)请求后突然卡住并且不执行任何操作。我认为这不是禁令，因为我可以在卡住后立即运行该程序并且它可以工作。

我想避免这种卡住。我的想法是计算函数的运行时间，如果时间跨越某条线(例如 10 秒)，它将重复代码。

您对如何避免卡住有什么建议吗？或者检查域的更好方法？

这是代码:

for keyword in keywords:
        try:
            details = pythonwhois.get_whois(keyword+'.com')
        except Exception as e:
            print e
            continue
        if 'status' not in details.keys():
            print 'Free domain!'
            print keyword

Answer 1

此方法很容易发生变化(如果底层库发生变化)，但是，您可以调用内部套接字函数来为所有 pythonwhois 网络调用设置超时。例如:

TIMEOUT = 5.0 # timeout in seconds
pythonwhois.net.socket.setdefaulttimeout(TIMEOUT)
pythonwhois.get_whois("example.com")