#!/usr/bin/env python # -*- coding: utf-8 -*- """ film property """ from rebulk import Rebulk, AppendMatch, Rule from rebulk.remodule import re from ..common import dash from ..common.formatters import cleanup from ..common.pattern import is_disabled from ..common.validators import seps_surround from ...config import load_config_patterns def film(config): # pylint:disable=unused-argument """ Builder for rebulk object. :return: Created Rebulk object :rtype: Rebulk """ rebulk = Rebulk(disabled=lambda context: is_disabled(context, 'film')) rebulk.regex_defaults(flags=re.IGNORECASE, abbreviations=[dash]).string_defaults(ignore_case=True) rebulk.defaults(name='film', validator=seps_surround) load_config_patterns(rebulk, config.get('film')) rebulk.rules(FilmTitleRule) return rebulk class FilmTitleRule(Rule): """ Rule to find out film_title (hole after film property """ consequence = AppendMatch properties = {'film_title': [None]} def enabled(self, context): return not is_disabled(context, 'film_title') def when(self, matches, context): # pylint:disable=inconsistent-return-statements bonus_number = matches.named('film', lambda match: not match.private, index=0) if bonus_number: filepath = matches.markers.at_match(bonus_number, lambda marker: marker.name == 'path', 0) hole = matches.holes(filepath.start, bonus_number.start + 1, formatter=cleanup, index=0) if hole and hole.value: hole.name = 'film_title' return hole