secstate/vrobbler
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[books] Fix koreader import with nul strings

Browse Source
This commit is contained in:
Colin Powell
2024-11-05 21:13:37 -05:00
parent 0679af3029
commit b65ebbe397
1 changed files with 11 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
vrobbler/apps/books/koreader.py
View File

@ -84,16 +84,22 @@ def lookup_or_create_authors_from_author_str(ko_author_str: str) -> list:
def create_book_from_row(row: list): def create_book_from_row(row: list):
# No KoReader book yet, create it # No KoReader book yet, create it
author_str = get_author_str_from_row(row) author_str = get_author_str_from_row(row).replace("\x00", "")
total_pages = row[KoReaderBookColumn.PAGES.value] total_pages = row[KoReaderBookColumn.PAGES.value]
run_time = total_pages * Book.AVG_PAGE_READING_SECONDS run_time = total_pages * Book.AVG_PAGE_READING_SECONDS
book_title = row[KoReaderBookColumn.TITLE.value] book_title = row[KoReaderBookColumn.TITLE.value].replace("\x00", "")
if " - " in book_title: if " - " in book_title:
split_title = book_title.split(" - ") split_title = book_title.split(" - ")
book_title = split_title[0] book_title = split_title[0]
if (not author_str or author_str == "N/A") and len(split_title) > 1: if (not author_str or author_str == "N/A") and len(split_title) > 1:
author_str = split_title[1].split("_")[0] author_str = split_title[1].split("_")[0]
clean_row = []
for value in row:
if isinstance(value, str):
value = value.replace("\x00", "")
clean_row.append(value)
book = Book.objects.create( book = Book.objects.create(
title=book_title.replace("_", ":"), title=book_title.replace("_", ":"),
pages=total_pages, pages=total_pages,
@ -102,7 +108,7 @@ def create_book_from_row(row: list):
"title": book_title, "title": book_title,
"author_str": author_str, "author_str": author_str,
"book_id": row[KoReaderBookColumn.ID.value], "book_id": row[KoReaderBookColumn.ID.value],
"raw_row_data": row, "raw_row_data": clean_row,
} }
}, },
run_time_seconds=run_time, run_time_seconds=run_time,
@ -142,11 +148,8 @@ def build_book_map(rows) -> dict:
).first() ).first()
if not book: if not book:
book = Book.objects.filter( title = book_row[KoReaderBookColumn.TITLE.value].split(" - ")[0].lower().replace("\x00", "")
title=book_row[KoReaderBookColumn.TITLE.value] book = Book.objects.filter(title=title).first()
.split(" - ")[0]
.lower()
).first()
if not book: if not book:
book = create_book_from_row(book_row) book = create_book_from_row(book_row)