right way to run some code with timeout in Python

后端 未结 9 1719
余生分开走
余生分开走 2020-12-13 18:18

I looked online and found some SO discussing and ActiveState recipes for running some code with a timeout. It looks there are some common approaches:

  • Use threa
相关标签:
9条回答
  • 2020-12-13 19:11

    For "normal" Python code, that doesn't linger prolongued times in C extensions or I/O waits, you can achieve your goal by setting a trace function with sys.settrace() that aborts the running code when the timeout is reached.

    Whether that is sufficient or not depends on how co-operating or malicious the code you run is. If it's well-behaved, a tracing function is sufficient.

    0 讨论(0)
  • 2020-12-13 19:21

    An other way is to use faulthandler:

    import time
    import faulthandler
    
    
    faulthandler.enable()
    
    
    try:
        faulthandler.dump_tracebacks_later(3)
        time.sleep(10)
    finally:
        faulthandler.cancel_dump_tracebacks_later()
    

    N.B: The faulthandler module is part of stdlib in python3.3.

    0 讨论(0)
  • 2020-12-13 19:21

    I've solved that in that way: For me is worked great (in windows and not heavy at all) I'am hope it was useful for someone)

    import threading
    import time
    
    class LongFunctionInside(object):
        lock_state = threading.Lock()
        working = False
    
        def long_function(self, timeout):
    
            self.working = True
    
            timeout_work = threading.Thread(name="thread_name", target=self.work_time, args=(timeout,))
            timeout_work.setDaemon(True)
            timeout_work.start()
    
            while True:  # endless/long work
                time.sleep(0.1)  # in this rate the CPU is almost not used
                if not self.working:  # if state is working == true still working
                    break
            self.set_state(True)
    
        def work_time(self, sleep_time):  # thread function that just sleeping specified time,
        # in wake up it asking if function still working if it does set the secured variable work to false
            time.sleep(sleep_time)
            if self.working:
                self.set_state(False)
    
        def set_state(self, state):  # secured state change
            while True:
                self.lock_state.acquire()
                try:
                    self.working = state
                    break
                finally:
                    self.lock_state.release()
    
    lw = LongFunctionInside()
    lw.long_function(10)
    

    The main idea is to create a thread that will just sleep in parallel to "long work" and in wake up (after timeout) change the secured variable state, the long function checking the secured variable during its work. I'm pretty new in Python programming, so if that solution has a fundamental errors, like resources, timing, deadlocks problems , please response)).

    0 讨论(0)
提交回复
热议问题