⚡️ Rewrite download manager

backend/__init__.py

@@ -23,9 +23,5 @@ def get_progress():
 @app.post("/download")
 def download():
     parent_dir = Path(request.json["parent_dir"])
-    id = request.json["id"]
-    try:
-        manager.download(id, parent_dir)
+    manager.download(parent_dir)
     return "OK"
-    except:
-        return "Not ready"

frontend/src/views/AudioDownload.vue

@@ -11,33 +11,8 @@ const state_list = ref([]);
 
 const auto_download = ref(false);
 
-onMounted(() => {
-  setInterval(() => {
-    axios.get("http://localhost:8000/progress").then((resp) => {
-      state_list.value = resp.data;
-      for (const [i, s] of state_list.value.entries()) {
-        if (
-          auto_download.value &&
-          counter.value.downloading == 0 &&
-          s.number &&
-          s.title &&
-          s.total == 0
-        ) {
-          axios.post("http://localhost:8000/download", {
-            id: i,
-            parent_dir: "/home/zhao/Desktop/tmp",
-          });
-          break;
-        }
-      }
-    });
-  }, 500);
-});
-
 const counter = computed(() => {
-  if (state_list.value.length == 0) {
-    return { fetching: true, downloading: false };
-  }
+  const total = state_list.value.length;
   let ready = 0;
   let downloading = 0;
   let finished = 0;
@@ -45,30 +20,55 @@ const counter = computed(() => {
     if (s.number && s.title && s.total == 0) {
       ready++;
     } else if (s.received < s.total) {
+      ready++;
       downloading++;
-    } else if (s.total <= s.received && s.total != 0) {
+    } else if (s.total != 0 && s.received == s.total) {
+      ready++;
       finished++;
     }
   });
-  return { ready: ready, downloading: downloading, finished: finished };
+  return {
+    total: total,
+    ready: ready,
+    downloading: downloading,
+    finished: finished,
+  };
 });
 
 const progress_info = computed(() => {
-  const total = state_list.value.length;
-  const { ready, downloading, finished } = counter.value;
+  const { ready, downloading, finished, total } = counter.value;
   const fetch_info = "Fetching Information";
+  const ready_info = "Ready";
   const download_info = "Downloading";
   const finished_info = "Downloads Complete";
-  if (total == 0) {
-    return { display: fetch_info, finished: 0, total: 0 };
-  } else if (ready < total && downloading == 0 && finished == 0) {
+  if (ready < total) {
     return { display: fetch_info, finished: ready, total: total };
-  } else if (finished + downloading + ready == total && finished < total) {
-    return { display: download_info, finished: finished + downloading, total: total };
+  } else if (ready == total && !downloading && !finished) {
+    return { display: ready_info, finished: ready, total: total };
+  } else if (finished < total) {
+    return {
+      display: download_info,
+      finished: finished,
+      total: total,
+    };
   } else if (finished == total) {
     return { display: finished_info, finished: finished, total: total };
   }
 });
+
+function start_download() {
+  axios.post("http://localhost:8000/download", {
+    parent_dir: "/home/zhao/Desktop/tmp",
+  });
+}
+
+onMounted(() => {
+  setInterval(() => {
+    axios.get("http://localhost:8000/progress").then((resp) => {
+      state_list.value = resp.data;
+    });
+  }, 200);
+});
 </script>
 
 <template>
@@ -87,7 +87,8 @@ const progress_info = computed(() => {
     </div>
     <DownloadButton
       @enable-download="auto_download = true"
-      v-if="counter.ready == state_list.length"
+      v-if="!auto_download && counter.ready == counter.total"
+      @click="start_download"
     ></DownloadButton>
     <TotalProgress v-else v-bind="progress_info"></TotalProgress>
   </div>

manager/__init__.py

@@ -1,13 +1,22 @@
 from video import Video
 from threading import Thread
+from queue import SimpleQueue
 from sanitize_filename import sanitize
 from pathlib import Path
+from time import sleep
 
 
 class Manager:
     def __init__(self):
         self.video_list: list[Video] = []
         self.bvid_set = set()
+        self.queue = SimpleQueue()
+
+    def get_info_worker(self):
+        while not self.queue.empty():
+            video: Video = self.queue.get()
+            video.get_info()
+            sleep(5)
 
     def add_videos_by_number(self, video_numbers: list[str]):
         for video_number in video_numbers:
@@ -18,19 +27,23 @@ class Manager:
             if (bvid := video.v.get_bvid()) not in self.bvid_set:
                 self.bvid_set.add(bvid)
                 self.video_list.append(video)
-                Thread(target=video.get_info).start()
+                self.queue.put(video)
+        Thread(target=self.get_info_worker).start()
 
-    def download(self, id: int, parent_dir: str | Path = Path(".")):
+    def download_worker(self, parent_dir: Path):
+        while not self.queue.empty():
+            video: Video = self.queue.get()
+            filename = sanitize(video.title) + ".m4a"
+            video.download(parent_dir / filename)
+            sleep(5)
+
+    def download(self, parent_dir: str | Path = Path(".")):
         if isinstance(parent_dir, str):
             parent_dir = Path(parent_dir)
-        if not 0 <= id < len(self.video_list):
-            raise Exception(f"id ({id}) out of range!")
-        video = self.video_list[id]
-        if not hasattr(video, "title"):
-            raise Exception(f"No information for video {video.number}")
-        filename = sanitize(video.title) + ".m4a"
-        t = Thread(target=video.download, args=(parent_dir / filename,))
-        t.start()
+        for video in self.video_list:
+            print(video)
+            self.queue.put(video)
+        Thread(target=self.download_worker, args=(parent_dir,)).start()
 
     def get_progress(self):
         return [

video/__init__.py

@@ -1,8 +1,13 @@
 from .exceptions import *
-from bilibili_api import video, sync, HEADERS
+from bilibili_api import video, sync
 from httpx import stream
 from os import PathLike
 
+HEADERS = {
+    "User-Agent": "Mozilla/5.0 (X11; Linux x86_64; rv:109.0) Gecko/20100101 Firefox/110.0",
+    "Referer": "https://www.bilibili.com",
+}
+
 
 class Video:
     def __init__(self, video_number: str):