mkv-episode-matcher 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

mkv_episode_matcher/__init__.py

@@ -0,0 +1 @@
+version = "0.1.1"

mkv_episode_matcher/__main__.py

@@ -1,9 +1,13 @@
 # __main__.py
 import argparse
 import os
+
 from loguru import logger
-from .config import set_config, get_config
 
+from .config import get_config, set_config
+
+# Log the start of the application
+logger.info("Starting the application")
 
 # Check if logs directory exists, if not create it
 if not os.path.exists('./logs'):
@@ -49,8 +53,6 @@ def main():
     The function logs its progress to two separate log files: one for standard output and one for errors.
     """
 
-    # Log the start of the application
-    logger.info("Starting the application")
 
     # Parse command-line arguments
     parser = argparse.ArgumentParser(description="Process shows with TMDb API")

mkv_episode_matcher/config.py

@@ -1,7 +1,8 @@
 # config.py
-import os
 import configparser
 import multiprocessing
+import os
+
 from loguru import logger
 
 MAX_THREADS = 4
@@ -13,6 +14,8 @@ def get_total_threads():
 
 total_threads = get_total_threads()
 
+if total_threads < MAX_THREADS:
+    MAX_THREADS = total_threads
 logger.info(f"Total available threads: {total_threads} -> Setting max to {MAX_THREADS}")
 
 

mkv_episode_matcher/episode_matcher.py

@@ -1,12 +1,14 @@
 # episode_matcher.py
 import os
-from mkv_episode_matcher.config import get_config
-from mkv_episode_matcher.tmdb_client import fetch_show_id
-from mkv_episode_matcher.utils import get_subtitles, cleanup_ocr_files,check_filename
+import re
+
 from loguru import logger
-from mkv_episode_matcher.__main__ import CONFIG_FILE, CACHE_DIR
+
+from mkv_episode_matcher.__main__ import CACHE_DIR, CONFIG_FILE
+from mkv_episode_matcher.config import get_config
 from mkv_episode_matcher.mkv_to_srt import convert_mkv_to_srt
-import re
+from mkv_episode_matcher.tmdb_client import fetch_show_id
+from mkv_episode_matcher.utils import check_filename, cleanup_ocr_files, get_subtitles
 
 
 # hash_data = {}
@@ -98,7 +100,7 @@ def extract_srt_text(filepath):
               Each inner list contains the lines of text for that block.
     """
     # extract the text from the file
