Source code for atlas_doc_parser.marks.parse_mark

# -*- coding: utf-8 -*-

"""
Auto-generated ADF Parser

This module provides functions to parse ADF JSON into Python objects.
"""

import typing as T

from ..type_hint import T_DATA
from ..type_enum import TypeEnum
from ..exc import UnimplementedTypeError

if T.TYPE_CHECKING:  # pragma: no cover
    from ..mark_or_node import T_MARK

# =============================================================================
# Mark imports
# =============================================================================
from .mark_background_color import MarkBackgroundColor
from .mark_code import MarkCode
from .mark_em import MarkEm
from .mark_link import MarkLink
from .mark_strike import MarkStrike
from .mark_strong import MarkStrong
from .mark_subsup import MarkSubsup
from .mark_text_color import MarkTextColor
from .mark_underline import MarkUnderline
from .mark_annotation import MarkAnnotation
from .mark_indentation import MarkIndentation
from .mark_border import MarkBorder
from .mark_alignment import MarkAlignment
from .mark_breakout import MarkBreakout
from .mark_data_consumer import MarkDataConsumer
from .mark_fragment import MarkFragment


# =============================================================================
# Mark registry
# =============================================================================
MARK_TYPE_TO_CLASS_MAPPING = {
    TypeEnum.backgroundColor.value: MarkBackgroundColor,
    TypeEnum.code.value: MarkCode,
    TypeEnum.em.value: MarkEm,
    TypeEnum.link.value: MarkLink,
    TypeEnum.strike.value: MarkStrike,
    TypeEnum.strong.value: MarkStrong,
    TypeEnum.subsup.value: MarkSubsup,
    TypeEnum.textColor.value: MarkTextColor,
    TypeEnum.underline.value: MarkUnderline,
    TypeEnum.annotation.value: MarkAnnotation,
    TypeEnum.indentation.value: MarkIndentation,
    TypeEnum.border.value: MarkBorder,
    TypeEnum.alignment.value: MarkAlignment,
    TypeEnum.breakout.value: MarkBreakout,
    TypeEnum.dataConsumer.value: MarkDataConsumer,
    TypeEnum.fragment.value: MarkFragment,
}



[docs]
def parse_mark(dct: T_DATA) -> "T_MARK":
    """
    Parse a mark dictionary into a Mark object.

    :param dct: The raw ADF mark dictionary from JSON.
    :return: The parsed mark instance.
    :raises UnimplementedTypeError: If the mark type is not registered.
    """
    type_ = dct["type"]
    try:
        klass = MARK_TYPE_TO_CLASS_MAPPING[type_]
    except KeyError:
        raise UnimplementedTypeError(type_, "mark")
    return klass.from_dict(dct)