我因为听了这个问题张贴在这里:设置超时mechanize.Browser
What should I do if socket.setdefaulttimeout() is not working?
,试图拿出一个解决方案,杀死请求时,我mechanize.Browser
对象花费的时间太长了,我有在尝试用在托马斯的编辑第一个解决方案(转贴在这里为清楚起见):
import signal, time
def request(arg):
"""Your http request"""
time.sleep(2)
return arg
class Timeout():
"""Timeout class using ALARM signal"""
class Timeout(Exception): pass
def __init__(self, sec):
self.sec = sec
def __enter__(self):
signal.signal(signal.SIGALRM, self.raise_timeout)
signal.alarm(self.sec)
def __exit__(self, *args):
signal.alarm(0) # disable alarm
def raise_timeout(self, *args):
raise Timeout.Timeout()
# Run block of code with timeouts
try:
with Timeout(3):
print request("Request 1")
with Timeout(1):
print request("Request 2")
except Timeout.Timeout:
print "Timeout"
# Prints "Request 1" and "Timeout"
当我从我的使用终端运行这个python timeout.py
(版本是Python 2.7.2+
,我在Ubuntu 11.10上eiric山猫),没有例外thrown-相反,它只是简单地打印
Request 1
Request 2
可能有人请解释如何解决这一问题?对这些signal.alarm
和signal.signal
调用进行的解释也很棒。
非常感谢您的宝贵时间!
编辑:
运行strace -f python timeout.py
产量:
alarm(3) = 0
select(0, NULL, NULL, NULL, {2, 0}) = 0 (Timeout)
fstat64(1, {st_mode=S_IFREG|0664, st_size=0, ...}) = 0
mmap2(NULL, 4096, PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANONYMOUS, -1, 0) = 0xb740c000
alarm(0) = 1
rt_sigaction(SIGALRM, {0x812f450, [], 0}, {0x812f450, [], 0}, 8) = 0
alarm(1) = 0
select(0, NULL, NULL, NULL, {2, 0}) = 0 (Timeout)
alarm(0) = 0
rt_sigaction(SIGINT, {SIG_DFL, [], 0}, {0x812f450, [], 0}, 8) = 0
rt_sigaction(SIGALRM, {SIG_DFL, [], 0}, {0x812f450, [], 0}, 8) = 0
write(1, "Request 1\nRequest 2\n", 20) = 20
exit_group(0) = ?
也许'sleep'使用'SIGALRM'内部,这是合理的,并会弄乱你的测试。尝试做一个阻塞I/O操作而不是睡眠(从标准输入读取,不要在控制台输入任何东西)。 – Useless 2012-03-16 16:11:45
它适用于我,同一版本的Ubuntu。你没有忘记'#!/ usr/bin/env python'或'chmod 777'是不是? –
John
2012-03-16 16:14:38
代码工作在我的环境预期(CPython的2.6.5/Ubuntu的9.04) – 2012-03-16 16:16:33