mirror of
https://github.com/clinton-hall/nzbToMedia.git
synced 2024-11-14 17:40:24 -08:00
48 lines
1.1 KiB
Python
48 lines
1.1 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
"""
|
|
Path markers
|
|
"""
|
|
from rebulk import Rebulk
|
|
|
|
from rebulk.utils import find_all
|
|
|
|
|
|
def path(config): # pylint:disable=unused-argument
|
|
"""
|
|
Builder for rebulk object.
|
|
|
|
:param config: rule configuration
|
|
:type config: dict
|
|
:return: Created Rebulk object
|
|
:rtype: Rebulk
|
|
"""
|
|
rebulk = Rebulk()
|
|
rebulk.defaults(name="path", marker=True)
|
|
|
|
def mark_path(input_string, context):
|
|
"""
|
|
Functional pattern to mark path elements.
|
|
|
|
:param input_string:
|
|
:param context:
|
|
:return:
|
|
"""
|
|
ret = []
|
|
if context.get('name_only', False):
|
|
ret.append((0, len(input_string)))
|
|
else:
|
|
indices = list(find_all(input_string, '/'))
|
|
indices += list(find_all(input_string, '\\'))
|
|
indices += [-1, len(input_string)]
|
|
|
|
indices.sort()
|
|
|
|
for i in range(0, len(indices) - 1):
|
|
ret.append((indices[i] + 1, indices[i + 1]))
|
|
|
|
return ret
|
|
|
|
rebulk.functional(mark_path)
|
|
return rebulk
|