mirror of
https://github.com/morpheus65535/bazarr
synced 2024-12-30 19:46:25 +00:00
61 lines
2.2 KiB
Python
61 lines
2.2 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(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
|