убить потоки позже время в python

Question

У меня есть код python с потоками, и мне нужно, если в, например, 1 час потоки не закончены, закончить все потоки и завершить скрипт, а если час не будет завершен, подождите что все мои нити заканчиваются.

Я пытаюсь с потоком демона и с часом, а если час завершен, используйте: sys.exit(), но он не работает для меня, потому что всегда жду моего сна threadh, тогда мой скрипт ждет, пока нить завершена и sys.exit() не работает.

import socket, threading, time, sys 
from sys import argv 
import os 

acc_time=0 
transactions_ps=5 

ins = open(sys.argv[1],'r') 
msisdn_list = [] 
for line in ins: 
    msisdn_list.append (line.strip('\n')) 
    # print line 
ins.close() 


def worker(msisdn_list): 
    semaphore.acquire() 
    global transactions_ps 
    print " ***** ", threading.currentThread().getName(), "Lanzado" 
    count=1 
    acc_time=0 
    print "len: ",len(msisdn_list) 
    for i in msisdn_list: 
     try: 
      init=time.time() 
      time.sleep(2) 
      print "sleeping...",i 
      time.sleep(4) 
      final=time.time() 
      acc_time = acc_time+final-init 
      print acc_time 
     except IOError: 
       print "Connection failed",sys.exc_info()[0] 

    print "Deteniendo ",threading.currentThread().getName() 
    semaphore.release() 
def kill_process(secs_to_die): 
    time.sleep(secs_to_die) 
    sys.exit() 

seconds_to_die=3600 

thread_kill = threading.Thread(target = kill_process, args=(seconds_to_die,)) 
thread_kill.start() 

max_con=5 
semaphore = threading.BoundedSemaphore(max_con) 
for i in range(0,28,transactions_ps): 
    w = threading.Thread(target=worker, args=(msisdn_list[i:i+transactions_ps-1],)) 
    w.setDaemon(True) 
    w.start() 

Как сделать это

Answer 1

Вы можете обратиться к этой реализации KTHREAD:

http://python.todaysummary.com/q_python_45717.html

Answer 2

Минимальное изменение в код, который бы решить проблему является threading.Barrier:

barrier = Barrier(number_of_threads, timeout=3600) 
# create (number_of_threads - 1) threads, pass them barrier 
# each thread calls barrier.wait() on exit 
barrier.wait() # after number_of_threads .wait() calls or on timeout it returns 

Проще альтернатива использовать multiprocessing.dummy.Pool, что создает демон темы:

from multiprocessing.dummy import Pool # use threads 

start = timer() 
endtime = start + 3600 
for result in pool.imap_unordered(work, args): 
    if timer() > endtime: 
     exit("timeout") 

код не тайм-аут, пока элемент работы не будет сделано то, что ожидает, что обработка одного элемента из списка не займет много времени.

Полный пример:

#!/usr/bin/env python3 
import logging 
import multiprocessing as mp 
from multiprocessing.dummy import Pool 
from time import monotonic as timer, sleep 

info = mp.get_logger().info 

def work(i): 
    info("start %d", i) 
    sleep(1) 
    info("end %d", i) 

seconds_to_die = 3600 
max_con = 5 
mp.log_to_stderr().setLevel(logging.INFO) # enable logging 
pool = Pool(max_con) # no more than max_con at a time 
start = timer() 
endtime = start + seconds_to_die 
for _ in pool.imap_unordered(work, range(10000)): 
    if timer() > endtime: 
     exit("timeout")