add audio metadata check of ffprobe data

yajrendrag · Oct 8, 2024 · 2dcb3fe · 2dcb3fe
1 parent f2e5daa
commit 2dcb3fe
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 5 deletions.
diff --git a/source/rename_file/changelog.md b/source/rename_file/changelog.md
@@ -1,4 +1,7 @@
 
+**<span style="color:#56adda">0.0.13</span>**
+- add check for audio metadata presence in ffprobe data
+
 **<span style="color:#56adda">0.0.12</span>**
 - fix typo in elif: clause
 

diff --git a/source/rename_file/info.json b/source/rename_file/info.json
@@ -11,5 +11,5 @@
         "on_postprocessor_task_results": 0
     },
     "tags": "rename, postprocessor",
-    "version": "0.0.12"
+    "version": "0.0.13"
 }
diff --git a/source/rename_file/plugin.py b/source/rename_file/plugin.py
@@ -293,13 +293,15 @@ def replace(data, settings, abspath, streams):
 
     acodec = ''
     if astream:
-        audio_codec = streams[astream]["codec_name"]
-        channel_layout = streams[astream]["channel_layout"]
-        channels = streams[astream]["channels"]
+        audio_codec = streams[astream]["codec_name"] if "codec_name" in streams[astream] else ""
+        channel_layout = streams[astream]["channel_layout"] if "channel_layout" in streams[astream] else ""
+        channels = streams[astream]["channels"] if "channels" in streams[astream] else ""
         if channel_layout == "stereo" and channels == 2:
             acodec = audio_codec + ' ' + str(channels) + '.0'
-        elif channels > 4:
+        elif channels > 4 and channel_layout:
             acodec = audio_codec + ' ' + str(channel_layout).replace('(side)','')
+        elif channels > 4 and not channel_layout:
+            acodec = audio_codec.upper() + ' ' + audio if audio_codec.upper() not in audio else audio
     logger.debug("acodec: '{}'".format(acodec))
 
     if basename.find(codec) > 0: