Merge branch 'develop' into daisy

Author: pauliyobo

bookworm/gui/book_viewer/menu_constants.py

@@ -45,6 +45,9 @@ class ViewerMenuIds(enum.IntEnum):
     about = next(ID_GEN)
     clear_documents_cache = next(ID_GEN)
 
+class ImportMenuIds(enum.IntEnum):
+    """Ids for the import submenu"""
+    qread = next(ID_GEN)
 
 FOCUS_TABLE_OF_CONTENTS = next(ID_GEN)
 

bookworm/gui/book_viewer/menubar.py

@@ -15,7 +15,7 @@
 import wx
 from slugify import slugify
 
-from bookworm import app, config, ocr, paths, speech
+from bookworm import app, config, ocr, paths, qread, speech
 from bookworm.commandline_handler import run_subcommand_in_a_new_process
 from bookworm.concurrency import call_threaded, process_worker
 from bookworm.document import READING_MODE_LABELS
@@ -89,12 +89,23 @@ def create(self):
         self.pinnedDocumentsMenu = wx.Menu()
         # A submenu for recent documents
         self.recentFilesMenu = wx.Menu()
+        # a submenu to import books from external sources
+        self.import_menu = wx.Menu()
         # File menu
         # Translators: the label of an item in the application menubar
         self.Append(wx.ID_OPEN, _("Open...\tCtrl-O"))
         # Translators: the label of an item in the application menubar
         self.Append(wx.ID_NEW, _("New Window...\tCtrl-N"))
         self.AppendSeparator()
+        self.AppendSubMenu(
+            self.import_menu,
+            # translators, the label of an item in the application menubar
+            _("i&mport"),
+            # Translators: the help text of an item in the application menubar
+            _("Import a book from an external resource")
+        )
+        self.import_menu.Append(ImportMenuIds.qread, _("&Import QRD file"))
+        self.AppendSeparator()
         # Translators: the label of an item in the application menubar
         self.Append(
             BookRelatedMenuIds.pin_document, _("&Pin\tCtrl-P"), kind=wx.ITEM_CHECK
@@ -163,6 +174,7 @@ def create(self):
         # Bind event handlers
         self.view.Bind(wx.EVT_MENU, self.onOpenEBook, id=wx.ID_OPEN)
         self.view.Bind(wx.EVT_MENU, self.onNewWindow, id=wx.ID_NEW)
+        self.view.Bind(wx.EVT_MENU, self.onImportQread, id=ImportMenuIds.qread)
         self.view.Bind(
             wx.EVT_MENU, self.onClearPinDocuments, id=self.clearPinnedDocumentsID
         )
@@ -230,6 +242,29 @@ def onNewWindow(self, event):
             args.append("--debug")
         run_subcommand_in_a_new_process(args, hidden=False)
 
+    def onImportQread(self, event):
+        with wx.FileDialog(
+            self.view,
+            # translators: the title of a file dialog to import qrd files
+            message = _("Import QRD file"),
+            style=wx.FD_OPEN | wx.FD_FILE_MUST_EXIST,
+            wildcard=("*.qrd")
+        ) as dlg:
+            if dlg.ShowModal() != wx.ID_OK:
+                return
+            
+            qrd_path = dlg.GetPath()
+            book_info = qread.get_book_info(qrd_path)
+        if not book_info:
+            return self.view.notify_user(
+                _("Failed to import QRD file"),
+                _("The QRD file could not be imported"),
+                icon=wx.ICON_ERROR
+            )
+        uri = DocumentUri.from_filename(book_info.original_path)
+        uri.openner_args = {"position": book_info.current_position}
+        return self.view.open_uri(uri)
+
     def onClearPinDocuments(self, event):
         recents_manager.clear_pinned()
         self.populate_pinned_documents_list()

bookworm/qread/__init__.py

@@ -0,0 +1,37 @@
+"""QRead interoperability"""
+import sqlite3
+from typing import Optional
+
+from pydantic import BaseModel, ValidationError
+
+from bookworm.logger import logger
+
+log = logger.getChild(__name__)
+
+class BookInfo(BaseModel):
+    """Saved book information
+    QRead stores book metadata inside a .qrd file, which is an unencrypted sqlite database
+    The table we are interested from is named shelf, and it contains information related to book position, among other things
+    """
+    # Current position of the book
+    current_position: int
+    # File location of the book
+    original_path: str
+
+
+def get_book_info(qrd_path: str) -> Optional[BookInfo]:
+    """Connects to the .qrd sqlite database and retrieves the book information we require, if any"""
+    log.debug(f"Connecting to {qrd_path}")
+    with sqlite3.connect(qrd_path) as conn:
+        cur = conn.cursor()
+        # the shelf table is made up by two columns, key and value
+        cur.execute('SELECT value FROM shelf WHERE key="book.json"')
+        try:
+            value = cur.fetchone()[0]
+            info = BookInfo.model_validate_json(value)
+            return info
+        except ValidationError:
+            log.error("Failed to obtain book information from QRD file", exc_info=True)
+            return None
+        finally:
+            cur.close()