subtitld 0.0.0.dev0__py3-none-any.whl

subtitld/modules/history.py

@@ -0,0 +1,31 @@
+"""History module (Ctrl+Z)
+
+"""
+
+import copy
+
+from subtitld.modules import session
+
+ALL_HISTORY = []
+REDO_HISTORY = []
+
+
+def history_append(subtitles):
+    """Append subtitles list to history"""
+    ALL_HISTORY.append(copy.deepcopy(subtitles))
+    REDO_HISTORY.clear()
+
+
+def history_undo():
+    """Revert to last subtitle on the history list"""
+    if ALL_HISTORY:
+        REDO_HISTORY.append(copy.deepcopy(session.SUBTITLE['segments']))
+        session.SUBTITLE['segments'].clear()
+        session.SUBTITLE['segments'].extend(copy.deepcopy(ALL_HISTORY.pop()))
+
+
+def history_redo():
+    """Redo subtitle on the history list"""
+    if REDO_HISTORY:
+        session.SUBTITLE['segments'].clear()
+        session.SUBTITLE['segments'].extend(copy.deepcopy(REDO_HISTORY.pop()))

subtitld/modules/quality_check.py

@@ -0,0 +1,53 @@
+"""Module to check subtitles quality
+
+"""
+
+from subtitld.modules import session
+
+
+def check_subtitle(subtitle=[]):
+    """Function to check individual subtitle"""
+    approved = True
+    reasons = []
+    issues = []
+
+    settings =  session.CONFIG.get('quality_check', {})
+
+    if subtitle and (subtitle['end'] - subtitle['start']):
+        if len(subtitle['text'].replace('\n', '')) / (subtitle['end'] - subtitle['start']) > settings.get('reading_speed_cps', 21):
+            approved = False
+            reasons.append('Reading speed: {ars} characters per second. It should be {rs} per second.'.format(rs=settings.get('reading_speed_cps', 21), ars=round(len(subtitle['text'].replace('\n', '')) / (subtitle['end'] - subtitle['start']), 2)))
+            issues.append('cps')
+        if len(subtitle['text'].replace('\n', '').split(' ')) / ((subtitle['end'] - subtitle['start']) / 60) > settings.get('reading_speed_wpm', 140):
+            approved = False
+            reasons.append('Reading speed: {ars} words per minute. It should be {rs} per minute.'.format(rs=settings.get('reading_speed_wpm', 140), ars=round(len(subtitle['text'].replace('\n', '').split(' ')) / ((subtitle['end'] - subtitle['start']) / 60))))
+            issues.append('wpm')
+        if (subtitle['end'] - subtitle['start']) < settings.get('minimum_duration', .7):
+            approved = False
+            reasons.append('The duration of this subtitle, {ad}, is less than the minimum of {d}.'.format(d=settings.get('minimum_duration', .7), ad=round((subtitle['end'] - subtitle['start']), 2)))
+            issues.append('duration')
+        if (subtitle['end'] - subtitle['start']) > settings.get('maximum_duration', 7):
+            approved = False
+            reasons.append('The duration of this subtitle, {ad}, is more than the maximum of {d}.'.format(d=settings.get('maximum_duration', 7), ad=round((subtitle['end'] - subtitle['start']), 2)))
+            issues.append('duration')
+        if len(subtitle['text'].split('\n')) > settings.get('maximum_lines', 2):
+            approved = False
+            reasons.append('There are {al} lines, more than the maximum of {l}.'.format(l=settings.get('maximum_lines', 2), al=round(len(subtitle['text'].split('\n')), 2)))
+            issues.append('number_of_lines')
+        for line in subtitle['text'].split('\n'):
+            if len(line) > settings.get('maximum_characters_per_line', 42):
+                approved = False
+                reasons.append('There are more than {c} characters per line ({ac}).'.format(c=settings.get('maximum_characters_per_line', 42), ac=len(line)))
+                issues.append('cpl')
+                break
+        if len(subtitle['text'].split('\n')) > 1:
+            if settings.get('prefer_compact', False):
+                if len(subtitle['text'].replace('\n', '')) < settings.get('maximum_characters_per_line', 42):
+                    approved = False
+                    reasons.append('The lines should be joined as there are less than {c} characters in total.'.format(c=settings.get('maximum_characters_per_line', 42)))
+            if settings.get('balance_ratio_enabled', False):
+                if len(min(subtitle['text'].split('\n'), key=len)) < len(max(subtitle['text'].split('\n'), key=len))*(settings.get('balance_ratio', 50)/100):
+                    approved = False
+                    reasons.append('The shortest line length is less than {p}% the largest.'.format(p=settings.get('balance_ratio', 50)))
+
+    return approved, reasons, issues

