60 lines
1.5 KiB
Python
60 lines
1.5 KiB
Python
import logging
|
|
import pathlib
|
|
|
|
import anyio
|
|
import pysubs2
|
|
from pysubs2.formats.substation import SubstationFormat
|
|
from pysubs2.time import ms_to_times
|
|
|
|
from .Utils import calc_download_path
|
|
|
|
LOG = logging.getLogger("player.subtitle")
|
|
|
|
|
|
def ms_to_timestamp(ms: int) -> str:
|
|
ms = max(0, ms)
|
|
h, m, s, ms = ms_to_times(ms)
|
|
cs = ms // 10
|
|
return f"{h:01d}:{m:02d}:{s:02d}.{cs:02d}"
|
|
|
|
|
|
SubstationFormat.ms_to_timestamp = ms_to_timestamp
|
|
|
|
|
|
class Subtitle:
|
|
allowed_extensions: tuple[str] = (
|
|
".srt",
|
|
".vtt",
|
|
".ass",
|
|
)
|
|
|
|
async def make(self, path: str, file: str) -> str:
|
|
realFile: str = calc_download_path(base_path=path, folder=file, create_path=False)
|
|
|
|
rFile = pathlib.Path(realFile)
|
|
|
|
if not rFile.exists():
|
|
msg = f"File '{file}' does not exist."
|
|
raise Exception(msg)
|
|
|
|
if rFile.suffix not in self.allowed_extensions:
|
|
msg = f"File '{file}' subtitle type is not supported."
|
|
raise Exception(msg)
|
|
|
|
if rFile.suffix == ".vtt":
|
|
async with await anyio.open_file(realFile) as f:
|
|
return await f.read()
|
|
|
|
subs = pysubs2.load(path=str(rFile))
|
|
|
|
if len(subs.events) < 1:
|
|
msg = f"No subtitle events were found in '{rFile}'."
|
|
raise Exception(msg)
|
|
|
|
if len(subs.events) < 2:
|
|
return subs.to_string("vtt")
|
|
|
|
if subs.events[0].end == subs.events[len(subs.events) - 1].end:
|
|
subs.events.pop(0)
|
|
|
|
return subs.to_string("vtt")
|