[Python Cheatsheet] 54. zipfile & tarfile - 압축 파일 읽기/쓰기/추출

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

import zipfile

# ZIP 파일 열기 및 목록 확인
with zipfile.ZipFile('archive.zip', 'r') as zf:
    # 파일 목록
    print(zf.namelist())
    # ['file1.txt', 'folder/file2.txt']
    
    # 상세 정보
    for info in zf.infolist():
        print(f"{info.filename}: {info.file_size} bytes")

# 특정 파일 읽기
with zipfile.ZipFile('archive.zip', 'r') as zf:
    content = zf.read('file1.txt')  # bytes
    text = content.decode('utf-8')
    print(text)

# 파일처럼 열기
with zipfile.ZipFile('archive.zip', 'r') as zf:
    with zf.open('file1.txt') as f:
        for line in f:
            print(line.decode('utf-8').strip())

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

import zipfile
from pathlib import Path

# 전체 추출
with zipfile.ZipFile('archive.zip', 'r') as zf:
    zf.extractall('output_dir')

# 특정 파일만 추출
with zipfile.ZipFile('archive.zip', 'r') as zf:
    zf.extract('file1.txt', 'output_dir')

# 안전한 추출 (경로 순회 공격 방지)
def safe_extract(zip_path: str, extract_dir: str) -> None:
    extract_path = Path(extract_dir).resolve()
    
    with zipfile.ZipFile(zip_path, 'r') as zf:
        for member in zf.namelist():
            member_path = (extract_path / member).resolve()
            # 추출 경로가 대상 디렉토리 안에 있는지 확인
            if not str(member_path).startswith(str(extract_path)):
                raise ValueError(f"Path traversal detected: {member}")
        
        zf.extractall(extract_dir)

safe_extract('archive.zip', 'output_dir')

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

import zipfile
from pathlib import Path

# 새 ZIP 파일 생성
with zipfile.ZipFile('new_archive.zip', 'w') as zf:
    # 파일 추가
    zf.write('file1.txt')
    
    # 아카이브 내 경로 지정
    zf.write('file2.txt', arcname='folder/file2.txt')
    
    # 문자열에서 직접 추가
    zf.writestr('hello.txt', 'Hello, World!')

# 압축 옵션
with zipfile.ZipFile('compressed.zip', 'w', zipfile.ZIP_DEFLATED) as zf:
    zf.write('large_file.txt')

# 압축률 조정 (compresslevel: 0-9)
with zipfile.ZipFile('compressed.zip', 'w', 
                     zipfile.ZIP_DEFLATED, 
                     compresslevel=9) as zf:
    zf.write('large_file.txt')

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12

# 디렉토리 전체 압축
import zipfile
from pathlib import Path

def zip_directory(dir_path: str, zip_path: str) -> None:
    with zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED) as zf:
        for file_path in Path(dir_path).rglob('*'):
            if file_path.is_file():
                arcname = file_path.relative_to(dir_path)
                zf.write(file_path, arcname)

zip_directory('my_folder', 'my_folder.zip')

1
2
3
4
5

import zipfile

# 기존 ZIP에 파일 추가
with zipfile.ZipFile('archive.zip', 'a') as zf:
    zf.write('new_file.txt')

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

import tarfile

# TAR 파일 열기 (자동 압축 감지)
with tarfile.open('archive.tar.gz', 'r:*') as tf:
    # 파일 목록
    print(tf.getnames())
    
    # 상세 정보
    for member in tf.getmembers():
        print(f"{member.name}: {member.size} bytes")

# 전체 추출
with tarfile.open('archive.tar.gz', 'r:gz') as tf:
    tf.extractall('output_dir')

# 특정 파일만 추출
with tarfile.open('archive.tar.gz', 'r:gz') as tf:
    tf.extract('file1.txt', 'output_dir')

# 안전한 추출 (Py3.12+)
with tarfile.open('archive.tar.gz', 'r:gz') as tf:
    tf.extractall('output_dir', filter='data')  # 안전한 필터

# Py3.11 이하 안전 추출
def safe_tar_extract(tar_path: str, extract_dir: str) -> None:
    from pathlib import Path
    extract_path = Path(extract_dir).resolve()
    
    with tarfile.open(tar_path, 'r:*') as tf:
        for member in tf.getmembers():
            member_path = (extract_path / member.name).resolve()
            if not str(member_path).startswith(str(extract_path)):
                raise ValueError(f"Path traversal: {member.name}")
            if member.issym() or member.islnk():
                raise ValueError(f"Symlink not allowed: {member.name}")
        
        tf.extractall(extract_dir)

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18

import tarfile

# 비압축 TAR
with tarfile.open('archive.tar', 'w') as tf:
    tf.add('file1.txt')
    tf.add('folder', arcname='renamed_folder')

# gzip 압축
with tarfile.open('archive.tar.gz', 'w:gz') as tf:
    tf.add('file1.txt')

# bzip2 압축
with tarfile.open('archive.tar.bz2', 'w:bz2') as tf:
    tf.add('file1.txt')

# xz 압축 (높은 압축률)
with tarfile.open('archive.tar.xz', 'w:xz') as tf:
    tf.add('file1.txt')

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11

# 특정 파일 제외
import tarfile

def exclude_filter(tarinfo):
    # .pyc 파일과 __pycache__ 제외
    if tarinfo.name.endswith('.pyc') or '__pycache__' in tarinfo.name:
        return None
    return tarinfo

with tarfile.open('archive.tar.gz', 'w:gz') as tf:
    tf.add('my_project', filter=exclude_filter)

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15

import zipfile
import io

# 메모리에서 ZIP 생성
buffer = io.BytesIO()
with zipfile.ZipFile(buffer, 'w', zipfile.ZIP_DEFLATED) as zf:
    zf.writestr('hello.txt', 'Hello from memory!')

# buffer.getvalue()로 bytes 얻기
zip_bytes = buffer.getvalue()

# 메모리에서 ZIP 읽기
buffer = io.BytesIO(zip_bytes)
with zipfile.ZipFile(buffer, 'r') as zf:
    print(zf.read('hello.txt').decode())