mirror of https://github.com/morpheus65535/bazarr
39 lines
1.3 KiB
Python
39 lines
1.3 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
# pylint: disable=no-self-use, pointless-statement, missing-docstring, invalid-name, len-as-condition
|
||
|
|
from rebulk.rules import Rule, RemoveMatch, CustomRule
|
||
|
|
|
||
|
|
|
||
|
|
class RemoveAllButLastYear(Rule):
|
||
|
|
consequence = RemoveMatch
|
||
|
|
def when(self, matches, context):
|
||
|
|
entries = matches.named('year')
|
||
|
|
return entries[:-1]
|
||
|
|
|
||
|
|
|
||
|
|
class PrefixedSuffixedYear(CustomRule):
|
||
|
|
def when(self, matches, context):
|
||
|
|
toRemove = []
|
||
|
|
years = matches.named('year')
|
||
|
|
for year in years:
|
||
|
|
if not matches.previous(year, lambda p: p.name == 'yearPrefix') and \
|
||
|
|
not matches.next(year, lambda n: n.name == 'yearSuffix'):
|
||
|
|
toRemove.append(year)
|
||
|
|
return toRemove
|
||
|
|
|
||
|
|
def then(self, matches, when_response, context):
|
||
|
|
for to_remove in when_response:
|
||
|
|
matches.remove(to_remove)
|
||
|
|
|
||
|
|
|
||
|
|
class PrefixedSuffixedYearNoLambda(Rule):
|
||
|
|
consequence = RemoveMatch
|
||
|
|
def when(self, matches, context):
|
||
|
|
toRemove = []
|
||
|
|
years = matches.named('year')
|
||
|
|
for year in years:
|
||
|
|
if not [m for m in matches.previous(year) if m.name == 'yearPrefix'] and \
|
||
|
|
not [m for m in matches.next(year) if m.name == 'yearSuffix']:
|
||
|
|
toRemove.append(year)
|
||
|
|
return toRemove
|