команда Run с ограниченным временем выполнения

Question

Если бы я был запустить некоторые Python через командную строку сказать, например, что-то вроде этого:

cat <<'PYSTUFF' | python 
print "Hi" 
print "There" 
print "Friend" 
PYSTUFF 

Это прекрасно работает и выводит отклик РЕПЛ. Я хочу ограничить выполнение этой команды. Например, если я написал:

cat <<'PYSTUFF' | python 
while(True): 
    print "Oh no!" 
PYSTUFF 

Это было бы неплохо и в конечном итоге что-то сработало. Как ограничить выполнение, чтобы сказать «если это занимает больше, чем х времени, убить его»? Я попытался использовать ulimit -t 2, но это, похоже, не достигает того, что я хочу.

Answer 1

ограничить выполнение этой команды.

Начиная с Python 2.6+ можно использовать multiprocessing модуль:

cat <<'PYSTUFF' | python 
from multiprocessing import Process 
from time import sleep 

def task(): 
    while True: 
     print "oh no!" 
     sleep(0.5) 

p = Process(target=task) 
p.start() 
p.join(1) 
print "*** Aborted" 
p.terminate() 
PYSTUFF 

Answer 2

Если вы хотите, чтобы контролировать это из оболочки, запустите сценарий питона на задний план, сон на некоторое время, а затем убить фоновый процесс:

# a sample command that generates some output 
{ while :; do date; sleep 1; done; } & 

# let it run for a while, see if it's still running, and kill it if it is 
sleep 10 && kill -0 $! 2>/dev/null && kill $! 

Answer 3

Благодаря @Biffen Использование команды timeout работала отлично. На mac я использовал gtimeout. Вся моя команда выглядит так:

cat <<'PYSTUFF' | gtimeout 0.5 python 
while(True): print("hi") 
PYSTUFF