使用Python 2.6,我在Windows XP中编写了一个脚本。Python:打印不起作用,脚本无休止地挂起
的脚本执行以下操作:
输入:域名(例如: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()
用途:蟒蛇myscript.py domain_to_ip amazon.com
2个参数,这个剧本,第一个映射到函数运行,第二个指定该域名。
Jason的回答是正确的,当阅读/ dev/random时脚本挂起。 我评论了/ dev/random读出,以便它总是通过time.time()生成它的编号,并且它工作正常。 – brildum 2009-11-30 14:38:31