[Python] OS, sys, glob 모듈 개념, 활용 예제
1. OS 모듈
환경 변수나 디렉터리, 파일 등의 OS 자원을 제어할 수 있게 해주는 모듈
1) 내 시스템의 환경 변수를 알고 싶을 때
print(os.environ)
2) 현재 디렉토리 출력
print(os.getcwd())
3) 현재 디렉토리 이동
#현재 디렉토리 이동
os.chdir("C:/spark/spark-3.1.2-bin-hadoop2.7")
#현재 디렉토리 확인
print(os.getcwd())
4) 파일 객체 읽기
#실행한 시스템 명령어의 결괏값 돌려받기 - os.popen
#os.popen은 시스템 명령어를 실행한 결괏값을 읽기 모드 형태의 파일 객체로 돌려준다.
f = os.popen("test.txt")
#읽어들인 파일 객체의 내용 보기
print(f.read())
5) 모든 파일, 폴더명 리스트로 반환
#입력 경로 내의 모든 파일과 폴더명 리스트 반환
#직접 경로를 주어도 된다.
dir_list = os.listdir(".")
print(dir_list)
6) 모든 경로의 하위 디렉토리,파일,path출력
#인자로 받은 모든 경로에 대해 모든 하위 디렉토리 출력
for path, direct, files in os.walk("."):#현재 디렉토리의 인자로 받은 모든 하위 디렉토리 출력
print(path)
"." : 현재 디렉토리
path: PATH
direct: 디렉토리
files: 파일
2. sys 모듈
sys 모듈은 파이썬 인터프리터가 제공하는 변수와 함수를 직접 제어할 수 있게 해주는 모듈
1) sys.path, sys.path.append
findspark를 설치하지않은 상태에서 아래의 코드를 실행 시키면 모듈이 없다는 에러가 나옵니다.
import findspark
findspark.init()
from pyspark.sql import SparkSession
spark = SparkSession.builder.master("local[*]").appName("pyspark").getOrCreate()
spark.sql("select '#' as col1").show()
import 문을 통해 다른 파이썬 파일을 불러올 때, 파이썬은 내부적으로 파일을 찾기 위해 sys.path와 PYTHONPATH에 있는 경로를 탐색
sys.path는 디렉터리의 경로들이 기록된 문자열 리스트
이 리스트에 경로를 추가하면 해당 경로에 있는 파이썬 파일을 import 문으로 불러올 수 있습니다.
현재의 파이썬 가상환경에 findspark를 설치하지 않고, 이미 설치되어있는 가상환경을 sys.path.append("경로")를 통해 sys.path에 필요한 라이브러리 및 모듈이 있는 경로를 지정해줍니다.
import sys
#sys 모듈을 통해 다른 모듈이 있는 경로를 추가해준다.
sys.path.append("C:/Users/user/Anaconda3/Lib/site-packages")
sys.path에는 파이썬에 의해 기본적으로 몇 가지 경로가 미리 추가되어 있습니다.
.py 파일이 속한 디렉터리의 절대 경로
sys.path에는 가장 먼저 .py 파일이 속한 디렉터리의 절대 경로가 추가됩니다.
특정 파이썬 파일을 실행하는 것 대신, 파이썬 인터프리터에서 직접 print(sys.path)를 실행하면 인터프리터를 실행할 당시의 경로(현재 작업 디렉터리)가 sys.path에 추가됩니다.
PYTHONPATH 환경 변수에 경로를 추가하면, 파이썬은 이 경로들을 sys.path에 추가해줍니다.
이를 통해 파이썬 코드 내부에서 뿐만 아니라 파이썬 코드 밖에서도 sys.path를 조작 가능합니다.
2) sys.argv()
명령 행에서 인수 전달하기
# argv_test.py
import sys
print(sys.argv)
명령 프롬프트 창에서 python_test디렉터리로 들어간 뒤 다음과 같이 실행
C:/python_test>python argv_test.py you need python
['argv_test.py', 'you', 'need', 'python']매개변수 입력
매개변수 출력
python 명령어 뒤의 모든 것들이 공백을 기준으로 나뉘어서 sys.argv 리스트의 요소가 됩니다.
3. glob 모듈
1) 현재 경로의 모든 디렉토리,파일을 출력
from glob import glob
import os
#현재 디렉토리 변경
os.chdir("C:\python_skillup")
#현재 경로의 모든 디렉토리,파일을 출력(숨긴 파일은 출력x)
print(glob("*"))
print("\n")
#현재 경로의 모든 디렉토리,파일을 출력(숨긴 파일도 출력o)
print(os.listdir("."))
print("\n")
2) 해당 경로의 모든 디렉토리, 파일 출력
#해당 경로의 모든 디렉토리,파일을 출력
print(glob("C:/python_skillup/os_sys_glob/*"))
3) 해당 경로의 모든 파이썬 파일 출력
#해당 경로의 모든 .py파일 출력
print(glob("C:/baekjoon_algorithm/*.py"))
.py뿐만아니라 .txt, .exe등 원하는 확장자의 파일을 출력할 수도 있습니다.
4) os 모듈+ glob 모듈 활용 예제
#현재 디렉토리의 모든 디렉토리,파일을 불러와 디렉토리일 경우 <DIR>를 붙여 출력하고 디렉토리가 아닐경우 그냥 출력
for x in glob("*"):
if os.path.isdir(x):
print(x,'<DIR>')
else:
print(x)
추가적으로 더 알게되는 내용이 있거나, 활용한 파이썬 스크립트를 만들게 될 경우 링크를 달거나 추가하겠습니다.