summaryrefslogtreecommitdiff
path: root/htmtl/htmtl.py
blob: ffbbe87427166789b989af85f07ce051523da7bc (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
from typing import Any
from dompa import Dompa
from .attribute_parser import AttributeParser
from .attribute_parsers.inner_text import InnerText
from .expression_modifier import ExpressionModifier
from .expression_parser import ExpressionParser


class Htmtl:
    __dom: Dompa
    __data: dict[str, Any]
    __attribute_parsers: list[type[AttributeParser]]
    __expression_modifiers: list[type[ExpressionModifier]]

    def __init__(self, template: str, data: dict[str, Any] = None):
        self.__dom = Dompa(template)
        self.__data = data or {}
        self.__attribute_parsers = self.__default_attribute_parsers()
        self.__expression_modifiers = []
        self.__parse()

    @staticmethod
    def __default_attribute_parsers() -> list[type[AttributeParser]]:
        return [
            InnerText
        ]

    def set_attribute_parsers(self, parsers: list[type[AttributeParser]]):
        for parser in parsers:
            if not isinstance(parser, AttributeParser):
                raise TypeError("Attribute parser must extend the AttributeParser class.")

        self.__attribute_parsers = parsers

    def set_expression_modifiers(self, modifiers: list[type[ExpressionModifier]]):
        for modifier in modifiers:
            if not isinstance(modifier, ExpressionModifier):
                raise NotImplementedError("Modifier must extend the ExpressionModifier class.")

        self.__expression_modifiers = modifiers

    def __parse(self) -> None:
        expression_parser = ExpressionParser(self.__data, self.__expression_modifiers)

        for parser in self.__attribute_parsers:
            parser_instance = parser(self.__data, expression_parser)
            self.__dom.update(parser_instance.traverse)

    def html(self) -> str:
        return self.__dom.html()