[공유] 파이썬에서 if __name__ == "__main__"을 쓰는 이유

질문

밑의 코드에서 if __name__ == ""__main__""은 왜 쓰는건가요?

소스코드

if __name__ == ""__main__""
    print ""hello""

답변

스크립트가 파이썬 인터프리터 명령어로 패싱되어 실행되면(python myscript.py같이) 다른 언어들과는 다르게, 파이썬은 자동으로 실행되는 메인함수가 없습니다. 파이썬은 메인 함수가 없는 대신 들여쓰기 하지 않은 모든 코드(level 0코드)를 실행합니다 다만, 함수나 클래스는 정의되었지만, 실행되지는 않습니다

질문하신 경우, 최 상위 코드는 if 블록이고, __name__은 현재 모듈의 이름을 담고있는 내장 변수입니다. python myscript.py 같이 이 모듈이 직접 실행되는 경우에만,__name__ 은 "__main__"으로 설정됩니다.

따라서 질문자의 코드가 다른 모듈에 의해 import된 경우 함수와 객체의 정의는 import되지만 __name__이 "__main__"이 아니기 때문에 if문은 실행되지 않습니다.

이처럼,

if __name__ == "__main__":

을 써서, 스크립트가 직접 실행되는지 혹은 import되어 실행되는지 테스트 할 수 있습니다.

예를들어 다음 코드의 경우

file A.py

def func():
    print("function A.py")

print("top-level A.py")

if __name__ == "__main__":
    print("A.py 직접 실행")
else:
    print("A.py가 임포트되어 사용됨")

file B.py

import A as one

print(""top-level in B.py"")
one.func()

if __name__ == "__main__":
    print("B.py가 직접 실행")
else:
    print("B.py가 임포트되어 사용됨")

>> python A.py
top-level in A.py
A.py가 직접 실행


>>python B.py
top-level in A.py
A.py가 임포트되어 사용됨
top-level in B.py
function A.py
B.py가 직접 실행

Reference

https://hashcode.co.kr/questions/3/if-__name__-__main__%EC%9D%80-%EC%99%9C%EC%93%B0%EB%82%98%EC%9A%94

제 글이 도움이 되셨다면 간단하게 '공감', '댓글' 부탁드립니다!