From f504d9f2a16dac0483cf35278e665f36dface75b Mon Sep 17 00:00:00 2001 From: Colin Powell Date: Sun, 12 Mar 2023 17:13:56 -0400 Subject: [PATCH] Fix none from KoReader imports --- vrobbler/apps/books/koreader.py | 16 +++++++++++----- 1 file changed, 11 insertions(+), 5 deletions(-) diff --git a/vrobbler/apps/books/koreader.py b/vrobbler/apps/books/koreader.py index 220fe48..6d68e96 100644 --- a/vrobbler/apps/books/koreader.py +++ b/vrobbler/apps/books/koreader.py @@ -62,7 +62,7 @@ def process_pages_for_book(book_id, sqlite_file_path): ] page.save() new_pages.append(page) - logger.info("Added {len(new_pages)} for book {book}") + logger.info(f"Added {len(new_pages)} for book {book}") return new_pages @@ -112,10 +112,16 @@ def process_koreader_sqlite_file(sqlite_file_path, user_id): if author_list: book.authors.add(*[a.id for a in author_list]) - playback_position_seconds = int( - book_row[KoReaderBookColumn.TOTAL_READ_TIME.value] - ) - pages_read = int(book_row[KoReaderBookColumn.TOTAL_READ_PAGES.value]) + playback_position_seconds = 0 + if book_row[KoReaderBookColumn.TOTAL_READ_TIME.value]: + playback_position_seconds = book_row[ + KoReaderBookColumn.TOTAL_READ_TIME.value + ] + pages_read = 0 + if book_row[KoReaderBookColumn.TOTAL_READ_PAGES.value]: + pages_read = int( + book_row[KoReaderBookColumn.TOTAL_READ_PAGES.value] + ) timestamp = datetime.utcfromtimestamp( book_row[KoReaderBookColumn.LAST_OPEN.value] ).replace(tzinfo=pytz.utc)