feat: support files from non-C drives

bugale · Apr 27, 2024 · eea4f3e · eea4f3e
1 parent 7e76eca
commit eea4f3e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/buganime/transcode.py b/buganime/transcode.py
@@ -4,6 +4,7 @@
 import asyncio
 import logging
 import warnings
+import shutil
 from dataclasses import dataclass
 from typing import AsyncIterator, cast, Optional
 
@@ -83,7 +84,10 @@ async def __read_input_frames(self) -> AsyncIterator[bytes]:
 
     async def __write_output_frames(self, frames: AsyncIterator[bytes]) -> None:
         with tempfile.TemporaryDirectory() as temp_dir:
-            os.link(self.__input_path, os.path.join(temp_dir, 'input.mkv'))
+            if os.path.splitdrive(self.__input_path)[0] == os.path.splitdrive(temp_dir)[0]:
+                os.link(self.__input_path, os.path.join(temp_dir, 'input.mkv'))
+            else:
+                shutil.copy(self.__input_path, os.path.join(temp_dir, 'input.mkv'))
             width_out = self.__width_out
             height_out = self.__height_out
             if self.__video_info.width / self.__video_info.height > self.__width_out / self.__height_out: