Merge pull request #41 from pganssle/add_reload_params

Add URL params to update

Author: pganssle

src/audio_feeder/cli.py

@@ -91,8 +91,13 @@ def run(host, port, config, profile):
     is_flag=True,
     help="Passed if metadata from all sources should be reloaded if present.",
 )
+@click.option(
+    "--check-hashes",
+    is_flag=True,
+    help="Passed if the updater should re-check hashes existing in the database",
+)
 @click.argument("path", metavar="PATH", type=str, required=True)
-def update(content_type, reload_metadata, path):
+def update(content_type, reload_metadata, check_hashes, path):
     """
     Add a specific path to the databases, loading all content and updating the
     database where necessary.
@@ -114,6 +119,8 @@ def pbar(msg: str) -> typing.Callable[[_T], _T]:
         content_type=content_type,
         path=path,
         progress_bar=pbar("Loading book metadata:"),
+        reload_metadata=reload_metadata,
+        check_hashes=check_hashes,
     )
 
 

src/audio_feeder/pages.py

@@ -361,12 +361,33 @@ def update_status():
     )
 
 
+def _evaluate_bool_param(param: str) -> bool:
+    return param.lower() == "true"
+
+
 @root.route("/update")
 def update():
+
     with UPDATE_LOCK:
         if not updater.UPDATE_IN_PROGRESS:
+            reload_metadata: bool = request.args.get(
+                "reloadMetadata", type=_evaluate_bool_param, default=False
+            )
+            check_hashes: bool = request.args.get(
+                "checkHashes", type=_evaluate_bool_param, default=False
+            )
+
+            if reload_metadata or check_hashes:
+                update_func = functools.partial(
+                    updater.update,
+                    reload_metadata=reload_metadata,
+                    check_hashes=check_hashes,
+                )
+            else:
+                update_func = updater.update
+
             updater.UPDATE_IN_PROGRESS = True
-            background_thread = threading.Thread(target=updater.update, daemon=True)
+            background_thread = threading.Thread(target=update_func, daemon=True)
             background_thread.start()
         return flask.redirect(flask.url_for("root.update_status"))
 

src/audio_feeder/updater.py

@@ -768,7 +768,10 @@ def _clear_update_output() -> None:
 
 
 def _update_books(
-    update_path: str, progress_bar: typing.Optional[_PBar] = None
+    update_path: str,
+    progress_bar: typing.Optional[_PBar] = None,
+    reload_metadata: bool = False,
+    check_hashes: bool = False,
 ) -> None:
     """
     Trigger an update to the database.
@@ -782,10 +785,19 @@ def _update_books(
 
     OpFunction = typing.Callable[[typing.Any], typing.Any]
     ops: typing.Sequence[typing.Tuple[OpFunction, str]] = [
-        (book_updater.update_db_entries, "Updating database entries."),
+        (
+            functools.partial(
+                book_updater.update_db_entries, check_hashes=check_hashes
+            ),
+            "Updating database entries.",
+        ),
         (book_updater.assign_books_to_entries, "Assigning books to entries"),
         (
-            functools.partial(book_updater.update_book_metadata, pbar=progress_bar),
+            functools.partial(
+                book_updater.update_book_metadata,
+                reload_metadata=reload_metadata,
+                pbar=progress_bar,
+            ),
             "Updating book metadata",
         ),
         (book_updater.update_author_db, "Updating author db"),
@@ -815,13 +827,20 @@ def update(
     content_type: typing.Optional[str] = None,
     path: typing.Optional[str] = None,
     progress_bar: typing.Optional[_PBar] = None,
+    reload_metadata: bool = False,
+    check_hashes: bool = False,
 ) -> None:
 
     if path is None:
         path = "."
 
     if content_type is None or content_type == "books":
-        _update_books(update_path=path, progress_bar=progress_bar)
+        _update_books(
+            update_path=path,
+            progress_bar=progress_bar,
+            reload_metadata=reload_metadata,
+            check_hashes=check_hashes,
+        )
     else:
         raise ValueError(f"Unknown content type: {content_type}")