diff options
Diffstat (limited to 'htmtl')
| -rw-r--r-- | htmtl/htmtl.py | 17 | ||||
| -rw-r--r-- | htmtl/parsers/__init__.py | 10 | ||||
| -rw-r--r-- | htmtl/parsers/iterate.py | 67 |
3 files changed, 89 insertions, 5 deletions
diff --git a/htmtl/htmtl.py b/htmtl/htmtl.py index 7732f06..6570e11 100644 --- a/htmtl/htmtl.py +++ b/htmtl/htmtl.py @@ -1,18 +1,21 @@ from typing import Any from dompa import Dompa +from dompa.actions import ToHtml +from dompa.nodes import Node from .parser import Parser from .parsers.generic_value import GenericValue from .parsers.inner_html import InnerHtml from .parsers.inner_partial import InnerPartial from .parsers.inner_text import InnerText +from .parsers.iterate import Iterate from .parsers.outer_html import OuterHtml from .parsers.outer_partial import OuterPartial from .parsers.outer_text import OuterText +from .parsers.when import When +from .parsers.when_not import WhenNot from .modifier import Modifier from .modifiers.truncate import Truncate from .expression_parser import ExpressionParser -from .parsers.when import When -from .parsers.when_not import WhenNot class Htmtl: @@ -27,6 +30,7 @@ class Htmtl: # set default attribute parsers self.__attribute_parsers = [ + Iterate, InnerText, InnerHtml, InnerPartial, @@ -64,7 +68,12 @@ class Htmtl: parser_instance = parser(self.__data, expression_parser) self.__dom.traverse(parser_instance.traverse) - def html(self) -> str: + def to_html(self) -> str: + self.__parse() + + return self.__dom.action(ToHtml) + + def nodes(self) -> list[Node]: self.__parse() - return self.__dom.html()
\ No newline at end of file + return self.__dom.get_nodes()
\ No newline at end of file diff --git a/htmtl/parsers/__init__.py b/htmtl/parsers/__init__.py index 4ad8185..5c98cba 100644 --- a/htmtl/parsers/__init__.py +++ b/htmtl/parsers/__init__.py @@ -1,2 +1,10 @@ from .inner_text import InnerText -from .outer_text import OuterText
\ No newline at end of file +from .inner_html import InnerHtml +from .inner_partial import InnerPartial +from .outer_text import OuterText +from .outer_html import OuterHtml +from .outer_partial import OuterPartial +from .generic_value import GenericValue +from .when import When +from .when_not import WhenNot +from .iterate import Iterate
\ No newline at end of file diff --git a/htmtl/parsers/iterate.py b/htmtl/parsers/iterate.py new file mode 100644 index 0000000..a70d026 --- /dev/null +++ b/htmtl/parsers/iterate.py @@ -0,0 +1,67 @@ +from collections.abc import Iterable +from typing import Optional + +from dompa.nodes.actions import ToHtml +from dompa.nodes import Node, FragmentNode +from ..parser import Parser +import htmtl + +class IterateOp: + var: str + iter_var_as: Optional[str] + iter_index_as: Optional[str] + + def __init__(self, var: str, iter_var_as: Optional[str] = None, iter_index_as: Optional[str] = None): + self.var = var + self.iter_var_as = iter_var_as or None + self.iter_index_as = iter_index_as or None + + +class Iterate(Parser): + def traverse(self, node: Node) -> Optional[Node]: + if "iterate" in node.attributes: + replacement_nodes = [] + iterate_op = self.__parse_exp(node.attributes["iterate"]) + collection = self.expression(iterate_op.var) + node.attributes.pop("iterate") + + if isinstance(collection, Iterable): + data = self.data() + + for idx, item in enumerate(collection): + if iterate_op.iter_var_as: + data[iterate_op.iter_var_as] = item + + if iterate_op.iter_index_as: + data[iterate_op.iter_index_as] = idx + + template = htmtl.Htmtl(node.action(ToHtml), data) + template_nodes = template.nodes() + + if len(template_nodes) > 0: + replacement_nodes.append(template_nodes[0]) + + return FragmentNode(children=replacement_nodes) + + return node + + @staticmethod + def __parse_exp(exp: str) -> IterateOp: + parts = exp.split(" ") + + if len(parts) == 3: + if parts[2].count(":") == 1: + return IterateOp( + var=parts[0].strip(), + iter_index_as=parts[2].split(":")[0].strip(), + iter_var_as=parts[2].split(":")[1].strip() + ) + + return IterateOp( + var=parts[0].strip(), + iter_var_as=parts[2].strip() + ) + + return IterateOp( + var=parts[0] + ) |