mirror of
https://github.com/lrnselfreliance/wrolpi.git
synced 2026-10-26 19:01:39 -04:00
55 lines
1.6 KiB
Python
Executable file
55 lines
1.6 KiB
Python
Executable file
#! /usr/bin/env python3
|
|
from pathlib import Path
|
|
from typing import Generator, Union
|
|
|
|
import srt
|
|
import webvtt
|
|
|
|
from wrolpi.common import logger
|
|
|
|
__all__ = ['get_captions']
|
|
|
|
|
|
def get_caption_text(caption_path: Union[str, Path]) -> Generator:
|
|
"""Return all text from each caption of a caption file."""
|
|
if str(caption_path).endswith('vtt'):
|
|
for caption in webvtt.read(caption_path):
|
|
text = str(caption.text).strip()
|
|
yield text
|
|
else:
|
|
with open(caption_path, 'rt') as fh:
|
|
contents = fh.read()
|
|
for subtitle in srt.parse(contents):
|
|
yield subtitle.content
|
|
|
|
|
|
def get_unique_caption_lines(caption_path: Union[str, Path]) -> Generator:
|
|
"""
|
|
Return all unique lines from each caption of a caption file.
|
|
"""
|
|
last_line = None
|
|
for text in get_caption_text(caption_path):
|
|
for line in text.split('\n'):
|
|
if line and line != last_line:
|
|
last_line = line
|
|
yield line
|
|
|
|
|
|
def get_captions(caption_path: Path):
|
|
"""Parse video captions from the video's captions file."""
|
|
try:
|
|
lines = get_unique_caption_lines(str(caption_path))
|
|
block = '\n'.join(lines)
|
|
return block
|
|
except UnicodeDecodeError:
|
|
# Failed to decode the caption file
|
|
# TODO handle this error
|
|
pass
|
|
except webvtt.errors.MalformedFileError:
|
|
# File format is broken
|
|
pass
|
|
except webvtt.errors.MalformedCaptionError:
|
|
# Captions form is broken somehow
|
|
pass
|
|
|
|
logger.debug(f'Failed to parse caption file {caption_path}')
|