From bba73ea1a7c5e9f2df17a99a4c9866cebe64f2e6 Mon Sep 17 00:00:00 2001 From: Mike Date: Fri, 25 Oct 2019 21:45:20 -0400 Subject: [PATCH] Added close method, and finished compare_shelf_current --- lib/library.py | 24 +++++++++++++++--------- 1 file changed, 15 insertions(+), 9 deletions(-) diff --git a/lib/library.py b/lib/library.py index 26ddec0..79c78f1 100755 --- a/lib/library.py +++ b/lib/library.py @@ -123,14 +123,17 @@ class Catalogue: except KeyError: return False def compare_shelf_current(self): - stored_books = Storage() - stored_books = stored_books.book_paths_list() - try: - self.books - except Exception: - self.filter_books() - unique = set(self.books) - set(stored_books) - return unique + db = Storage() + stored = db.book_paths_list() + closed = db.close() + try: self.books + except Exception: self.filter_books() + on_disk, in_storage = [], [] + for _x in self.books: on_disk.append(_x) + for _y in stored: in_storage.append(_y[0]) + a, b, = set(on_disk), set(in_storage) + c = set.difference(a, b) + return c def import_books(self, list=None): book_list = self.compare_shelf_current() @@ -139,4 +142,7 @@ class Catalogue: book = self.process_book(book) extracted = self.extract_metadata(book) db.insert_book(extracted) - db.commit() + inserted = db.commit() + if inserted != True: + print(inserted) + db.close() \ No newline at end of file