subtitld/modules/session.py

@@ -0,0 +1,270 @@
+"""All path definitions for Subtitld"""
+
+import pathlib
+import sys
+import tempfile
+import subprocess
+import subtitld
+import platformdirs
+
+PATH_SUBTITLD = pathlib.Path(subtitld.__file__).parent
+PATH_HOME = pathlib.Path.home()
+PATH_LOCALE = PATH_SUBTITLD / 'locale'
+PATH_SUBTITLD_GRAPHICS = PATH_SUBTITLD / 'graphics'
+PATH_SUBTITLD_USER_CONFIG = pathlib.Path(platformdirs.user_config_dir('subtitld'))
+PATH_SUBTITLD_USER_CACHE = pathlib.Path(platformdirs.user_cache_dir('subtitld'))
+FFMPEG_EXECUTABLE = 'ffmpeg'
+FFPROBE_EXECUTABLE = 'ffprobe'
+STARTUPINFO = None
+
+ACTUAL_OS = 'linux'
+
+tempdir = tempfile.TemporaryDirectory()
+PATH_TEMP = tempdir.name
+
+if sys.platform == 'darwin':
+    ACTUAL_OS = 'macos'
+    if getattr(sys, "frozen", False):
+        FFMPEG_EXECUTABLE = str(pathlib.Path(PATH_SUBTITLD).parent / 'ffmpeg')
+        FFPROBE_EXECUTABLE = str(pathlib.Path(PATH_SUBTITLD).parent / 'ffprobe')
+elif sys.platform == 'win32':
+    ACTUAL_OS = 'windows'
+
+    if getattr(sys, "frozen", False):
+        FFMPEG_EXECUTABLE = pathlib.Path(PATH_SUBTITLD).parent / 'ffmpeg.exe'
+        FFPROBE_EXECUTABLE = pathlib.Path(PATH_SUBTITLD).parent / 'ffprobe.exe'
+    else:
+        script_dir = pathlib.Path(sys.argv[0]).parent
+        FFMPEG_EXECUTABLE = script_dir / 'ffmpeg.exe'
+        FFPROBE_EXECUTABLE = script_dir / 'ffprobe.exe'
+    STARTUPINFO = subprocess.STARTUPINFO()
+    STARTUPINFO.dwFlags |= subprocess.STARTF_USESHOWWINDOW
+    STARTUPINFO.wShowWindow = subprocess.SW_HIDE
+    import multiprocessing
+    multiprocessing.freeze_support()
+else:
+    # Linux (AppImage, Snap, Flatpak, pip install). When frozen by
+    # PyInstaller we ship ffmpeg/ffprobe next to the main executable;
+    # for regular installs we fall back to whatever the distro provides.
+    if getattr(sys, "frozen", False):
+        FFMPEG_EXECUTABLE = str(pathlib.Path(PATH_SUBTITLD).parent / 'ffmpeg')
+        FFPROBE_EXECUTABLE = str(pathlib.Path(PATH_SUBTITLD).parent / 'ffprobe')
+
+if not PATH_SUBTITLD_USER_CONFIG.exists():
+    PATH_SUBTITLD_USER_CONFIG.mkdir(parents=True)
+
+if not PATH_SUBTITLD_USER_CACHE.exists():
+    PATH_SUBTITLD_USER_CACHE.mkdir(parents=True)
+
+PATH_SUBTITLD_DATA_BACKUP = PATH_SUBTITLD_USER_CACHE / 'backup'
+
+if not PATH_SUBTITLD_DATA_BACKUP.exists():
+    PATH_SUBTITLD_DATA_BACKUP.mkdir(parents=True)
+
+PATH_SUBTITLD_DATA_MODELS = PATH_SUBTITLD_USER_CACHE / 'models'
+
+if not PATH_SUBTITLD_DATA_MODELS.exists():
+    PATH_SUBTITLD_DATA_MODELS.mkdir(parents=True)
+
+PATH_SUBTITLD_DATA_AUDIOSEPARATION = PATH_SUBTITLD_USER_CACHE / 'audioseparation'
+
+if not PATH_SUBTITLD_DATA_AUDIOSEPARATION.exists():
+    PATH_SUBTITLD_DATA_AUDIOSEPARATION.mkdir(parents=True)
+
+PATH_SUBTITLD_USER_CONFIG_FILE = PATH_SUBTITLD_USER_CONFIG / 'subtitld.config'
+
+# VERSION_NUMBER = '20.07.0.0'
+# if os.path.isfile(os.path.join(PATH_SUBTITLD, 'current_version')):
+#     VERSION_NUMBER = open(os.path.join(PATH_SUBTITLD, 'current_version')).read().strip()
+
+LIST_OF_SUPPORTED_VIDEO_EXTENSIONS = {
+    'MP4': {'description': 'MPEG-4 Video format', 'extensions': ['mp4']},
+    'MKV': {'description': 'Matroska Video format', 'extensions': ['mkv']},
+    'MOV': {'description': 'Quicktime Video format', 'extensions': ['mov']},
+    'MPG': {'description': 'MPEG Video format', 'extensions': ['mpg']},
+    'WEBM': {'description': 'WebM Video format', 'extensions': ['webm']},
+    'OGV': {'description': 'Ogg Video format', 'extensions': ['ogv']},
+    'M4V': {'description': 'MPEG-4 Video format', 'extensions': ['m4v']},
+}
+
+LIST_OF_SUPPORTED_SUBTITLE_EXTENSIONS = {
+    'SRT': {'description': 'SubRip Subtitle format', 'extensions': ['srt']},
+    'DFXP': {'description': 'DFXP ITT Subtitle format', 'extensions': ['dfxp', 'itt']},
+    'TTML': {'description': 'TTML Subtitle format', 'extensions': ['ttml']},
+    'SAMI': {'description': 'SAMI Subtitle format', 'extensions': ['smi', 'sami']},
+    'SCC': {'description': 'SCC Subtitle format', 'extensions': ['scc']},
+    'VTT': {'description': 'VTT Subtitle format', 'extensions': ['webvtt', 'vtt']},
+    'ASS': {'description': 'SubStation Alpha Subtitle format', 'extensions': ['ass', 'ssa']},
+    'SBV': {'description': 'SBV Subtitle format', 'extensions': ['sbv']},
+    'SUB': {'description': 'MicroDVD Subtitle format', 'extensions': ['sub']},
+    'XML': {'description': 'XML Subtitle format', 'extensions': ['xml']},
+    'USF': {'description': 'Universal Subtitle Format', 'extensions': ['usf']},
+    'USFX': {'description': 'Universal Subtitle Format bundled with assets', 'extensions': ['usfx']},
+    'JSON': {'description': 'JSON format', 'extensions': ['json']},
+}
+
+LIST_OF_SUPPORTED_IMPORT_EXTENSIONS = {
+    'TXT': {'description': 'Simple TXT file', 'extensions': ['txt']},
+    'DOCX': {'description': 'Microsoft Word document (docx)', 'extensions': ['docx']},
+    'SRT': {'description': 'SubRip Subtitle format', 'extensions': ['srt']}
+}
+
+LIST_OF_SUPPORTED_EXPORT_EXTENSIONS = {
+    'TXT': {'description': 'Simple TXT file', 'extensions': ['txt']},
+    'KDENLIVE': {'description': 'Kdenlive format', 'extensions': ['kdenlive']}
+}
+
+LIST_OF_SUPPORTED_AUDIO_EXPORT_EXTENSIONS = {
+    'WAV': {'description': 'WAV audio', 'extensions': ['wav']},
+    'FLAC': {'description': 'FLAC lossless audio', 'extensions': ['flac']},
+    'MP3': {'description': 'MP3 audio', 'extensions': ['mp3']},
+    'MP4': {'description': 'MP4 video with dubbed audio', 'extensions': ['mp4']},
+}
+
+LANGUAGE_DICT_LIST = {
+    'Afrikaans (South Africa)': 'af-za',
+    'Amharic (Ethiopia)': 'am-et',
+    'Arabic (United Arab Emirates)': 'ar-ae',
+    'Arabic (Bahrain)': 'ar-bh',
+    'Arabic (Algeria)': 'ar-dz',
+    'Arabic (Egypt)': 'ar-eg',
+    'Arabic (Israel)': 'ar-il',
+    'Arabic (Iraq)': 'ar-iq',
+    'Arabic (Jordan)': 'ar-jo',
+    'Arabic (Kuwait)': 'ar-kw',
+    'Arabic (Lebanon)': 'ar-lb',
+    'Arabic (Morocco)': 'ar-ma',
+    'Arabic (Oman)': 'ar-om',
+    'Arabic (State of Palestine)': 'ar-ps',
+    'Arabic (Qatar)': 'ar-qa',
+    'Arabic (Saudi Arabia)': 'ar-sa',
+    'Arabic (Tunisia)': 'ar-tn',
+    'Azerbaijani (Azerbaijan)': 'az-az',
+    'Bulgarian (Bulgaria)': 'bg-bg',
+    'Bengali (Bangladesh)': 'bn-bd',
+    'Bengali (India)': 'bn-in',
+    'Catalan (Spain)': 'ca-es',
+    'Chinese, Mandarin (Simplified, China)': 'cmn-hans-cn',
+    'Chinese, Mandarin (Simplified, Hong Kong)': 'cmn-hans-hk',
+    'Chinese, Mandarin (Traditional, Taiwan)': 'cmn-hant-tw',
+    'Czech (Czech Republic)': 'cs-cz',
+    'Danish (Denmark)': 'da-dk',
+    'German (Germany)': 'de-de',
+    'Greek (Greece)': 'el-gr',
+    'English (Australia)': 'en-au',
+    'English (Canada)': 'en-ca',
+    'English (United Kingdom)': 'en-gb',
+    'English (Ghana)': 'en-gh',
+    'English (Ireland)': 'en-ie',
+    'English (India)': 'en-in',
+    'English (Kenya)': 'en-ke',
+    'English (Nigeria)': 'en-ng',
+    'English (New Zealand)': 'en-nz',
+    'English (Philippines)': 'en-ph',
+    'English (Singapore)': 'en-sg',
+    'English (Tanzania)': 'en-tz',
+    'English (United States)': 'en-us',
+    'English (South Africa)': 'en-za',
+    'Spanish (Argentina)': 'es-ar',
+    'Spanish (Bolivia)': 'es-bo',
+    'Spanish (Chile)': 'es-cl',
+    'Spanish (Colombia)': 'es-co',
+    'Spanish (Costa Rica)': 'es-cr',
+    'Spanish (Dominican Republic)': 'es-do',
+    'Spanish (Ecuador)': 'es-ec',
+    'Spanish (Spain)': 'es-es',
+    'Spanish (Guatemala)': 'es-gt',
+    'Spanish (Honduras)': 'es-hn',
+    'Spanish (Mexico)': 'es-mx',
+    'Spanish (Nicaragua)': 'es-ni',
+    'Spanish (Panama)': 'es-pa',
+    'Spanish (Peru)': 'es-pe',
+    'Spanish (Puerto Rico)': 'es-pr',
+    'Spanish (Paraguay)': 'es-py',
+    'Spanish (El Salvador)': 'es-sv',
+    'Spanish (United States)': 'es-us',
+    'Spanish (Uruguay)': 'es-uy',
+    'Spanish (Venezuela)': 'es-ve',
+    'Basque (Spain)': 'eu-es',
+    'Persian (Iran)': 'fa-ir',
+    'Finnish (Finland)': 'fi-fi',
+    'Filipino (Philippines)': 'fil-ph',
+    'French (Canada)': 'fr-ca',
+    'French (France)': 'fr-fr',
+    'Galician (Spain)': 'gl-es',
+    'Gujarati (India)': 'gu-in',
+    'Hebrew (Israel)': 'he-il',
+    'Hindi (India)': 'hi-in',
+    'Croatian (Croatia)': 'hr-hr',
+    'Hungarian (Hungary)': 'hu-hu',
+    'Armenian (Armenia)': 'hy-am',
+    'Indonesian (Indonesia)': 'id-id',
+    'Icelandic (Iceland)': 'is-is',
+    'Italian (Italy)': 'it-it',
+    'Japanese (Japan)': 'ja-jp',
+    'Javanese (Indonesia)': 'jv-id',
+    'Georgian (Georgia)': 'ka-ge',
+    'Khmer (Cambodia)': 'km-kh',
+    'Kannada (India)': 'kn-in',
+    'Korean (South Korea)': 'ko-kr',
+    'Lao (Laos)': 'lo-la',
+    'Lithuanian (Lithuania)': 'lt-lt',
+    'Latvian (Latvia)': 'lv-lv',
+    'Malayalam (India)': 'ml-in',
+    'Marathi (India)': 'mr-in',
+    'Malay (Malaysia)': 'ms-my',
+    'Norwegian Bokmal (Norway)': 'nb-no',
+    'Nepali (Nepal)': 'ne-np',
+    'Dutch (Netherlands)': 'nl-nl',
+    'Polish (Poland)': 'pl-pl',
+    'Portuguese (Brazil)': 'pt-br',
+    'Portuguese (Portugal)': 'pt-pt',
+    'Romanian (Romania)': 'ro-ro',
+    'Russian (Russia)': 'ru-ru',
+    'Sinhala (Sri Lanka)': 'si-lk',
+    'Slovak (Slovakia)': 'sk-sk',
+    'Slovenian (Slovenia)': 'sl-si',
+    'Serbian (Serbia)': 'sr-rs',
+    'Sundanese (Indonesia)': 'su-id',
+    'Swedish (Sweden)': 'sv-se',
+    'Swahili (Kenya)': 'sw-ke',
+    'Swahili (Tanzania)': 'sw-tz',
+    'Tamil (India)': 'ta-in',
+    'Tamil (Sri Lanka)': 'ta-lk',
+    'Tamil (Malaysia)': 'ta-my',
+    'Tamil (Singapore)': 'ta-sg',
+    'Telugu (India)': 'te-in',
+    'Thai (Thailand)': 'th-th',
+    'Turkish (Turkey)': 'tr-tr',
+    'Ukrainian (Ukraine)': 'uk-ua',
+    'Urdu (India)': 'ur-in',
+    'Urdu (Pakistan)': 'ur-pk',
+    'Vietnamese (Vietnam)': 'vi-vn',
+    'Chinese, Cantonese (Traditional, Hong Kong)': 'yue-hant-hk',
+    'Zulu (South Africa)': 'zu-za'
+}
+
+CONFIG = {}
+
+SUBTITLE = {
+    'segments': [],
+}
+
+SPEAKERS = {}
+
+VIDEO = {}
+
+FORMAT = {}
+
+REPEAT_DURATION_BUFFER = []
+
+UNSAVED = False
+_unsaved_change_callbacks = []
+def set_unsaved(value=True):
+    global UNSAVED
+    old_value = UNSAVED
+    UNSAVED = value
+    if old_value != value:
+        for callback in _unsaved_change_callbacks:
+            callback()
+

