20. 미니 프로젝트: 도서 관리 CLI

book_project/
├── book.py        ← Book 클래스 (책 한 권)
├── library.py     ← Library 클래스 (책들의 관리)
├── main.py        ← CLI 메인 (사용자 상호작용)
└── test_library.py ← 테스트

# book.py
"""도서 한 권을 나타내는 모듈."""

class Book:
    def __init__(self, title: str, author: str, read: bool = False) -> None:
        self.title = title
        self.author = author
        self.read = read

    def mark_read(self) -> None:
        """읽음으로 표시."""
        self.read = True

    def to_dict(self) -> dict:
        """JSON 저장용 딕셔너리로 변환."""
        return {"title": self.title, "author": self.author, "read": self.read}

    @classmethod
    def from_dict(cls, data: dict) -> "Book":
        """딕셔너리에서 Book 객체를 복원."""
        return cls(data["title"], data["author"], data["read"])

    def __str__(self) -> str:
        status = "✓ 읽음" if self.read else "· 안 읽음"
        return f"[{status}] {self.title} — {self.author}"

    def __repr__(self) -> str:
        return f"Book({self.title!r}, {self.author!r}, read={self.read})"

# library.py
"""도서 컬렉션을 관리하는 모듈."""
import json
from book import Book

class Library:
    def __init__(self) -> None:
        self.books: list[Book] = []

    def add(self, title: str, author: str) -> Book:
        book = Book(title, author)
        self.books.append(book)
        return book

    def find(self, keyword: str) -> list[Book]:
        """제목이나 저자에 keyword가 든 책을 찾는다."""
        kw = keyword.lower()
        return [b for b in self.books
                if kw in b.title.lower() or kw in b.author.lower()]

    def unread(self) -> list[Book]:
        """안 읽은 책 목록."""
        return [b for b in self.books if not b.read]

    def stats(self) -> dict:
        """전체·읽음·안읽음 권수."""
        total = len(self.books)
        read = sum(1 for b in self.books if b.read)
        return {"total": total, "read": read, "unread": total - read}

    def save(self, path: str) -> None:
        data = [b.to_dict() for b in self.books]
        with open(path, "w", encoding="utf-8") as f:
            json.dump(data, f, ensure_ascii=False, indent=2)

    def load(self, path: str) -> None:
        try:
            with open(path, encoding="utf-8") as f:
                data = json.load(f)
            self.books = [Book.from_dict(d) for d in data]
        except FileNotFoundError:
            self.books = []      # 파일이 없으면 빈 상태로 시작

from library import Library

lib = Library()
lib.add("파이썬 입문", "김파이")
lib.add("자료구조", "이구조")
algo = lib.add("알고리즘", "박알고")
algo.mark_read()              # 알고리즘은 읽음 처리

print("=== 전체 목록 ===")
for book in lib.books:
    print(book)               # __str__ 덕분에 보기 좋게

print("\n=== '파이' 검색 ===")
for book in lib.find("파이"):
    print(book)

print("\n=== 통계 ===")
print(lib.stats())

=== 전체 목록 ===
[· 안 읽음] 파이썬 입문 — 김파이
[· 안 읽음] 자료구조 — 이구조
[✓ 읽음] 알고리즘 — 박알고

=== '파이' 검색 ===
[· 안 읽음] 파이썬 입문 — 김파이

=== 통계 ===
{'total': 3, 'read': 1, 'unread': 2}

[
  {
    "title": "파이썬 입문",
    "author": "김파이",
    "read": false
  },
  ...
]

# main.py
"""도서 관리 CLI 메인 프로그램."""
from library import Library

DATA_FILE = "books.json"

def print_menu() -> None:
    print("\n=== 도서 관리 ===")
    print("1) 추가  2) 목록  3) 검색  4) 통계  5) 종료")

def main() -> None:
    lib = Library()
    lib.load(DATA_FILE)             # 시작 시 저장된 데이터 불러오기
    while True:
        print_menu()
        choice = input("선택: ").strip()
        if choice == "1":
            title = input("제목: ")
            author = input("저자: ")
            lib.add(title, author)
            lib.save(DATA_FILE)     # 변경 즉시 저장
            print("추가됨!")
        elif choice == "2":
            for book in lib.books:
                print(book)
        elif choice == "3":
            kw = input("검색어: ")
            for book in lib.find(kw):
                print(book)
        elif choice == "4":
            print(lib.stats())
        elif choice == "5":
            print("안녕히 가세요!")
            break
        else:
            print("1~5만 입력하세요")

if __name__ == "__main__":          # 15장: 직접 실행할 때만 main() 호출
    main()

# test_library.py
from library import Library
from book import Book

def test_add():
    lib = Library()
    lib.add("책1", "저자1")
    assert len(lib.books) == 1

def test_find():
    lib = Library()
    lib.add("파이썬", "김씨")
    lib.add("자바", "이씨")
    assert len(lib.find("파이")) == 1     # 제목 매치
    assert len(lib.find("씨")) == 2       # 두 저자 모두 '씨'

def test_stats():
    lib = Library()
    b = lib.add("책", "저자")
    b.mark_read()
    lib.add("책2", "저자2")
    assert lib.stats() == {"total": 2, "read": 1, "unread": 1}

def test_book_str():
    b = Book("제목", "작가", read=True)
    assert "✓ 읽음" in str(b)

$ pytest test_library.py -v
test_library.py::test_add PASSED
test_library.py::test_find PASSED
test_library.py::test_stats PASSED
test_library.py::test_book_str PASSED
===== 4 passed =====

# library.py의 Library 클래스에 추가
def by_author(self, author: str) -> list[Book]:
    return [b for b in self.books if b.author == author]

# test_library.py에 테스트 추가
def test_by_author():
    lib = Library()
    lib.add("책A", "김작가")
    lib.add("책B", "김작가")
    lib.add("책C", "이작가")
    assert len(lib.by_author("김작가")) == 2