From a5c19db6bfbf03e17287de378c4bf556a2c908ea Mon Sep 17 00:00:00 2001
From: Audionut <audionut11@gmail.com>
Date: Sun, 18 Aug 2024 14:26:18 +1000
Subject: [PATCH] Use command line mediainfo

---
 src/prep.py | 79 +++++++++++++++++++++++++++++------------------------
 1 file changed, 44 insertions(+), 35 deletions(-)

diff --git a/src/prep.py b/src/prep.py
index 806035afc..5a0caf1f7 100644
--- a/src/prep.py
+++ b/src/prep.py
@@ -476,53 +476,62 @@ def get_video(self, videoloc, mode):
     Get and parse mediainfo
     """
     def exportInfo(self, video, isdir, folder_id, base_dir, export_text):
-        media_info_path = f"{base_dir}/tmp/{folder_id}/MEDIAINFO.txt"
-        media_info_clean_path = f"{base_dir}/tmp/{folder_id}/MEDIAINFO_CLEANPATH.txt"
-        media_info_json_path = f"{base_dir}/tmp/{folder_id}/MediaInfo.json"
+        # Path to the MediaInfo CLI executable
+        mediainfo_cli_path = 'D:\\mediainfo\\MediaInfo.exe'  # Correct path to the MediaInfo CLI executable
 
-        if not os.path.exists(media_info_path) and export_text:
+        if not os.path.exists(f"{base_dir}/tmp/{folder_id}/MEDIAINFO.txt") and export_text:
             console.print("[bold yellow]Exporting MediaInfo...")
             
             # MediaInfo to text
-            media_info = MediaInfo.parse(video, output="STRING", full=False, mediainfo_options={'inform_version': '1'})
+            if not isdir:
+                os.chdir(os.path.dirname(video))
             
-            with open(media_info_path, 'w', newline="", encoding='utf-8') as export:
-                export.write(media_info)
+            # Use MediaInfo CLI to get text output with utf-8 encoding
+            result = subprocess.run(
+                [mediainfo_cli_path, '--Output=TEXT', video],
+                stdout=subprocess.PIPE,
+                stderr=subprocess.PIPE,
+                encoding='utf-8'
+            )
             
-            with open(media_info_clean_path, 'w', newline="", encoding='utf-8') as export_cleanpath:
-                export_cleanpath.write(media_info.replace(video, os.path.basename(video)))
-            
-            # Read the original content
-            with open(media_info_path, 'r', encoding='utf-8') as infile:
-                original_content = infile.read()
-
-            # Apply transformations similar to `tr` and `sed`
-            processed_content = original_content.replace('\n', '\r')
-            processed_content = processed_content.replace('\r\r', '\n')
-            processed_content = processed_content.replace('\r', '')
+            if result.returncode != 0:
+                console.print(f"[bold red]Error: {result.stderr}")
+                return None
 
-            # Write the processed content to the same file
-            with open(media_info_path, 'w', encoding='utf-8') as outfile:
-                outfile.write(processed_content)
+            media_info = result.stdout
+            
+            if media_info:
+                with open(f"{base_dir}/tmp/{folder_id}/MEDIAINFO.txt", 'w', newline="", encoding='utf-8') as export:
+                    export.write(media_info)
 
-            # Calculate and report the number of characters removed
-            original_length = len(original_content)
-            processed_length = len(processed_content)
-            characters_removed = original_length - processed_length
+                # Save the MediaInfo with clean path
+                with open(f"{base_dir}/tmp/{folder_id}/MEDIAINFO_CLEANPATH.txt", 'w', newline="", encoding='utf-8') as export_cleanpath:
+                    export_cleanpath.write(media_info.replace(video, os.path.basename(video)))
 
-            console.print(f"[bold cyan]{characters_removed} characters were removed during processing.")
             console.print("[bold green]MediaInfo Exported.")
-        
-        if not os.path.exists(f"{media_info_json_path}.txt"):
-            # MediaInfo to JSON
-            media_info = MediaInfo.parse(video, output="JSON", mediainfo_options={'inform_version': '1'})
+
+        if not os.path.exists(f"{base_dir}/tmp/{folder_id}/MediaInfo.json"):
+            # MediaInfo to JSON using the CLI
+            result = subprocess.run(
+                [mediainfo_cli_path, '--Output=JSON', video],
+                stdout=subprocess.PIPE,
+                stderr=subprocess.PIPE,
+                encoding='utf-8'
+            )
             
-            with open(media_info_json_path, 'w', encoding='utf-8') as export:
-                export.write(media_info)
-        
-        with open(media_info_json_path, 'r', encoding='utf-8') as f:
+            if result.returncode != 0:
+                console.print(f"[bold red]Error: {result.stderr}")
+                return None
+            
+            media_info_json = result.stdout
+            
+            if media_info_json:
+                with open(f"{base_dir}/tmp/{folder_id}/MediaInfo.json", 'w', encoding='utf-8') as export:
+                    export.write(media_info_json)
+            
+        with open(f"{base_dir}/tmp/{folder_id}/MediaInfo.json", 'r', encoding='utf-8') as f:
             mi = json.load(f)
-        
+
         return mi