mirror of https://github.com/morpheus65535/bazarr
62 lines
2.3 KiB
Python
62 lines
2.3 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
"""
|
|
container property
|
|
"""
|
|
from rebulk.remodule import re
|
|
|
|
from rebulk import Rebulk
|
|
|
|
from ..common import seps
|
|
from ..common.pattern import is_disabled
|
|
from ..common.validators import seps_surround
|
|
from ...reutils import build_or_pattern
|
|
|
|
|
|
def container(config):
|
|
"""
|
|
Builder for rebulk object.
|
|
|
|
:param config: rule configuration
|
|
:type config: dict
|
|
:return: Created Rebulk object
|
|
:rtype: Rebulk
|
|
"""
|
|
rebulk = Rebulk(disabled=lambda context: is_disabled(context, 'container'))
|
|
rebulk = rebulk.regex_defaults(flags=re.IGNORECASE).string_defaults(ignore_case=True)
|
|
rebulk.defaults(name='container',
|
|
formatter=lambda value: value.strip(seps),
|
|
tags=['extension'],
|
|
conflict_solver=lambda match, other: other
|
|
if other.name in ('source', 'video_codec') or
|
|
other.name == 'container' and 'extension' not in other.tags
|
|
else '__default__')
|
|
|
|
subtitles = config['subtitles']
|
|
info = config['info']
|
|
videos = config['videos']
|
|
torrent = config['torrent']
|
|
nzb = config['nzb']
|
|
|
|
rebulk.regex(r'\.'+build_or_pattern(subtitles)+'$', exts=subtitles, tags=['extension', 'subtitle'])
|
|
rebulk.regex(r'\.'+build_or_pattern(info)+'$', exts=info, tags=['extension', 'info'])
|
|
rebulk.regex(r'\.'+build_or_pattern(videos)+'$', exts=videos, tags=['extension', 'video'])
|
|
rebulk.regex(r'\.'+build_or_pattern(torrent)+'$', exts=torrent, tags=['extension', 'torrent'])
|
|
rebulk.regex(r'\.'+build_or_pattern(nzb)+'$', exts=nzb, tags=['extension', 'nzb'])
|
|
|
|
rebulk.defaults(clear=True,
|
|
name='container',
|
|
validator=seps_surround,
|
|
formatter=lambda s: s.lower(),
|
|
conflict_solver=lambda match, other: match
|
|
if other.name in ('source',
|
|
'video_codec') or other.name == 'container' and 'extension' in other.tags
|
|
else '__default__')
|
|
|
|
rebulk.string(*[sub for sub in subtitles if sub not in ('sub', 'ass')], tags=['subtitle'])
|
|
rebulk.string(*videos, tags=['video'])
|
|
rebulk.string(*torrent, tags=['torrent'])
|
|
rebulk.string(*nzb, tags=['nzb'])
|
|
|
|
return rebulk
|