feat: 优化批量下载工具命名和下载高码率音频

hanxi · Dec 11, 2024 · 23882cf · 23882cf
1 parent 73df9de
commit 23882cf
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/xiaomusic/utils.py b/xiaomusic/utils.py
@@ -866,6 +866,8 @@ async def download_playlist(config, url, dirname):
         "-x",
         "--audio-format",
         "mp3",
+        "--audio-quality",
+        "0",
         "--paths",
         config.download_path,
         "-o",
@@ -899,6 +901,8 @@ async def download_one_music(config, url, name=""):
         "-x",
         "--audio-format",
         "mp3",
+        "--audio-quality",
+        "0",
         "--paths",
         config.download_path,
         "-o",
@@ -947,13 +951,17 @@ def remove_common_prefix(directory):
 
     log.info(f'Common prefix identified: "{common_prefix}"')
 
+    pattern = re.compile(r"(\d+)[\t 　]*\1")
     for filename in files:
         if filename == common_prefix:
             continue
         # 检查文件名是否以共同前缀开头
         if filename.startswith(common_prefix):
             # 构造新的文件名
             new_filename = filename[len(common_prefix) :]
+            match = pattern.match(new_filename)
+            if match:
+                new_filename = match.group(1) + new_filename[match.end() :]
             # 生成完整的文件路径
             old_file_path = os.path.join(directory, filename)
             new_file_path = os.path.join(directory, new_filename)

diff --git a/xiaomusic/xiaomusic.py b/xiaomusic/xiaomusic.py
@@ -1640,6 +1640,8 @@ async def download(self, search_key, name):
             "-x",
             "--audio-format",
             "mp3",
+            "--audio-quality",
+            "0",
             "--paths",
             self.download_path,
             "-o",