[BW 18] __missing__을 사용해 키에 따라 다른 디폴트 값을 생성하는 방법

만약 defaultdict 의 기본 값을 어떤 인자가 필요한 함수로 지정하면 다음과 같이 오류가 발생한다.

from collections import defaultdict

path = "/"

def open_file(file_path):
    try:
        return open(file_path, 'a+b')
    except OSError:
        print(f"경로를 알 수 없습니다: {file_path}")
        raise

files = defaultdict(open_file)
handle = files[path]
handle.seek(0)
file_data = handle.read()

>>>
TypeError: open_file() missing 1 required positional argument: 'file_path'

함수는 하나의 인수가 필요한데, defaultdict를 생성할 때 인수를 같이 넣어주지 않았기 때문이다.

이를 위해 우리는 __missing__ 메서드를 구현할 수 있다.

path = "test.jpg"

# 딕셔너리에 key가 없을 경우 커스텀 로직을 처리할 수 있도록 하는 클래스
class Files(dict):
    def __missing__(self, key):
        # value에 어떤 함수의 로직을 처리할 것인지 지정한다.
        value = open_file(key)
        self[key] = value
        return value

def open_file(file_path):
    try:
        return open(file_path, 'a+b')
    except OSError:
        print(f"경로를 알 수 없습니다: {file_path}")
        raise

files = Files()
handle = files[path]
handle.seek(0)
file_data = handle.read()
print(file_data)

>>>
b''