mirror of https://github.com/morpheus65535/bazarr
64 lines
2.1 KiB
Python
64 lines
2.1 KiB
Python
import dataclasses
|
|
import json
|
|
from typing import Any, Optional, TextIO
|
|
|
|
from ..common import Color
|
|
from ..ssaevent import SSAEvent
|
|
from ..ssastyle import SSAStyle
|
|
from .base import FormatBase
|
|
from ..ssafile import SSAFile
|
|
|
|
|
|
# We're using Color dataclass
|
|
# https://stackoverflow.com/questions/51286748/make-the-python-json-encoder-support-pythons-new-dataclasses
|
|
class EnhancedJSONEncoder(json.JSONEncoder):
|
|
def default(self, o: Any) -> Any:
|
|
if dataclasses.is_dataclass(o):
|
|
return dataclasses.asdict(o)
|
|
return super().default(o)
|
|
|
|
|
|
class JSONFormat(FormatBase):
|
|
"""
|
|
Implementation of JSON subtitle pseudo-format (serialized pysubs2 internal representation)
|
|
|
|
This is essentially SubStation Alpha as JSON.
|
|
"""
|
|
@classmethod
|
|
def guess_format(cls, text: str) -> Optional[str]:
|
|
"""See :meth:`pysubs2.formats.FormatBase.guess_format()`"""
|
|
if text.startswith("{\"") and "\"info\":" in text:
|
|
return "json"
|
|
else:
|
|
return None
|
|
|
|
@classmethod
|
|
def from_file(cls, subs: "SSAFile", fp: TextIO, format_: str, **kwargs: Any) -> None:
|
|
"""See :meth:`pysubs2.formats.FormatBase.from_file()`"""
|
|
data = json.load(fp)
|
|
|
|
subs.info.clear()
|
|
subs.info.update(data["info"])
|
|
|
|
subs.styles.clear()
|
|
for name, fields in data["styles"].items():
|
|
subs.styles[name] = sty = SSAStyle()
|
|
for k, v in fields.items():
|
|
if "color" in k:
|
|
setattr(sty, k, Color(**v))
|
|
else:
|
|
setattr(sty, k, v)
|
|
|
|
subs.events = [SSAEvent(**fields) for fields in data["events"]]
|
|
|
|
@classmethod
|
|
def to_file(cls, subs: "SSAFile", fp: TextIO, format_: str, **kwargs: Any) -> None:
|
|
"""See :meth:`pysubs2.formats.FormatBase.to_file()`"""
|
|
data = {
|
|
"info": dict(**subs.info),
|
|
"styles": {name: sty.as_dict() for name, sty in subs.styles.items()},
|
|
"events": [ev.as_dict() for ev in subs.events]
|
|
}
|
|
|
|
json.dump(data, fp, cls=EnhancedJSONEncoder)
|