Python 코드상에서 Interpreter 환경 알아내기

1. 필요한 이유

Python interpreter 환경에서 특정 코드를 실행해야 하는 경우가 발생하였다.

검색해본 결과 몇가지 방법이 있어, 그중 쓸 수 있는 방법을 정리해둔다.

2. 확인해본 방법들

2-1. sys.flags.interactive

import sys

def is_interpreter():
    return bool(sys.flags.interactive)

print(is_interpreter())

# Python IDLE : False  
# Notebook : False
# Python 파일 실행 (intelliJ) : False

이 경우에는 interpreter 환경 구분 불가했다.

2-2. sys.ps1

import sys
print(sys.ps1)

# Python IDLE : >>>
# Notebook : In :
# Python 파일 실행 (intelliJ) : AttributeError: module 'sys' has no attribute 'ps1'

prompt가 없는 interpreter는 구분 불가했고, 예외 처리가 필요한점이 마음에 안들었다..

2-3. hasattr(__main__, "__file__")

import __main__
def is_interpreter():
    return not hasattr(__main__, "__file__")

print(is_interpreter())

# Python IDLE : True
# Notebook : True
# Python 파일 실행 (intelliJ) : False

최종적으로 위 방법으로 interpreter 환경을 구분할 수 있었다.