bazarr/libs/subzero/modification/mods/offset.py

# coding=utf-8

import logging

from subzero.modification.mods import SubtitleModification
from subzero.modification import registry

logger = logging.getLogger(__name__)


class ShiftOffset(SubtitleModification):
    identifier = "shift_offset"
    description = "Change the timing of the subtitle"
    exclusive = False
    advanced = True
    args_mergeable = True
    modifies_whole_file = True

    long_description = "Adds or substracts a certain amount of time from the whole subtitle to match your media"

    @classmethod
    def merge_args(cls, args1, args2):
        new_args = dict((key, int(value)) for key, value in args1.iteritems())

        for key, value in args2.iteritems():
            if not int(value):
                continue

            if key in new_args:
                new_args[key] += int(value)
            else:
                new_args[key] = int(value)

        return dict(filter(lambda kv: bool(kv[1]), new_args.items()))

    def modify(self, content, debug=False, parent=None, **kwargs):
        parent.f.shift(h=int(kwargs.get("h", 0)), m=int(kwargs.get("m", 0)), s=int(kwargs.get("s", 0)),
                       ms=int(kwargs.get("ms", 0)))


registry.register(ShiftOffset)