From 4e8c0e6ffe08b7afeb3d212c2a60c74d76b77d61 Mon Sep 17 00:00:00 2001 From: Kristofers Solo Date: Wed, 8 Nov 2023 13:51:28 +0200 Subject: [PATCH] feat(Book): add `get` method --- src/bookstore/book.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/src/bookstore/book.py b/src/bookstore/book.py index 606ab5e..a02e02d 100644 --- a/src/bookstore/book.py +++ b/src/bookstore/book.py @@ -32,5 +32,20 @@ class Book: def values(self) -> tuple[ISBN, str, str, float, int]: return self.isbn, self.title, self.author, self.price, self.stock + def get(self, field: str, default: str = "") -> ISBN | str | float | int: + match field: + case "ISBN": + return self.isbn + case "Title": + return self.title + case "Author": + return self.author + case "Price": + return self.price + case "Stock": + return self.stock + case _: + return default + def __iter__(self): yield from self.values()