28 lines
594 B
28 lines
594 B
5 years ago
|
#!/usr/bin/env python
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
import logging
|
||
|
|
import sys
|
||
|
|
|
||
|
|
from sklearn.pipeline import Pipeline
|
||
|
|
|
||
|
|
from .subtitle_parser import GenericSubtitleParser
|
||
|
|
from .subtitle_transformers import SubtitleShifter
|
||
|
|
|
||
|
|
logging.basicConfig(level=logging.INFO)
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
td = float(sys.argv[3])
|
||
|
|
pipe = Pipeline([
|
||
|
|
('parse', GenericSubtitleParser()),
|
||
|
|
('offset', SubtitleShifter(td)),
|
||
|
|
])
|
||
|
|
pipe.fit_transform(sys.argv[1])
|
||
|
|
pipe.steps[-1][1].write_file(sys.argv[2])
|
||
|
|
return 0
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
sys.exit(main())
|