ommlds 0.0.0.dev494__py3-none-any.whl → 0.0.0.dev496__py3-none-any.whl

ommlds/minichain/content/transforms/interleave.py

@@ -1,70 +0,0 @@
-import typing as ta
-
-from omlish import dataclasses as dc
-from omlish import dispatch
-from omlish import lang
-
-from ..sequence import BlockContent
-from ..sequence import InlineContent
-from ..types import Content
-
-
-##
-
-
-class ContentInterleaver:
-    def __init__(
-            self,
-            *,
-            separator: Content | None = None,
-            sequence_separator: Content | None = None,
-            inline_separator: Content | None = None,
-            block_separator: Content | None = None,
-    ) -> None:
-        super().__init__()
-
-        self._sequence_separator = sequence_separator if sequence_separator is not None else separator
-        self._inline_separator = inline_separator if inline_separator is not None else separator
-        self._block_separator = block_separator if block_separator is not None else separator
-
-    def _interleave(self, l: ta.Iterable[Content], separator: Content | None) -> ta.Sequence[Content]:
-        cs: ta.Iterable[Content] = map(self.interleave, l)
-        if separator is not None:
-            cs = lang.interleave(cs, separator)
-        return list(cs)
-
-    @dispatch.method()
-    def interleave(self, c: Content) -> Content:
-        return c
-
-    @interleave.register
-    def interleave_str(self, c: str) -> Content:
-        return c
-
-    @interleave.register
-    def interleave_sequence(self, c: ta.Sequence) -> Content:
-        return self._interleave(c, self._sequence_separator)
-
-    @interleave.register
-    def interleave_inline_content(self, c: InlineContent) -> Content:
-        return dc.replace(c, l=self._interleave(c.l, self._inline_separator))
-
-    @interleave.register
-    def interleave_block_content(self, c: BlockContent) -> Content:
-        return dc.replace(c, l=self._interleave(c.l, self._block_separator))
-
-
-def interleave_content(
-        c: Content,
-        *,
-        separator: Content | None = None,
-        sequence_separator: Content | None = None,
-        inline_separator: Content | None = None,
-        block_separator: Content | None = None,
-) -> Content:
-    return ContentInterleaver(
-        separator=separator,
-        sequence_separator=sequence_separator,
-        inline_separator=inline_separator,
-        block_separator=block_separator,
-    ).interleave(c)

ommlds/minichain/content/transforms/squeeze.py

@@ -1,72 +0,0 @@
-import typing as ta
-
-from omlish import dataclasses as dc
-from omlish import dispatch
-
-from ..sequence import SequenceContent
-from ..text import TextContent
-from ..types import Content
-from ..types import SingleContent
-
-
-##
-
-
-class ContentSqueezer:
-    def __init__(
-            self,
-            *,
-            strip_strings: bool = False,
-    ) -> None:
-        super().__init__()
-
-        self._strip_strings = strip_strings
-
-    @dispatch.method()
-    def squeeze(self, c: Content) -> ta.Iterable[SingleContent]:
-        raise TypeError(c)
-
-    #
-
-    @squeeze.register
-    def squeeze_str(self, c: str) -> ta.Iterable[SingleContent]:
-        if self._strip_strings:
-            c = c.strip()
-
-        if c:
-            yield c
-
-    @squeeze.register
-    def squeeze_sequence(self, c: ta.Sequence) -> ta.Iterable[SingleContent]:
-        for e in c:
-            yield from self.squeeze(e)
-
-    #
-
-    @squeeze.register
-    def squeeze_single_content(self, c: SingleContent) -> ta.Iterable[SingleContent]:
-        return [c]
-
-    @squeeze.register
-    def squeeze_text_content(self, c: TextContent) -> ta.Iterable[SingleContent]:
-        if self._strip_strings:
-            if (ss := c.s.strip()) != c.s:
-                c = dc.replace(c, s=ss)
-
-        if c.s:
-            yield c.s
-
-    @squeeze.register
-    def squeeze_sequence_content(self, c: SequenceContent) -> ta.Iterable[SingleContent]:
-        for e in c.l:
-            yield from self.squeeze(e)
-
-
-def squeeze_content(
-        c: Content,
-        *,
-        strip_strings: bool = False,
-) -> ta.Sequence[SingleContent]:
-    return list(ContentSqueezer(
-        strip_strings=strip_strings,
-    ).squeeze(c))