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):
# 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]
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:
split_title = book_title.split(" - ")
book_title = split_title[0]
if (not author_str or author_str == "N/A") and len(split_title) > 1:
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(
title=book_title.replace("_", ":"),
pages=total_pages,
@ -102,7 +108,7 @@ def create_book_from_row(row: list):
"title": book_title,
"author_str": author_str,
"book_id": row[KoReaderBookColumn.ID.value],
"raw_row_data": row,
"raw_row_data": clean_row,
}
},
run_time_seconds=run_time,
@ -142,11 +148,8 @@ def build_book_map(rows) -> dict:
).first()
if not book:
book = Book.objects.filter(
title=book_row[KoReaderBookColumn.TITLE.value]
.split(" - ")[0]
.lower()
).first()
title = book_row[KoReaderBookColumn.TITLE.value].split(" - ")[0].lower().replace("\x00", "")
book = Book.objects.filter(title=title).first()
if not book:
book = create_book_from_row(book_row)