Fix web ui progress indicator and improve restart workers.

justin025 · Jan 14, 2025 · 633fa5a · 633fa5a
1 parent 501e1ee
commit 633fa5a
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 3 deletions.
diff --git a/src/onthespot/downloader.py b/src/onthespot/downloader.py
@@ -349,10 +349,11 @@ def run(self):
                                 downloaded += len(data)
                                 data_chunks += data
 
-                                if self.gui and downloaded != total_size:
+                                if downloaded != total_size:
                                     if item['item_status'] == 'Cancelled':
                                         raise Exception("Download cancelled by user.")
-                                    self.progress.emit(item, self.tr("Downloading"), int((downloaded / total_size) * 100))
+                                    if self.gui:
+                                        self.progress.emit(item, self.tr("Downloading"), int((downloaded / total_size) * 100))
 
                             key = calcbfkey(song["SNG_ID"])
 

diff --git a/src/onthespot/resources/web/download_queue.html b/src/onthespot/resources/web/download_queue.html
@@ -96,7 +96,7 @@
                                     copyButton = createButton('icons/link.png', 'Copy Link', `copyToClipboard('${item.item_url}')`);
                                 }
 
-                                if (item.item_status === 'Waiting') {
+                                if (!['Failed', 'Cancelled', 'Deleted', 'Already Exists', 'Downloaded'].includes(item.item_status)) {
                                     cancelButton = createButton('icons/stop.png', 'Stop Download', `handleCancel('${item.local_id}')`);
                                 }
 
@@ -186,7 +186,13 @@
             let cancelledCount = 0;
 
             for (let row of tableBody.rows) {
+
+                {% if config.show_download_thumbnails %}
                 const statusCell = row.cells[5];
+                {% else %}
+                const statusCell = row.cells[4];
+                {% endif %}
+
                 if (statusCell) {
                     const statusText = statusCell.textContent.trim();
 

diff --git a/src/onthespot/web.py b/src/onthespot/web.py
@@ -181,6 +181,16 @@ def restart_workers():
 
     download_workers.clear()
 
+    item_urls = []
+    for key, value in download_queue.items():
+        item_urls.append(value.get('item_url'))
+
+    with download_queue_lock:
+        download_queue.clear()
+
+    for url in item_urls:
+        parse_url(url)
+
     for i in range(config.get('maximum_download_workers')):
         download_worker = DownloadWorker()
         download_worker.start()