mkv-episode-matcher 0.3.4__py3-none-any.whl → 0.3.5__py3-none-any.whl

mkv_episode_matcher/episode_identification.py

@@ -9,6 +9,10 @@ from loguru import logger
 import whisper
 import numpy as np
 import re
+from pathlib import Path
+import chardet
+from loguru import logger
+
 class EpisodeMatcher:
     def __init__(self, cache_dir, show_name, min_confidence=0.6):
         self.cache_dir = Path(cache_dir)
@@ -50,34 +54,32 @@ class EpisodeMatcher:
         return str(chunk_path)
 
     def load_reference_chunk(self, srt_file, chunk_idx):
-        """Load reference subtitles for a specific time chunk."""
+        """
+        Load reference subtitles for a specific time chunk with robust encoding handling.
+        
+        Args:
+            srt_file (str or Path): Path to the SRT file
+            chunk_idx (int): Index of the chunk to load
+            
+        Returns:
+            str: Combined text from the subtitle chunk
+        """
         chunk_start = chunk_idx * self.chunk_duration
         chunk_end = chunk_start + self.chunk_duration
-        text_lines = []
         
-        with open(srt_file, 'r', encoding='utf-8') as f:
-            content = f.read().strip()
+        try:
+            # Read the file content using our robust reader
+            reader = SubtitleReader()
+            content = reader.read_srt_file(srt_file)
             
-        for block in content.split('\n\n'):
-            lines = block.split('\n')
-            if len(lines) < 3 or '-->' not in lines[1]:  # Skip malformed blocks
-                continue
-                
-            try:
-                timestamp = lines[1]
-                text = ' '.join(lines[2:])
-                
-                end_time = timestamp.split(' --> ')[1].strip()
-                hours, minutes, seconds = map(float, end_time.replace(',','.').split(':'))
-                total_seconds = hours * 3600 + minutes * 60 + seconds
-                
-                if chunk_start <= total_seconds <= chunk_end:
-                    text_lines.append(text)
-                    
-            except (IndexError, ValueError):
-                continue
-                
-        return ' '.join(text_lines)
+            # Extract subtitles for the time chunk
+            text_lines = reader.extract_subtitle_chunk(content, chunk_start, chunk_end)
+            
+            return ' '.join(text_lines)
+            
+        except Exception as e:
+            logger.error(f"Error loading reference chunk from {srt_file}: {e}")
+            return ''
 
     def identify_episode(self, video_file, temp_dir, season_number):
         try:
@@ -147,4 +149,121 @@ class EpisodeMatcher:
         finally:
             # Cleanup temp files
             for file in self.temp_dir.glob("chunk_*.wav"):
