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

Question

使用 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 个参数，第一个映射到要运行的函数，第二个指定域。

Answer 1

显然 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 静音。