feat: adding title

get_metadata.py

@@ -10,10 +10,10 @@ SUPPORTED_EXTENSIONS = (".mp4", ".mkv", ".mov", ".avi")
 def get_video_metadata(file_path):
     """
     Extract metadata from a video file using ffprobe.
-    
+
     Args:
         file_path (str): Path to the video file
-        
+
     Returns:
         dict: Metadata information
     """
@@ -22,29 +22,31 @@ def get_video_metadata(file_path):
         "ffprobe", "-v", "quiet", "-print_format", "json",
         "-show_format", "-show_streams", file_path
     ]
-    
+
     try:
         result = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
         if result.returncode != 0:
             print(f"❌ Error processing {file_path}: {result.stderr}")
             return None
-            
+
         data = json.loads(result.stdout)
-        
+
-        # Extract filename
+        # Extract filename and title
         filename = os.path.basename(file_path)
-        
+        title = data.get("format", {}).get("tags", {}).get("title", filename)
+
         # Initialize metadata structure
         metadata = {
             "filename": filename,
+            "title": title,
             "audio_tracks": [],
             "subtitle_tracks": []
         }
-        
+
         # Process streams
         for stream in data.get("streams", []):
             codec_type = stream.get("codec_type")
-            
+
             if codec_type == "audio":
                 track = {
                     "index": stream.get("index"),
@@ -63,7 +65,7 @@ def get_video_metadata(file_path):
                     }
                 }
                 metadata["audio_tracks"].append(track)
-                
+
             elif codec_type == "subtitle":
                 track = {
                     "index": stream.get("index"),
@@ -81,9 +83,9 @@ def get_video_metadata(file_path):
                     }
                 }
                 metadata["subtitle_tracks"].append(track)
-        
+
         return metadata
-        
+
     except Exception as e:
         print(f"❌ Error processing {file_path}: {str(e)}")
         return None
@@ -91,7 +93,7 @@ def get_video_metadata(file_path):
 def process_file(file_path, output_file=None):
     """
     Process a single video file and write metadata to JSON.
-    
+
     Args:
         file_path (str): Path to the video file
         output_file (str, optional): Path to output JSON file
@@ -99,33 +101,33 @@ def process_file(file_path, output_file=None):
     if not os.path.isfile(file_path):
         print(f"❌ File not found: {file_path}")
         return False
-        
+
     if not file_path.lower().endswith(SUPPORTED_EXTENSIONS):
         print(f"❌ Unsupported file format: {file_path}")
         return False
-    
+
     print(f"📊 Extracting metadata from {os.path.basename(file_path)}")
     metadata = get_video_metadata(file_path)
-    
+
     if metadata:
         if not output_file:
             # Generate output filename based on input file
             base_name = os.path.splitext(file_path)[0]
             output_file = f"{base_name}_metadata.json"
-            
+
         # Write metadata to JSON file
         with open(output_file, 'w', encoding='utf-8') as f:
             json.dump(metadata, f, indent=2, ensure_ascii=False)
-            
+
         print(f"✅ Metadata saved to {output_file}")
         return True
-    
+
     return False
 
 def process_directory(directory_path, output_file=None):
     """
     Process all video files in a directory and write metadata to JSON.
-    
+
     Args:
         directory_path (str): Path to the directory
         output_file (str, optional): Path to output JSON file
@@ -133,36 +135,36 @@ def process_directory(directory_path, output_file=None):
     if not os.path.isdir(directory_path):
         print(f"❌ Directory not found: {directory_path}")
         return False
-    
+
     all_metadata = {}
     file_count = 0
-    
+
     for root, _, files in os.walk(directory_path):
         for file in files:
             if file.lower().endswith(SUPPORTED_EXTENSIONS):
                 file_path = os.path.join(root, file)
                 print(f"📊 Extracting metadata from {file}")
                 metadata = get_video_metadata(file_path)
-                
+
                 if metadata:
                     # Use relative path as key
                     rel_path = os.path.relpath(file_path, directory_path)
                     all_metadata[rel_path] = metadata
                     file_count += 1
-    
+
     if file_count == 0:
         print(f"❌ No supported video files found in {directory_path}")
         return False
-    
+
     if not output_file:
         # Generate output filename based on directory name
         dir_name = os.path.basename(os.path.normpath(directory_path))
         output_file = f"{dir_name}_metadata.json"
-    
+
     # Write all metadata to a single JSON file
     with open(output_file, 'w', encoding='utf-8') as f:
         json.dump(all_metadata, f, indent=2, ensure_ascii=False)
-    
+
     print(f"✅ Metadata for {file_count} files saved to {output_file}")
     return True
 
@@ -171,10 +173,10 @@ def main():
     parser.add_argument("input", help="Path to input video file or directory")
     parser.add_argument("-o", "--output", help="Path to output JSON file")
     args = parser.parse_args()
-    
+
     input_path = args.input
     output_file = args.output
-    
+
     if os.path.isfile(input_path):
         process_file(input_path, output_file)
     elif os.path.isdir(input_path):