如果需要太长时间，如何终止 python 中的 exec() function？

Question

I am trying to make a function that takes in user's code and runs it:

code = input("Your code: ")

def execute(c):
    exec(c)
    
execute(code)

However, if the user enters in an infinite loop, the function runs forever.

Is there a way to terminate the exec() function if the function takes too long? For example, if the code takes longer than 15 seconds to execute, the programme terminates.

Thanks in advance.

Answer 1

There are multiple solutions, For example you could pass a global variable stop condition and raise exception from inside the exec once condition has met. Other solution would be to run exec in a separate thread/process and sending some stop signal once desired.

Answer 2

You can use wrapt_timeout_decorator

from wrapt_timeout_decorator import *

@timeout(15)
def execute(c):
    exec(c)

code = input("Your code: ")
execute(code)

This will throw an exception if the function takes more than 15 sec to finish - TimeoutError: Function execute timed out after 15.0 seconds