[Python] 파일 및 디렉터리 제어

 

※ SK 쉴더스 루키즈 24기 (클라우드기반 스마트 융합보안 과정) 교육을 들으면서 작성한 내용입니다.

 

1️⃣ 기본 개념: 파일과 파일 경로

🔹1. 파일과 파일 경로의 이해

• 파일: 데이터를 저장하는 단위
  • 텍스트 파일(``.txt``), 이미지 파일(``.png``), 실행 파일(``.exe``) 등 다양한 형식이 있음
• 파일 경로: 컴퓨터에서 파일의 위치를 나타내는 문자열. 경로는 두 가지로 구분 됨
  • 절대 경로: 파일의 전체 위칠르 나타냄 (예: ``C:\Users\User\example.txt``)
  • 상대 경로: 현재 작업 디렉터리를 기준으로 파일의 위치를 나타냄 (예: ``../example.txt``)

---

🔹2. 주요 경로 표기법

• Windows: 경로 구분자로 백슬래시(``\``)를 사용
• maxOS/Linux: 경로 구분자로 슬래시(``/``)를 사

file_path = "C:\\Users\\User\\example.txt"  # Windows 경로
file_path = "/home/user/example.txt"  # macOS/Linux 경로

루트 폴더

• 모든 폴더가 포함된 최상위 폴더를 뜻함
• Windows의 루트 폴더: ``C:\``
• macOS 또는 Linux의 루트 폴더: ``/``

---

🔹3. os 라이브러리

• 파이썬의 내장 모듈인 os는 운영 체제에서 파일과 디렉터리를 제어할 수 있는 다양한 기능을 제공

import os

---

2️⃣ 디렉터리 작업

🔹1. 현재 작업 디렉터리 확인: os.getcwd()

• 현재 작업 중인 디렉터리를 확인하려면 ``os.getcwd()``를 사용

import os

current_dir = os.getcwd()
print(f"현재 작업 디렉터리: {current_dir}")

---

🔹2. 디렉터리 내용 확인: os.listdir()

• 지정된 디렉터리의 파일 및 하위 디렉터리를 확인하려면 ``os.listdir()을 사용``

contents = os.listdir('.')  # 현재 디렉터리 내용 확인
print(contents)

---

🔹3. 디렉터리 탐색: os.walk()

• 디렉터리 구조를 재귀적으로 탐색하려면 ``os.walk()``를 사용

for dirpath, dirnames, filenames in os.walk('.'):
    print(f"디렉터리: {dirpath}")
    print(f"하위 디렉터리: {dirnames}")
    print("파일: {filenames}")

---

3️⃣ 파일 작업

🔹1. 파일 존재 여부 확인: os.path.isfile()

• 파일인지 확인하려면 ``os.path.isfile()``를 사용

import os

file_path = 'example.txt'
if os.path.isfile(file_path):
    print(f"{file_path}는 파일입니다.")
else:
    print(f"{file_path}는 파일이 아닙니다.")

---

🔹2. 디렉터리 존재 여부 확인: os.path.isdir()

• 디렉터리인지 확인하려면 ``os.path.isdir()``를 사용

if os.path.isdir('test_dir'):
    print("디렉터리가 존재합니다.")
else:
    print("디렉터리가 존재하지 않습니다.")

---

🔹3. 파일 경로 결합: os.path.join()

• 파일 경로를 안전하게 결합하려면 ``os.path.join()``를 사용

file_path = os.path.join('test_dir', 'example.txt')
print(f"파일 경로: {file_path}")

---

4️⃣ 파일 읽기와쓰기 

🔹1. 파일 열기와 읽기: with open()

• 파일을 읽거나 쓸 때는 ``with open()``을 사용하여 파일을 안전하게 열고, 닫을 수 있음
• ``open()``함수는 파일을 여는 모드 옵션을 통해 다양한 작업을 지원
  • ``'r'`` (읽기 모드): 파일을 읽기 전용으로 여는데, 파일이 없으면 오류 발생
  • ``'w'`` (쓰기 모드): 파일에 내용 작성, 파일이 없으면 생성하고, 기존 내용은 덮어씀
  • ``'a'`` (추가 모드): 파일 끝에 내용을 추가, 파일이 없으면 생성

# 파일 읽기
with open('example.txt', 'r') as f:
    content = f.read()
    print(content)

# 파일 쓰기
with open('example.txt', 'w') as f:
    f.write("Hello, Python!")

 

• 파일 읽기 관련 메서드:
  • ``f.read()``: 파일 전체 읽기
  • ``f.readline()``: 파일에서 한 줄 읽기
  • ``f.readlines()``: 파일의 모든 줄을 리스트로 반환

---

5️⃣ 파일 이름 및 확장자 처리

🔹1. 특정 확장자로 필터링: ``str.endswith()``

• 파일 이름이 특정 확장자로 끝나는지 확인하려면 ``str.endswith()``를 사용

files = ['example.txt', 'data.csv', 'image.png']

for file in files:
    if file.endswith('.txt'):
        print(f"텍스트 파일: {file}")

---

6️⃣ 실용 예제: 특정 확장자 파일 검사 및 조건 탐지

• 다음은 특정 폴더에 있는 특정 확장자 파일을 검사하고, 각 파일에서 특정 조검을 탐지하는 예제임
• 예를 들어 .txt 파일에서 "#error"라는 단어로 시작되는 줄을 찾는 코드는 다음과 같음:

import os

dir_path = ""	# 특정 디렉터리 경로

all_files = os.listdir(dir_path)

txt_files = []

for file in all_files:
	if file.endswitch(.'txt'):
    	txt_files.append(file)
        
for file_name in txt_files:
    file_path = os.path.join(dir_path, file_name)
 
    with open(file_path, "r", encoding='utf-8') as f:
        lines = f.readlines()

        for i, line in enumerate(lines):
            if line.startswith('#error'):
                print(f"인덱스: {i+1}, 탐지 문자: {line}")

---