🥒 What is 'pickle' ?

pickle 모듈은 파이썬 객체 구조의 직렬화와 역 직렬화를 위한 바이너리 프로토콜을 구현합니다. “피클링(pickling)”은 파이썬 객체 계층 구조가 바이트 스트림으로 변환되는 절차이며, “역 피클링(unpickling)”은 반대 연산으로, (바이너리 파일 이나 바이트열류 객체로 부터의) 바이트 스트림을 객체 계층 구조로 복원합니다.

import pickle

1. pickle.dump(obj, file, protocol=None)

: 피클 모듈로 저장하기 → 객체 obj를 직렬화하여 파일 file에 저장합니다.

✅ 파일 오픈할 때 with 함수를 사용하면 명시적으로 파일을 닫지 않아도 됩니다. with 블록을 떠나는 시점에 닫는 동작이 수행되기 때문에 예외 또는 중단시에도 파일이 닫히게 됩니다.

✅ wb 는 write binary를 의미합니다. 저장할 데이터를 생성할 파일에 적는 것을 의미합니다.

저는 saved_model.pkl 이라는 파일 이름으로 저장하였습니다.

def save():
    # 파일로 저장
    with open('saved_model.pkl', 'wb') as f:
        pickle.dump(model, f)
    return {"Saved": "OK"}

2. pickle.load(file)

: 피클 모듈로 불러오기 → 파일에서 직렬화된 객체를 읽어 역직렬화(복원)합니다.
✅ wb 는 read binary를 의미합니다.
✅ 파일에서 직렬화된 파이썬 객체를 읽어옵니다.

def load():
    global model  # ✅ FastAPI 앱의 글로벌 model 변수 사용
    with open('saved_model.pkl', 'rb') as f:
        model = pickle.load(f)
        return {"Loaded": "OK"}

3. pickle.dumps(obj, protocol=None)

: 객체 obj를 직렬화하여 바이너리 문자열(bytes) 로 반환합니다.메모리 내 데이터를 변환하는 방식
파일에 저장하지 않고 메모리 상에서 직렬화된 데이터를 다룰 때 유용합니다.

binary_data = pickle.dumps(data)
print(binary_data)  # b'\x80\x04\x95...'

4. pickle.loads(bytes_obj)

: pickle.dumps() 로 직렬화된 바이너리 데이터를 복원합니다.메모리 내 데이터를 변환하는 방식
dumps() 로 변환된 데이터를 다시 원래의 객체로 변환합니다.

original_data = pickle.loads(binary_data)
print(original_data)  # {'name': 'Alice', 'age': 25, 'city': 'Seoul'}

⚠️ pickle 사용 시 주의할 점

1. 보안 위험
   pickle.load()를 사용할 때 신뢰할 수 없는 소스에서 받은 파일을 로드하면 악의적인 코드가 실행될 위험이 있습니다.
   → 대안: 안전한 데이터 저장이 필요할 경우 JSON(json 모듈) 사용 고려

2. 버전 문제
   파이썬 버전이 다르면 pickle 직렬화된 데이터를 복원할 때 문제가 발생할 수 있음.

3. 호환되지 않는 객체
   모든 객체가 pickle로 직렬화 가능한 것은 아닙니다.
   파일 핸들, 소켓, 스레드, 람다 함수 등은 pickle로 저장할 수 없습니다.

보안이 중요한 경우는 json을 사용하는 것이 더 안전하다.

직렬화(serialization)란 ?
객체를 일련의 바이트(byte)스트림으로 변환하는 과정을 의미합니다. 이 과정을 통해 객체를 파일에 저장하거나 네트워크를 통해 전송할 수 있습니다.직렬화된 바이트 스트림을 다시 객체로 변환하는 것을 역직렬화(deserialization)라고 합니다.

참고
Python pickle