[python] 파이썬에서 특정 키로 함수 종료하기

파이썬에서 함수를 실행 중인 도중에 특정 키 입력을 받아 함수를 종료하고 싶을 때가 있습니다. 이를 위해서는 KeyboardInterrupt 예외를 처리하는 방법을 사용할 수 있습니다.

아래의 예제 코드를 살펴봅시다.

import sys

def my_function():
    try:
        while True:
            # 코드 실행
            pass
    except KeyboardInterrupt:
        print("프로그램이 종료되었습니다.")
        sys.exit()

my_function()

위의 코드에서는 my_function이라는 함수를 정의하고, KeyboardInterrupt 예외를 처리하는 부분을 추가했습니다. 함수 내부에서는 무한 루프로 코드를 실행하고 있습니다. 그러나 사용자가 Ctrl+C 키를 입력하면 KeyboardInterrupt 예외가 발생하며 이에 대한 처리가 실행됩니다.

KeyboardInterrupt 예외가 발생하면 “프로그램이 종료되었습니다.”라는 메시지를 출력한 후 sys.exit()를 호출하여 프로그램을 종료합니다.

이 예제 코드를 실행하고 함수가 실행 중인 도중에 Ctrl+C 키를 입력해보면 프로그램이 종료되는 것을 확인할 수 있습니다.

이와 같은 방법을 사용하여 원하는 키를 입력받아 특정 조건일 때 함수를 종료하고, 원하는 처리를 할 수 있습니다.