Python: print无效，脚本持续挂起

Question

我在Windows XP上用Python 2.6写了一个脚本。

这个脚本的功能是：

输入：域名（比如：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

这个脚本有两个参数，第一个是要运行的功能，第二个是指定的域名。

Linux windows 脚本进程管理 dns dnspython 输出缓存系统挂起

Python: print无效，脚本持续挂起

2 个回答

撰写回答