[Python] 모듈

🍔 모듈

• 함수나 변수 또는 클래스를 모아 놓은 파일
• 다른 파이썬 프로그램에서 불러와 사용할 수 있게끔 만든 파이썬 파일

1. 모듈 만들기

# mod1.py
def add(a, b):
    return a + b

def sub(a, b): 
    return a-b

• add, sub 함수가 있는 mod1.py 파일을 현 디렉터리에 저장
• mod1.py가 모듈 (** 파이썬 확장자 .py로 만든 파이썬 파일은 모두 모듈)

2. 모듈 불러오기

import 모듈

>>> import mod1    # import mod1.py로 입력하지 않도록 주의
>>> print(mod1.add(3, 4))    # 모듈명.함수명
7

from 모듈 import 함수

>>> from mod1 import add    # from 모듈명 import 함수명
>>> add(3, 4)    # 쓸때마다 모듈명을 쓰지 않아도 됨
7

두개 이상의 함수 호출 + 모듈명 없이 사용

>>> from mod1 import add, sub    # from 모듈명 import 함수명1, 함수명2, ...
>>> add(3, 4)
7

>>> from mod1 import *    # *는 '모든 것'이라는 뜻 -> 모듈 내 모든 함수 불러오기
>>> add(3, 4)
7

3. if __name__ == "__main__"

🍕 import가 아닌, 파일을 직접 실행할 때만 수행하고 싶은 내용을 위해

# mod1.py 
def add(a, b): 
    return a+b

def sub(a, b): 
    return a-b

print(add(1, 4))
print(sub(4, 2))

# 파일을 직접 실행할 경우
C:\doit>python mod1.py
5
2

# 모듈로 import할 경우
>>> import mod1
5
2

• 모듈 내 함수를 사용하기 위해 모듈을 import했는데, 파일이 실행되어 결괏값을 출력해버림

🍕 if __name__ == "__main__"

직접 파일 실행 시 if절이 참이 되어, if 이하 수행
모듈을 불러서 사용 시 if절이 거짓이 되어, if 이하 수행하지 않음

# mod1.py 
def add(a, b): 
    return a+b

def sub(a, b): 
    return a-b

if __name__ == "__main__":
    print(add(1, 4))
    print(sub(4, 2))

# 모듈로 import할 경우
>>> import mod1    
>>>                      # if 이하 수행되지 않음

🍔 __name__ 변수

• 파이썬 내부적으로 사용하는 특별한 변수 이름
• mod1.py 파일을 직접 실행할 경우, __name__ 변수에는 __main__값이 저장됨
• mod1을 import 할 경우, mod1.py의 __name__ 변수에는 mod1이 저장됨.

4. 다른 파일에서 모듈 불러오기

🍕 다른 파일에서 모듈 불러오기 (feat. 같은 디렉터리)

# modtest.py
import mod2
result = mod2.add(3, 4)
print(result)

🍕 다른 디렉터리에 있는 모듈 불러오기

1. sys.path.append(모듈을 저장한 디렉터리)

C:\doit>python
>>> import sys

# 파이썬 라이브러리가 설치되어있는 디렉터리 리스트에 해당 디렉터리 추가
>>> sys.path.append("C:/doit/mymod")    

# 불러울 모듈이 있는 디렉터리 추가 완료
>>> sys.path
['', 'C:\\Windows\\SYSTEM32\\python37.zip', 'c:\\Python37\\DLLs', 
'c:\\Python37\\lib', 'c:\\Python37', 'c:\\Python37\\lib\\site-packages', 
'C:/doit/mymod']

>>> import mod2
>>> print(mod2.add(3,4))
7

2. PYTHONPATH 환경 변수 사용하기

C:\doit>set PYTHONPATH=C:\doit\mymod    # set PYTHONPATH로 환경변수에 디렉터리 설정   
C:\doit>python
>>> import mod2    # 별도의 디렉터리 이동 없이 모듈 불러오기 가능
>>> print(mod2.add(3,4))
7

아래 내용을 바탕으로 정리한 것이다.

점프투파이썬 (https://wikidocs.net/29)