[scrobbles] Fix dataclass parsing and add puzzles to urls

vrobbler/apps/books/models.py

@@ -1,6 +1,8 @@
 from collections import OrderedDict
+from dataclasses import dataclass
 import logging
 from datetime import timedelta, datetime
+from typing import Optional
 from uuid import uuid4
 
 import requests
@@ -35,9 +37,9 @@ from vrobbler.apps.books.locg import (
     lookup_comic_from_locg,
     lookup_comic_writer_by_locg_slug,
 )
-from vrobbler.apps.books.sources.google import lookup_book_from_google
-from vrobbler.apps.books.sources.semantic import lookup_paper_from_semantic
-from vrobbler.apps.scrobbles.dataclasses import BookLogData
+from books.sources.google import lookup_book_from_google
+from books.sources.semantic import lookup_paper_from_semantic
+from scrobbles.dataclasses import BaseLogData, LongPlayLogData
 
 COMICVINE_API_KEY = getattr(settings, "COMICVINE_API_KEY", "")
 
@@ -46,6 +48,23 @@ User = get_user_model()
 BNULL = {"blank": True, "null": True}
 
 
+@dataclass
+class BookPageLogData(BaseLogData):
+    page_number: Optional[int] = None
+    end_ts: Optional[int] = None
+    start_ts: Optional[int] = None
+    duration: Optional[int] = None
+
+
+@dataclass
+class BookLogData(BaseLogData, LongPlayLogData):
+    koreader_hash: Optional[str] = None
+    page_data: Optional[dict[int, BookPageLogData]] = None
+    pages_read: Optional[int] = None
+    page_start: Optional[int] = None
+    page_end: Optional[int] = None
+
+
 class Author(TimeStampedModel):
     name = models.CharField(max_length=255)
     uuid = models.UUIDField(default=uuid4, editable=False, **BNULL)
@@ -161,7 +180,9 @@ class Book(LongPlayScrobblableMixin):
         return reverse("books:book_detail", kwargs={"slug": self.uuid})
 
     @classmethod
-    def find_or_create(cls, title: str, enrich: bool = False, commit: bool = True):
+    def find_or_create(
+        cls, title: str, enrich: bool = False, commit: bool = True
+    ):
         """Given a title, get a Book instance.
 
         If the book is not already in our database, or overwrite is True,
@@ -173,10 +194,15 @@ class Book(LongPlayScrobblableMixin):
         # TODO use either a Google Books id identifier or author name like for tracks
         book, created = cls.objects.get_or_create(title=title)
         if not created:
-            logger.info("Found exact match for book by title", extra={"title": title})
+            logger.info(
+                "Found exact match for book by title", extra={"title": title}
+            )
 
         if not enrich:
-            logger.info("Found book by title, but not enriching", extra={"title": title})
+            logger.info(
+                "Found book by title, but not enriching",
+                extra={"title": title},
+            )
             return book
 
         book_dict = lookup_book_from_google(title)