2012-03-16 153 views
2

我因为听了这个问题张贴在这里:设置超时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.alarmsignal.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)       = ? 
+1

也许'sleep'使用'SIGALRM'内部,这是合理的,并会弄乱你的测试。尝试做一个阻塞I/O操作而不是睡眠(从标准输入读取,不要在控制台输入任何东西)。 – Useless 2012-03-16 16:11:45

+1

它适用于我,同一版本的Ubuntu。你没有忘记'#!/ usr/bin/env python'或'chmod 777 '是不是? – John 2012-03-16 16:14:38

+0

代码工作在我的环境预期(CPython的2.6.5/Ubuntu的9.04) – 2012-03-16 16:16:33

回答

1

如果你想知道这里发生了什么,请尝试:

$ strace -f python timeout.py 

对我来说,(与Python 2.6运行Debian 6)这工作。 strace输出的重要部分:

alarm(3)        = 0 
select(0, NULL, NULL, NULL, {2, 0})  = 0 (Timeout) 
fstat(1, {st_mode=S_IFCHR|0620, st_rdev=makedev(136, 15), ...}) = 0 
mmap(NULL, 4096, PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANONYMOUS, -1, 0) = x7f0fbbe06000 
write(1, "Request 1\n", 10Request 1)    = 10 
alarm(0)        = 1 
rt_sigaction(SIGALRM, {0x4d0a90, [], SA_RESTORER, 0x7f0fbb9deff0}, {0x4d0a90, [], SA_RESTORER, 0x7f0fbb9deff0}, 8) = 0 
alarm(1)        = 0 
select(0, NULL, NULL, NULL, {2, 0})  = ? ERESTARTNOHAND (To be restarted) 
--- SIGALRM (Alarm clock) @ 0 (0) --- 
rt_sigreturn(0xffffffff)    = -1 EINTR (Interrupted system call) 
alarm(0)        = 0 
write(1, "Timeout\n", 8Timeout 
)    = 8 

报警(3)被调用;请求1通过;警报(1)被称为超时。

+0

我已经在编辑中发布了我的'strace'的输出。我真的很困惑,为什么这对我不起作用,但是对于其他人来说 - 似乎在我的'strace'中'write'命令只被调用一次,并且异常不会被抛出。有任何想法吗?谢谢。 – ZenLikeThat 2012-03-16 17:32:30