From 7dff2a98f23dd5e11f34444e3a6f17ce3f53ec95 Mon Sep 17 00:00:00 2001
From: Siarhei Siniak <siarheisiniak@gmail.com>
Date: Sat, 6 Jul 2024 17:49:44 +0300
Subject: [PATCH] [~] Refactor

---
 python/tasks/tiktok/__init__.py | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/python/tasks/tiktok/__init__.py b/python/tasks/tiktok/__init__.py
index bdc249f..25052c6 100644
--- a/python/tasks/tiktok/__init__.py
+++ b/python/tasks/tiktok/__init__.py
@@ -289,12 +289,15 @@ def tiktok_videos_process(meta: Iterable[dict[str, Any]]) -> dict[str, Any]:
             path_parts = os.path.splitext(path)
 
             processed_path = path_parts[0] + '-proc' + path_parts[1]
+            processed_path_tmp = path_parts[0] + '-proc.tmp' + path_parts[1]
 
-            print(processed_path)
             if not os.path.exists(path) or os.path.exists(processed_path):
                 stats['skipped'] += 1
                 continue
 
+            if os.path.exists(processed_path_tmp):
+                os.unlink(processed_path_tmp)
+
             ffmpeg = [
                 'ffmpeg',
                 '-i', path,
@@ -308,9 +311,18 @@ def tiktok_videos_process(meta: Iterable[dict[str, Any]]) -> dict[str, Any]:
                 '-sws_flags', 'bilinear',
                 '-map', '0:v:0',
                 '-map', '1:a:0',
-                processed_path,
+                processed_path_tmp,
             ]
-            subprocess.check_call(ffmpeg)
+
+            subprocess.check_call(
+                ffmpeg,
+                stdin=subprocess.DEVNULL,
+                stderr=subprocess.DEVNULL,
+                stdout=subprocess.DEVNULL
+            )
+
+            os.rename(processed_path_tmp, processed_path)
+
             stats['saved'] += 1
         except KeyboardInterrupt:
             break