Python:打印不起作用，脚本无休止地挂起

使用 Python 2.6，我在 Windows XP 中编写了一个脚本。

该脚本执行以下操作:

输入:域名(即:amazon.com)

该脚本通过 dnspython 模块查询 DNS 并返回任何 A 记录 IP 地址。 输出采用使用此数据的特定应用程序所需的特殊格式。

这在 Windows 中运行良好，但是当我将它放在我的 Linux 服务器上时，我得到了一些不寻常且不一致的结果。

第一次运行时，它按预期完成。如果我立即再次运行它，脚本将挂起，什么也不做，没有输出，脚本也不会结束。如果我使用 CTRL-C 退出进程，它会打印出来! (几乎就像它被缓冲了一样，但没有写入终端)

我已经尝试了各种技术来解决这个问题，比如在打印后强制执行 sys.stdout.flush()(尽管无论如何，打印应该自动刷新)但没有成功。

如果我等待一段时间(几分钟)，然后再次运行脚本，它将再次运行(一次)，然后后续尝试将继续失败。我不确定这是怎么回事...还有其他人遇到过这样的事情吗？

Windows 和 Linux (Ubuntu) 上的 Python 2.6。

这是我的脚本:

from dns.resolver import Resolver 
from dns.exception import DNSException 
from cStringIO import StringIO 
import sys 
 
def maltego_transform(entities, messages = ''):     
    print '''<MaltegoMessage> 
<MaltegoTransformResponseMessage> 
<Entities> 
{0} 
</Entities> 
<UIMessages> 
{1} 
</UIMessages> 
</MaltegoTransformResponseMessage> 
</MaltegoMessage>'''.format(entities, messages) 
 
def domain_to_ip(domain): 
    resolver = Resolver() 
    results = [] 
    for type in ['A', 'AAAA']: 
        try: 
            query = resolver.query(domain, type) 
        except DNSException: 
            query = [] 
        results += query 
 
    entities = StringIO() 
    for answer in results: 
        entities.write('''<Entity Type="IPAddress"><Value>{0}</Value></Entity>'''.format(answer)) 
    maltego_transform(entities.getvalue()) 
 
def domain_to_mxdomain(domain): 
    resolver = Resolver() 
    try: 
        query = resolver.query(domain, 'MX') 
    except DNSException: 
        query = [] 
 
    entities = StringIO() 
    for answer in query: 
        entities.write('''<Entity Type="Domain"><Value>{0}</Value> 
<AdditionalFields><Field Name="DomainType" DisplayName="Domain Type">Mail Exchange</Field></AdditionalFields> 
</Entity>'''.format(answer.exchange)) 
    maltego_transform(entities.getvalue()) 
 
def main(): 
    options = {'domain_to_ip' : domain_to_ip, 
               'domain_to_mxdomain' : domain_to_mxdomain} 
    if len(sys.argv) > 2: 
        func = options.get(sys.argv[1], None) 
        if func: 
            func(sys.argv[2]) 
 
if __name__ == '__main__': 
    main() 

使用:python myscript.py domain_to_ip amazon.com

此脚本的 2 个参数，第一个映射到要运行的函数，第二个指定域。

请您参考如下方法:

显然 dnspython 需要 16 个字节的 high-quality random numbers在启动时。获取它们(从/dev/random)可能会阻塞。

如果您按下 Ctrl+C，它实际上会捕捉到 KeyboardInterupt 异常并回退到不太安全的随机数(从当前系统时间获取)。然后你的程序结束运行。

有问题的代码在这里:http://www.dnspython.com/docs/1.7.1/html/dns.entropy-pysrc.html

我想我会认为这是 dnspython 中的错误。它应该找到一种不阻塞那里的方法，并返回到/dev/urandom。在任何情况下，它都不应该使 KeyboardInterrupt 静音。