-    with open(filepath, "r") as f:
+    with open(filepath) as f:
         filepath = f.read()
     text_lines = [
         filepath.split("\n\n")[i].split("\n")[2:]
@@ -172,7 +174,7 @@ def process_reference_srt_files(series_name):
         for filename in filenames:
             if filename.lower().endswith(".srt"):
                 srt_file = os.path.join(dirpath, filename)
-                print(f"Processing {srt_file}")
+                logger.info(f"Processing {srt_file}")
                 srt_text = extract_srt_text(srt_file)
                 season, episode = extract_season_episode(filename)
                 mkv_filename = f"{series_name} - S{season:02}E{episode:02}.mkv"
@@ -195,7 +197,7 @@ def process_srt_files(show_dir):
         for filename in filenames:
             if filename.lower().endswith(".srt"):
                 srt_file = os.path.join(dirpath, filename)
-                print(f"Processing {srt_file}")
+                logger.info(f"Processing {srt_file}")
                 srt_text = extract_srt_text(srt_file)
                 srt_files[srt_file] = srt_text
     return srt_files

mkv_episode_matcher/mkv_to_srt.py

@@ -1,6 +1,5 @@
 import os
 import subprocess
-
 import sys
 
 # Get the absolute path of the parent directory of the current script.
@@ -12,16 +11,18 @@ sys.path.append(parent_dir)
 sys.path.append(os.path.join(parent_dir, "libraries"))
 # Add the 'libraries' directory to the Python path.
 sys.path.append(os.path.join(parent_dir, "..", "libraries", "pgs2srt"))
-import pytesseract
 import re
-from PIL import Image, ImageOps
-from mkv_episode_matcher.__main__ import CONFIG_FILE
-from mkv_episode_matcher.config import get_config
-from datetime import datetime, timedelta
 from concurrent.futures import ThreadPoolExecutor
-from pgsreader import PGSReader
+from datetime import datetime, timedelta
+
+import pytesseract
 from imagemaker import make_image
 from loguru import logger
+from pgsreader import PGSReader
+from PIL import Image, ImageOps
+
+from mkv_episode_matcher.__main__ import CONFIG_FILE
+from mkv_episode_matcher.config import get_config
 
 
 def convert_mkv_to_sup(mkv_file, output_dir):
@@ -80,7 +81,7 @@ def perform_ocr(sup_file_path):
     si = 0
 
     tesseract_lang = "eng"
-    tesseract_config = "-c tessedit_char_blacklist=[] --psm 6 --oem {}".format(1)
+    tesseract_config = f"-c tessedit_char_blacklist=[] --psm 6 --oem {1}"
 
     config = get_config(CONFIG_FILE)
     tesseract_path = config.get("tesseract_path")

mkv_episode_matcher/tmdb_client.py

@@ -1,10 +1,12 @@
 # tmdb_client.py
+import time
+from threading import Lock
+
 import requests
 from loguru import logger
-from mkv_episode_matcher.config import get_config
+
 from mkv_episode_matcher.__main__ import CONFIG_FILE
-from threading import Lock
-import time
+from mkv_episode_matcher.config import get_config
 
 BASE_IMAGE_URL = "https://image.tmdb.org/t/p/original"
 

mkv_episode_matcher/utils.py

@@ -1,14 +1,16 @@
 # utils.py
 import os
+import re
+import shutil
 from typing import Set
+
+import requests
 from loguru import logger
-import re
-from mkv_episode_matcher.__main__ import CONFIG_FILE, CACHE_DIR
+from opensubtitlescom import OpenSubtitles
+
+from mkv_episode_matcher.__main__ import CACHE_DIR, CONFIG_FILE
 from mkv_episode_matcher.config import get_config
 from mkv_episode_matcher.tmdb_client import fetch_season_details
-import requests
-from opensubtitlescom import OpenSubtitles
-import shutil
 
 
 def check_filename(filename, series_title, season_number, episode_number):
@@ -199,7 +201,7 @@ def get_subtitles(show_id, seasons: Set[int]):
                     else:
                         continue
             else:
-                print(
+                logger.info(
                     f"Subtitle already exists for {series_name} - S{season:02d}E{episode:02d}"
                 )
                 continue

{mkv_episode_matcher-0.1.0.dist-info → mkv_episode_matcher-0.1.1.dist-info}/METADATA

@@ -1,8 +1,29 @@
 Metadata-Version: 2.3
 Name: mkv-episode-matcher
-Version: 0.1.0
+Version: 0.1.1
 Summary: The MKV Episode Matcher is a tool for identifying TV series episodes from MKV files and renaming the files accordingly.
-Author-email: Jonathan Sakkos <jonathansakkos@protonmail.com>
+Project-URL: Documentation, https://github.com/Jsakkos/mkv-episode-matcher#readme
+Project-URL: Issues, https://github.com/Jsakkos/mkv-episode-matcher/issues
+Project-URL: Source, https://github.com/Jsakkos/mkv-episode-matcher
+Author-email: Jsakkos <jonathansakkos@protonmail.com>
+License-Expression: MIT
+Classifier: Development Status :: 4 - Beta
+Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: Implementation :: CPython
+Classifier: Programming Language :: Python :: Implementation :: PyPy
+Requires-Python: >=3.8
+Requires-Dist: configparser
+Requires-Dist: loguru
+Requires-Dist: opensubtitlescom
+Requires-Dist: pillow
+Requires-Dist: pytesseract
+Requires-Dist: requests
+Requires-Dist: tmdb-client
 Description-Content-Type: text/markdown
 
 # MKV Episode Matcher
@@ -32,19 +53,8 @@ python __main__.py --api-key `your-api-key` --show-dir /path/to/show
 
 ## How it works
 
-MKV Episode Matcher compares reference images from TMDb with frames from the mkv content using image hashing. 
+MKV Episode Matcher extracts the subtitle text from each MKV file, then crossreferences the text against .srt subtitle files that are either user-provided or downloaded from Opensubtitles.com
 
-## Caveats (WIP)
-
-Currently, MKV Episode Matcher is slow (several minutes per episode), CPU intensive, and error-prone.
-
-# Known issues
-
-When reading BluRay files, the following warning pops up in the terminal:
-```
-Could not find codec parameters for stream 3 (Subtitle: hdmv_pgs_subtitle (pgssub)): unspecified size
-Consider increasing the value for the 'analyzeduration' (0) and 'probesize' (5000000) options
-```
 
 # Contributing
 

mkv_episode_matcher-0.1.1.dist-info/RECORD

@@ -0,0 +1,13 @@
+mkv_episode_matcher/.gitattributes,sha256=Gh2-F2vCM7SZ01pX23UT8pQcmauXWfF3gwyRSb6ZAFs,66
+mkv_episode_matcher/__init__.py,sha256=LM2LaE6ld1IdTocZPzvWFmhHRfeoMYFg3PuXn0cJvmU,19
+mkv_episode_matcher/__main__.py,sha256=UW2-p_vhGCSlimKdVlpf-5mRR5_VIbIuZXB_Q9RXNGA,6672
+mkv_episode_matcher/config.py,sha256=zDDKBcsDt5fME9BRqiTi7yWKeast1pZh36BNYMvIBYM,2419
+mkv_episode_matcher/episode_matcher.py,sha256=UBT53tiC0xryl9k9XXoRUY1-08u5DH-NcwnmD-MSWG8,8914
+mkv_episode_matcher/mkv_to_srt.py,sha256=6boR-jJ5fOCLZxHapTSFj7b01daQZNMWI_jBD__pQZE,6462
+mkv_episode_matcher/requirements.txt,sha256=0JLuUm69lLp8anUgtW48CuULZ_lSwd-1XL3eoShVWjI,93
+mkv_episode_matcher/tmdb_client.py,sha256=LbMCgjmp7sCbrQo_CDlpcnryKPz5S7inE24YY9Pyjk4,4172
+mkv_episode_matcher/utils.py,sha256=UMygaE4ZUvV9BRyXxmgfd1g06-0pk1eGlM56NzfgjYM,9305
+mkv_episode_matcher-0.1.1.dist-info/METADATA,sha256=kvxVKoydueoVzbMRg3cv6wWgM8tCa0pnWwTMyvq6GFg,4351
+mkv_episode_matcher-0.1.1.dist-info/WHEEL,sha256=zEMcRr9Kr03x1ozGwg5v9NQBKn3kndp6LSoSlVg-jhU,87
+mkv_episode_matcher-0.1.1.dist-info/entry_points.txt,sha256=IglJ43SuCZq2eQ3shMFILCkmQASJHnDCI3ogohW2Hn4,64
+mkv_episode_matcher-0.1.1.dist-info/RECORD,,

{mkv_episode_matcher-0.1.0.dist-info → mkv_episode_matcher-0.1.1.dist-info}/WHEEL

@@ -1,5 +1,4 @@
 Wheel-Version: 1.0
 Generator: hatchling 1.24.2
 Root-Is-Purelib: true
-Tag: py2-none-any
 Tag: py3-none-any

mkv_episode_matcher-0.1.1.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+mkv-match = mkv_episode_matcher.__main__:main

mkv_episode_matcher-0.1.0.dist-info/RECORD

@@ -1,12 +0,0 @@
-mkv_episode_matcher/.gitattributes,sha256=Gh2-F2vCM7SZ01pX23UT8pQcmauXWfF3gwyRSb6ZAFs,66
-mkv_episode_matcher/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mkv_episode_matcher/__main__.py,sha256=kCsrekSaNOkrfaY8Lm-VzVzALsxcmuaEWeeyCE5deEQ,6678
-mkv_episode_matcher/config.py,sha256=2Ui0f9LUc0r6pmdRUmcopdykotHoFxjqJavLKLOzy5w,2354
-mkv_episode_matcher/episode_matcher.py,sha256=IAUDOHyMzmoqBDilA5GIXiQdomfEWR7mn4xXU5XtvsM,8904
-mkv_episode_matcher/mkv_to_srt.py,sha256=BSDgNCgrkpr451X-P0A3-Q4bENfItv2A43yp5dtB430,6468
-mkv_episode_matcher/requirements.txt,sha256=0JLuUm69lLp8anUgtW48CuULZ_lSwd-1XL3eoShVWjI,93
-mkv_episode_matcher/tmdb_client.py,sha256=3sWC0tHvsW2XAYA4ndXh3PjUFCobQRpXzykNP-Z4rAA,4170
-mkv_episode_matcher/utils.py,sha256=ZkqGV3ZNPwpTvN1dHNZb-iLwJnk4ldk6w-Znh3TPH70,9297
-mkv_episode_matcher-0.1.0.dist-info/METADATA,sha256=Z6_kFPF6S49njL8CzfhrvxMuuXH9qNueD3FyZrTHx5c,3759
-mkv_episode_matcher-0.1.0.dist-info/WHEEL,sha256=cDcbFFSNXOE-241I5PFuLkIYfR_FM7WTlPEi33njInY,105
-mkv_episode_matcher-0.1.0.dist-info/RECORD,,