支持配置 ffmpeg 路径 #15

hanxi · Feb 24, 2024 · 54b4417 · 54b4417
1 parent 17d7f54
commit 54b4417
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/xiaomusic/config.py b/xiaomusic/config.py
@@ -88,6 +88,7 @@ class Config:
     search_prefix: str = os.getenv(
         "XIAOMUSIC_SEARCH", "ytsearch:"
     )  # "bilisearch:" or "ytsearch:"
+    ffmpeg_location: str = os.getenv("XIAOMUSIC_FFMPEG_LOCATION", "./ffmpeg/bin")
 
     def __post_init__(self) -> None:
         if self.proxy:

diff --git a/xiaomusic/xiaomusic.py b/xiaomusic/xiaomusic.py
@@ -57,6 +57,7 @@ def __init__(self, config: Config):
         self.port = config.port
         self.proxy = config.proxy
         self.search_prefix = config.search_prefix
+        self.ffmpeg_location = config.ffmpeg_location
 
         # 下载对象
         self.download_proc = None
@@ -311,7 +312,7 @@ async def download(self, name):
             "-o",
             f"{name}.mp3",
             "--ffmpeg-location",
-            "./ffmpeg/bin",
+            f"{self.ffmpeg_location}",
             "--no-playlist",
         )