bazarr/libs/knowit/rules/subtitle.py

import re

from knowit.core import Rule


class ClosedCaptionRule(Rule):
    """Closed caption rule."""

    cc_re = re.compile(r'(\bcc\d\b)', re.IGNORECASE)

    def execute(self, props, pv_props, context):
        """Execute closed caption rule."""
        for name in (pv_props.get('_closed_caption'), props.get('name')):
            if name and self.cc_re.search(name):
                return True


class HearingImpairedRule(Rule):
    """Hearing Impaired rule."""

    hi_re = re.compile(r'(\bsdh\b)', re.IGNORECASE)

    def execute(self, props, pv_props, context):
        """Hearing Impaired."""
        name = props.get('name')
        if name and self.hi_re.search(name):
            return True