python 실행 vs module 실행/ jupyter notebook 실행 vs shell 실행 self check

- python 파일 실행 vs module로 import 실행

 

파이썬에 argument 입력하고 싶을때 등장하는 것이 

 

if __name__ == “__main__” : 
	print('바로 py파일 실행할때')
else : 
	print('import 해서 모듈처럼 사용할때')

 

그런데 나는 인터프리터에서 사용 vs terminal에서 실행  이 둘을 구별하고 싶은데 그이유는 ipynb 파일에서 예시로 몇개 실행해보고 py로 바꾸기 때문이다. 

즉 내가 원하는 두 케이스다 __name__ == “__main__”  로 인식된다.

 

 

- 인터프리터에서 사용 vs terminal에서 실행

 

def isnotebook():
    try:
        shell = get_ipython().__class__.__name__
        if shell == 'ZMQInteractiveShell':
            return True   # Jupyter notebook or qtconsole
        elif shell == 'TerminalInteractiveShell':
            return False  # Terminal running IPython
        else:
            return False  # Other type (?)
    except NameError:
        return False      # Probably standard Python interpreter

최종적으로 활용예시는 다음과 같다

if isnotebook():
    ## jupyter notebook 내에서만 실행
else :
    ## shell에서 실행
    if len(sys.argv) != 2:
        print("Insufficient arguments")
    else : 
        {$variable} = sys.argv[1]

 

Reference:

https://stackoverflow.com/questions/15411967/how-can-i-check-if-code-is-executed-in-the-ipython-notebook

 

https://medium.com/@chullino/if-name-main-%EC%9D%80-%EC%99%9C-%ED%95%84%EC%9A%94%ED%95%A0%EA%B9%8C-bc48cba7f720