Fix checking for deletion when saving book

Discard non book files: cover.png and cover-min.png

1 files changed, 4 insertions, 3 deletions

diff --git a/src/book_store.py b/src/book_store.py
index af1402e..ece5415 100644
--- a/src/book_store.py
+++ b/src/book_store.py
@@ -7,6 +7,7 @@ import glob
 import logging
 
 import src.db as db
+import src.book_files as book_files
 
 logger = logging.getLogger(__name__)
 
@@ -20,8 +21,8 @@ def store(library, conn, data, cover, books, book = None):
         save_cover(cover, directory)
 
         if book:
-            if not already_exist(directory, books) or has_delete(directory, books):
                 update_books(directory, books)
+            if not already_exist(directory, books) or has_delete(library, book_id, books):
         else:
             create_books(directory, books)
 
@@ -48,9 +49,9 @@ def already_exist(directory, books):
             return False
     return True
 
-def has_delete(directory, books):
+def has_delete(library, book_id, books):
     new_paths = books.keys()
-    for path in glob.glob(f'{directory}/*'):
+    for path in book_files.get(library, book_id):
         if not path in new_paths:
             return True
     return False