kristoferssolo/LU-bookstore
1
0
Fork 0
mirror of https://github.com/kristoferssolo/LU-bookstore.git synced 2025-10-21 18:00:34 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat(sql): add edit method

Browse Source
This commit is contained in:
Kristofers Solo 2023-11-08 11:53:13 +02:00
parent aa53cd3223
commit 2dd9fac786
1 changed files with 16 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
src/bookstore/inventory.py
View File

@ -34,17 +34,28 @@ class Inventory:
for book in books: for book in books:
try: try:
self.cursor.execute("INSERT INTO Book VALUES (?, ?, ?, ?, ?)", (book.isbn, book.title, book.author, book.price, book.stock)) self.cursor.execute("INSERT INTO Book VALUES (?, ?, ?, ?, ?)", (book.isbn, book.title, book.author, book.price, book.stock))
self.conn.commit() self.save()
print(f"Book with ISBN: {book.isbn} was saved") print(f"Book with ISBN: {book.isbn} was successfully saved")
except sqlite3.InternalError: except sqlite3.InternalError as e:
print(f"A book with ISBN {book.isbn} already exists in the database.") print(f"A book with ISBN {book.isbn} already exists in the database.\n{e}")
def edit(self, book: Book) -> None:
"""Edit `Book`."""
try:
self.cursor.execute(
"UPDATE Book SET title = ?, author = ?, price = ?, stock = ? WHERE isbn = ?", (book.title, book.author, book.price, book.stock, book.isbn)
)
self.save()
print(f"Book with ISBN: {book.isbn} was successfully updated!")
except sqlite3.InternalError as e:
print(f"Something went wrong.\n{e}")
def delete(self, isbn: ISBN) -> Book | None: def delete(self, isbn: ISBN) -> Book | None:
"""Deletes `Book` from `Inventory` by `ISBN` and returns deleted `Book`""" """Deletes `Book` from `Inventory` by `ISBN` and returns deleted `Book`"""
deleted_book = self.find_by_isbn(isbn) deleted_book = self.find_by_isbn(isbn)
self.cursor.execute("DELETE FROM Book WHERE isbn = ?", (isbn,)) self.cursor.execute("DELETE FROM Book WHERE isbn = ?", (isbn,))
self.conn.commit() self.save()
return deleted_book return deleted_book