-                file.unlink()
+                file.unlink()
+
+def detect_file_encoding(file_path):
+    """
+    Detect the encoding of a file using chardet.
+    
+    Args:
+        file_path (str or Path): Path to the file
+        
+    Returns:
+        str: Detected encoding, defaults to 'utf-8' if detection fails
+    """
+    try:
+        with open(file_path, 'rb') as f:
+            raw_data = f.read()
+        result = chardet.detect(raw_data)
+        encoding = result['encoding']
+        confidence = result['confidence']
+        
+        logger.debug(f"Detected encoding {encoding} with {confidence:.2%} confidence for {file_path}")
+        return encoding if encoding else 'utf-8'
+    except Exception as e:
+        logger.warning(f"Error detecting encoding for {file_path}: {e}")
+        return 'utf-8'
+
+def read_file_with_fallback(file_path, encodings=None):
+    """
+    Read a file trying multiple encodings in order of preference.
+    
+    Args:
+        file_path (str or Path): Path to the file
+        encodings (list): List of encodings to try, defaults to common subtitle encodings
+        
+    Returns:
+        str: File contents
+        
+    Raises:
+        ValueError: If file cannot be read with any encoding
+    """
+    if encodings is None:
+        # First try detected encoding, then fallback to common subtitle encodings
+        detected = detect_file_encoding(file_path)
+        encodings = [detected, 'utf-8', 'latin-1', 'cp1252', 'iso-8859-1']
+    
+    file_path = Path(file_path)
+    errors = []
+    
+    for encoding in encodings:
+        try:
+            with open(file_path, 'r', encoding=encoding) as f:
+                content = f.read()
+            logger.debug(f"Successfully read {file_path} using {encoding} encoding")
+            return content
+        except UnicodeDecodeError as e:
+            errors.append(f"{encoding}: {str(e)}")
+            continue
+            
+    error_msg = f"Failed to read {file_path} with any encoding. Errors:\n" + "\n".join(errors)
+    logger.error(error_msg)
+    raise ValueError(error_msg)
+
+class SubtitleReader:
+    """Helper class for reading and parsing subtitle files."""
+    
+    @staticmethod
+    def parse_timestamp(timestamp):
+        """Parse SRT timestamp into seconds."""
+        hours, minutes, seconds = timestamp.replace(',', '.').split(':')
+        return float(hours) * 3600 + float(minutes) * 60 + float(seconds)
+    
+    @staticmethod
+    def read_srt_file(file_path):
+        """
+        Read an SRT file and return its contents with robust encoding handling.
+        
+        Args:
+            file_path (str or Path): Path to the SRT file
+            
+        Returns:
+            str: Contents of the SRT file
+        """
+        return read_file_with_fallback(file_path)
+    
+    @staticmethod
+    def extract_subtitle_chunk(content, start_time, end_time):
+        """
+        Extract subtitle text for a specific time window.
+        
+        Args:
+            content (str): Full SRT file content
+            start_time (float): Chunk start time in seconds
+            end_time (float): Chunk end time in seconds
+            
+        Returns:
+            list: List of subtitle texts within the time window
+        """
+        text_lines = []
+        
+        for block in content.strip().split('\n\n'):
+            lines = block.split('\n')
+            if len(lines) < 3 or '-->' not in lines[1]:
+                continue
+                
+            try:
+                timestamp = lines[1]
+                text = ' '.join(lines[2:])
+                
+                end_stamp = timestamp.split(' --> ')[1].strip()
+                total_seconds = SubtitleReader.parse_timestamp(end_stamp)
+                
+                if start_time <= total_seconds <= end_time:
+                    text_lines.append(text)
+                    
+            except (IndexError, ValueError) as e:
+                logger.warning(f"Error parsing subtitle block: {e}")
+                continue
+                
+        return text_lines

