Book on flex

books/models.py

@@ -32,6 +32,54 @@ def cleanhtml(raw_html):
     return cleantext
 
 
+def prefetch_book_resources(queryset):
+    """Prefetch related faculty and student resources for a queryset of books."""
+    return queryset.prefetch_related(
+        models.Prefetch('bookfacultyresources_set', queryset=BookFacultyResources.objects.all(), to_attr='prefetched_faculty_resources'),
+        models.Prefetch('bookstudentresources_set', queryset=BookStudentResources.objects.all(), to_attr='prefetched_student_resources')
+    )
+
+def get_book_data(book):
+    """Return the book data dict for a single Book instance, matching BookIndex.books property."""
+    has_faculty_resources = hasattr(book, 'prefetched_faculty_resources') and bool(book.prefetched_faculty_resources)
+    has_student_resources = hasattr(book, 'prefetched_student_resources') and bool(book.prefetched_student_resources)
+    try:
+        return {
+            'id': book.id,
+            'cnx_id': book.cnx_id,
+            'slug': f'books/{book.slug}',
+            'book_state': book.book_state,
+            'title': book.title,
+            'subjects': book.subjects(),
+            'is_ap': book.is_ap,
+            'cover_url': book.cover_url,
+            'cover_color': book.cover_color,
+            'high_resolution_pdf_url': book.high_resolution_pdf_url,
+            'low_resolution_pdf_url': book.low_resolution_pdf_url,
+            'ibook_link': book.ibook_link,
+            'ibook_link_volume_2': book.ibook_link_volume_2,
+            'webview_link': book.webview_link,
+            'webview_rex_link': book.webview_rex_link,
+            'bookshare_link': book.bookshare_link,
+            'kindle_link': book.kindle_link,
+            'amazon_coming_soon': book.amazon_coming_soon,
+            'amazon_link': book.amazon_link,
+            'bookstore_coming_soon': book.bookstore_coming_soon,
+            'comp_copy_available': book.comp_copy_available,
+            'salesforce_abbreviation': book.salesforce_abbreviation,
+            'salesforce_name': book.salesforce_name,
+            'urls': book.book_urls(),
+            'last_updated_pdf': book.last_updated_pdf,
+            'has_faculty_resources': has_faculty_resources,
+            'has_student_resources': has_student_resources,
+            'assignable_book': book.assignable_book,
+            'promote_tags': [snippet.value.name for snippet in book.promote_snippet],
+        }
+    except Exception as e:
+        capture_exception(e)
+        return None
+
+
 class VideoFacultyResource(models.Model):
     resource_heading = models.CharField(max_length=255)
     resource_description = RichTextField(blank=True, null=True)
@@ -1101,42 +1149,9 @@ def books(self):
         books = Book.objects.live().filter(locale=self.locale).exclude(book_state='unlisted').order_by('title')
         book_data = []
         for book in books:
-            has_faculty_resources = BookFacultyResources.objects.filter(book_faculty_resource=book).exists()
-            has_student_resources = BookStudentResources.objects.filter(book_student_resource=book).exists()
-            try:
-                book_data.append({
-                    'id': book.id,
-                    'cnx_id': book.cnx_id,
-                    'slug': 'books/{}'.format(book.slug),
-                    'book_state': book.book_state,
-                    'title': book.title,
-                    'subjects': book.subjects(),
-                    'is_ap': book.is_ap,
-                    'cover_url': book.cover_url,
-                    'cover_color': book.cover_color,
-                    'high_resolution_pdf_url': book.high_resolution_pdf_url,
-                    'low_resolution_pdf_url': book.low_resolution_pdf_url,
-                    'ibook_link': book.ibook_link,
-                    'ibook_link_volume_2': book.ibook_link_volume_2,
-                    'webview_link': book.webview_link,
-                    'webview_rex_link': book.webview_rex_link,
-                    'bookshare_link': book.bookshare_link,
-                    'kindle_link': book.kindle_link,
-                    'amazon_coming_soon': book.amazon_coming_soon,
-                    'amazon_link': book.amazon_link,
-                    'bookstore_coming_soon': book.bookstore_coming_soon,
-                    'comp_copy_available': book.comp_copy_available,
-                    'salesforce_abbreviation': book.salesforce_abbreviation,
-                    'salesforce_name': book.salesforce_name,
-                    'urls': book.book_urls(),
-                    'last_updated_pdf': book.last_updated_pdf,
-                    'has_faculty_resources': has_faculty_resources,
-                    'has_student_resources': has_student_resources,
-                    'assignable_book': book.assignable_book,
-                    'promote_tags': [snippet.value.name for snippet in book.promote_snippet],
-                })
-            except Exception as e:
-                capture_exception(e)
+            data = get_book_data(book)
+            if data:
+                book_data.append(data)
         return book_data
 
     content_panels = Page.content_panels + [

codecov.yml

@@ -27,3 +27,7 @@ ignore:
   - '*/settings/*'
   - '*settings.py'
   - '*wsgi.py'
+  - '*/__init__.py'
+  - '*/urls.py'
+  - 'versions/*'  # this is internal only, and not mission critical
+  - 'wagtailimportexport/*'  # this doesn't work, candidate for removal / offloading / rewriting

pages/custom_blocks.py

@@ -8,6 +8,7 @@
 from api.serializers import ImageSerializer
 from openstax.functions import build_image_url, build_document_url
 from wagtail.rich_text import expand_db_html
+from books.models import get_book_data
 
 
 class APIRichTextBlock(blocks.RichTextBlock):
@@ -144,6 +145,7 @@ class QuoteBlock(StructBlock):
     name = blocks.CharBlock(help_text="The name of the person or entity to attribute the quote to.")
     title = blocks.CharBlock(requred=False, help_text="Additional title or label about the quotee.")
 
+
 class DividerBlock(StructBlock):
     image = APIImageChooserBlock()
     config = blocks.StreamBlock([
@@ -323,3 +325,14 @@ def get_api_representation(self, value, context=None):
                 'cover': build_document_url(value['cover'].url),
                 'title': value['title'],
             }
+
+class BookListBlock(blocks.StreamBlock):
+    books = blocks.PageChooserBlock(page_type=['books.Book'], required=False)
+
+    class Meta:
+        icon = 'placeholder'
+        label = "Book List"
+
+    def get_api_representation(self, value, context=None):
+        # value is a StreamValue of blocks, each with .value as a Book page
+        return [book_data for book in value if (book_data := get_book_data(book.value))]