subtitld/modules/shortcuts.py

@@ -0,0 +1,84 @@
+"""Shortcuts module - Robust customizable shortcut system with decorators
+
+"""
+
+from PySide6.QtGui import QAction
+
+
+class ShortcutRegistry:
+    """Registry for managing shortcuts and their associated actions"""
+    
+    def __init__(self):
+        self._shortcuts = {}
+    
+    def register(self, command_id, description, default_keys, handler):
+        """Register a shortcut with its handler"""
+        self._shortcuts[command_id] = {
+            'description': description,
+            'default_keys': default_keys,
+            'handler': handler
+        }
+    
+    def shortcut(self, command_id, description, keys):
+        """Decorator to register a function as a shortcut handler
+        
+        Usage:
+            @shortcut('my_command', 'My Command Description', ['Ctrl+K'])
+            def my_handler(self):
+                # handler code
+        """
+        def decorator(func):
+            self.register(command_id, description, keys, func)
+            return func
+        return decorator
+    
+    def get_description(self, command_id):
+        return self._shortcuts.get(command_id, {}).get('description', '')
+    
+    def get_default_keys(self, command_id):
+        return self._shortcuts.get(command_id, {}).get('default_keys', [])
+    
+    def get_handler(self, command_id):
+        return self._shortcuts.get(command_id, {}).get('handler')
+    
+    def get_all_commands(self):
+        return list(self._shortcuts.keys())
+
+
+# Global registry
+_registry = ShortcutRegistry()
+shortcut = _registry.shortcut
+
+# Legacy compatibility
+shortcuts_dict = {}
+default_shortcuts_dict = {}
+
+
+def load(self, shortcut_commands):
+    """Load shortcuts commands on widgets"""
+    # Update legacy dicts
+    shortcuts_dict.update({cmd: _registry.get_description(cmd) for cmd in _registry.get_all_commands()})
+    default_shortcuts_dict.update({cmd: _registry.get_default_keys(cmd) for cmd in _registry.get_all_commands()})
+    
+    for command_id in _registry.get_all_commands():
+        if command_id not in shortcut_commands:
+            shortcut_commands[command_id] = _registry.get_default_keys(command_id)
+    
+    for command_id, keys in shortcut_commands.items():
+        handler = _registry.get_handler(command_id)
+        if handler:
+            action = QAction(_registry.get_description(command_id), self)
+            action.setShortcuts(keys)
+            action.triggered.connect(lambda checked=False, h=handler: h(self))
+            self.addAction(action)
+
+
+def disable_actions(self):
+    for action in self.actions():
+        action.setEnabled(False)
+
+
+def enable_actions(self):
+    for action in self.actions():
+        action.setEnabled(True)
+