{mkv_episode_matcher-0.3.4.dist-info → mkv_episode_matcher-0.3.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: mkv-episode-matcher
-Version: 0.3.4
+Version: 0.3.5
 Summary: The MKV Episode Matcher is a tool for identifying TV series episodes from MKV files and renaming the files accordingly.
 Home-page: https://github.com/Jsakkos/mkv-episode-matcher
 Author: Jonathan Sakkos
@@ -51,6 +51,14 @@ Automatically match and rename your MKV TV episodes using The Movie Database (TM
 - ✨ **Bulk Processing**: Handle entire seasons at once
 - 🧪 **Dry Run Mode**: Test changes before applying
 
+## Prerequisites
+
+- Python 3.9 or higher
+- [FFmpeg](https://ffmpeg.org/download.html) installed and available in system PATH
+- [Tesseract OCR](https://github.com/UB-Mannheim/tesseract/wiki) installed (required for image-based subtitle processing)
+- TMDb API key
+- OpenSubtitles account (optional, for subtitle downloads)
+
 ## Quick Start
 
 1. Install the package:
@@ -60,37 +68,13 @@ pip install mkv-episode-matcher
 
 2. Run on your show directory:
 ```bash
-mkv-match --show-dir "path/to/your/show" --season 1
+mkv-match --show-dir "path/to/your/show" --get-subs true
 ```
 
-## Requirements
-
-- Python 3.8 or higher
-- TMDb API key
-- OpenSubtitles account (optional, for subtitle downloads)
-
 ## Documentation
 
 Full documentation is available at [https://jsakkos.github.io/mkv-episode-matcher/](https://jsakkos.github.io/mkv-episode-matcher/)
 
-## Basic Usage
-
-```python
-from mkv_episode_matcher import process_show
-
-# Process all seasons
-process_show()
-
-# Process specific season
-process_show(season=1)
-
-# Test run without making changes
-process_show(season=1, dry_run=True)
-
-# Process and download subtitles
-process_show(get_subs=True)
-```
-
 ## Directory Structure
 
 MKV Episode Matcher expects your TV shows to be organized as follows:
@@ -105,6 +89,23 @@ Show Name/
 │   └── episode2.mkv
 ```
 
+## Reference Subtitle File Structure
+
+Subtitle files that are not automatically downloaded using the `--get-subs` flag should be named as follows:
+
+```
+
+~/.mkv-episode-matcher/cache/data/Show Name/
+├── Show Name - S01E01.srt
+├── Show Name - S01E02.srt
+└── ...
+```
+
+On Windows, the cache directory is located at `C:\Users\{username}\.mkv-episode-matcher\cache\data\`
+
+Reference subtitle files should follow this naming pattern:
+`{show_name} - S{season:02d}E{episode:02d}.srt`
+
 ## Contributing
 
 1. Fork the repository

{mkv_episode_matcher-0.3.4.dist-info → mkv_episode_matcher-0.3.5.dist-info}/RECORD

@@ -2,7 +2,7 @@ mkv_episode_matcher/.gitattributes,sha256=Gh2-F2vCM7SZ01pX23UT8pQcmauXWfF3gwyRSb
 mkv_episode_matcher/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 mkv_episode_matcher/__main__.py,sha256=3ZcCUxeI7rUA-4oiCD2WXBiOFJAqLsVVWfZKN446FwQ,6792
 mkv_episode_matcher/config.py,sha256=zDDKBcsDt5fME9BRqiTi7yWKeast1pZh36BNYMvIBYM,2419
-mkv_episode_matcher/episode_identification.py,sha256=NopEkcBFFUjjrAujogeVcdISv8UZHFjYr5RJLM0j468,5875
+mkv_episode_matcher/episode_identification.py,sha256=xYqHq1YFbZT8L1Gfa_DhSStrLblKTWxZte__B0qikQU,9739
 mkv_episode_matcher/episode_matcher.py,sha256=BJ76DPxsmZs-KfHZZ_0WvKSBZWXsUEO6lW34YdYEaxM,3979
 mkv_episode_matcher/mkv_to_srt.py,sha256=4yxBHRVhgVby0UtQ2aTXGuoQpid8pkgjMIaHU6GCdzc,10857
 mkv_episode_matcher/speech_to_text.py,sha256=-bnGvmtPCKyHFPEaXwIcEYTf_P13rNpAJA-2UFeRFrs,2806
@@ -19,8 +19,8 @@ mkv_episode_matcher/libraries/pgs2srt/requirements.txt,sha256=sg87dqWw_qpbwciw-M
 mkv_episode_matcher/libraries/pgs2srt/Libraries/SubZero/SubZero.py,sha256=geT1LXdVd8yED9zoJ9K1XfP2JzGcM7u1SslHYrJI09o,10061
 mkv_episode_matcher/libraries/pgs2srt/Libraries/SubZero/post_processing.py,sha256=GKtVy_Lxv-z27mkRG8pJF2znKWXwZTot7jL6kN-zIxM,10503
 mkv_episode_matcher/libraries/pgs2srt/Libraries/SubZero/dictionaries/data.py,sha256=AlJHUYXl85J95OzGRik-AHVfzDd7Q8BJCvD4Nr8kRIk,938598
-mkv_episode_matcher-0.3.4.dist-info/METADATA,sha256=QPa0StsF0ADrzxSEswvQ4tTbBkztRjR82hNFsXnJwCc,4640
-mkv_episode_matcher-0.3.4.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-mkv_episode_matcher-0.3.4.dist-info/entry_points.txt,sha256=IglJ43SuCZq2eQ3shMFILCkmQASJHnDCI3ogohW2Hn4,64
-mkv_episode_matcher-0.3.4.dist-info/top_level.txt,sha256=XRLbd93HUaedeWLtkyTvQjFcE5QcBRYa3V-CfHrq-OI,20
-mkv_episode_matcher-0.3.4.dist-info/RECORD,,
+mkv_episode_matcher-0.3.5.dist-info/METADATA,sha256=mTKSbM9Ai5UDKyj2K4AKgkdjdPVEaxylfHHp95wVZv4,5048
+mkv_episode_matcher-0.3.5.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+mkv_episode_matcher-0.3.5.dist-info/entry_points.txt,sha256=IglJ43SuCZq2eQ3shMFILCkmQASJHnDCI3ogohW2Hn4,64
+mkv_episode_matcher-0.3.5.dist-info/top_level.txt,sha256=XRLbd93HUaedeWLtkyTvQjFcE5QcBRYa3V-CfHrq-OI,20
+mkv_episode_matcher-0.3.5.dist-info/RECORD,,

{mkv_episode_matcher-0.3.4.dist-info → mkv_episode_matcher-0.3.5.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.6.0)
+Generator: setuptools (75.8.0)
 Root-Is-Purelib: true
 Tag: py3-none-any