파이썬에서 특정 작업을 실행할 때 일정 시간내에 완료되지 않으면 예외를 발생시킬 수 있습니다. 어떤 작업이 대해 시간을 제한하고 싶을 때 사용할 수 있겠습니다. try/except 구문과 signal.alarm() 함수를 이용합니다. 예제 코드는 아래와 같습니다.
import time
import signal
import traceback
# 알람 처리 함수
def alarm_handler(signum, frame):
# 예외 발생시키기
raise Exception("시간 초과")
# 루프 함수 (n초 동안 대기)
def loop_func(num):
for i in range(num):
print(i)
time.sleep(1)
# alarm signal이 발생할 경우 처리할 함수 지정
signal.signal(signal.SIGALRM, alarm_handler)
try:
# 5초 동안 작업이 완료되지 않으면 예외 발생
signal.alarm(5)
# 함수 호출
loop_func(10)
except:
print("예외 발생")
traceback.print_exc
# 시간 제한 해제
signal.alarm(0)
위의 예시를 보면 5초안에 작업이 완료되지 않으면 예외를 발생하도록 한 후, loop_func 함수를 호출하여 10초 동안 대기하도록 했습니다. 따라서 5초후에 예외가 발생하게 됩니다. 만약 예를 들어 3초 동안 대기하도록 loop_func 함수의 인자를 3으로 지정하여 호출했다면 예외 발생이 없이 정상적으로 완료될 것입니다.
여기서 유의할 점이 있는데, signal.alarm()에 대한 처리 함수를 만들고 이 함수를 이용해서 예외를 발생시키도록 해 주어야 합니다. 위의 코드에서는 alarm_handler() 함수를 만들고 signal.signal() 함수에서 해당 함수를 지정해 주었습니다. 이렇게 하지 않으면 알람이 발생하는 시점에서 프로그램이 그대로 종료되므로, 위와 같이 별도의 함수를 지정하여 예외를 발생시키도록 해 줍니다.
또 한가지 유의할 점이 있습니다. 시간제한을 적용하고 싶은 작업이 완료되었으면 시간 제한을 해제해 주어야 합니다. signal.alarm() 함수에서 인자를 0으로 지정하면 됩니다. 해제하지 않으면 이후 다른 작업을 진행하는 도중에 제한 시간이 만료되어 예외가 발생할 수 있습니다.
이와 관련된 문서는 아래의 링크에 있습니다.
https://docs.python.org/ko/3.8/library/signal.html
https://docs.python.org/3/library/signal.html