satori-python-core 0.13.0__tar.gz → 0.13.2__tar.gz

{satori_python_core-0.13.0 → satori_python_core-0.13.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: satori-python-core
-Version: 0.13.0
+Version: 0.13.2
 Summary: Satori Protocol SDK for python, specify common part
 Home-page: https://github.com/RF-Tar-Railt/satori-python
 Author-Email: RF-Tar-Railt <rf_tar_railt@qq.com>

{satori_python_core-0.13.0 → satori_python_core-0.13.2}/pyproject.toml

@@ -23,7 +23,7 @@ classifiers = [
     "Programming Language :: Python :: 3.12",
     "Operating System :: OS Independent",
 ]
-version = "0.13.0"
+version = "0.13.2"
 
 [project.license]
 text = "MIT"

{satori_python_core-0.13.0 → satori_python_core-0.13.2}/src/satori/__init__.py

@@ -25,6 +25,7 @@ from .element import Superscript as Superscript
 from .element import Text as Text
 from .element import Underline as Underline
 from .element import Video as Video
+from .element import select as select
 from .element import transform as transform
 from .model import ArgvInteraction as ArgvInteraction
 from .model import ButtonInteraction as ButtonInteraction
@@ -42,4 +43,4 @@ from .model import Role as Role
 from .model import Upload as Upload
 from .model import User as User
 
-__version__ = "0.13.0"
+__version__ = "0.13.2"

{satori_python_core-0.13.0 → satori_python_core-0.13.2}/src/satori/element.py

@@ -2,11 +2,12 @@ from base64 import b64encode
 from dataclasses import InitVar, dataclass, field, fields
 from io import BytesIO
 from pathlib import Path
-from typing import Any, ClassVar, Dict, List, Optional, Tuple, TypeVar, Union, get_args
+from typing import Any, ClassVar, Dict, List, Optional, Tuple, Type, TypeVar, Union, get_args, overload
 from typing_extensions import override
 
 from .parser import Element as RawElement
 from .parser import escape, param_case
+from .parser import select as select_raw
 
 TE = TypeVar("TE", bound="Element")
 
@@ -566,6 +567,32 @@ def transform(elements: List[RawElement]) -> List[Element]:
     return msg
 
 
+@overload
+def select(elements: Union[Element, List[Element]], query: Type[TE]) -> List[TE]: ...
+
+
+@overload
+def select(elements: Union[Element, List[Element]], query: str) -> List[Element]: ...
+
+
+def select(elements: Union[Element, List[Element]], query: Union[Type[TE], str]):
+    if not elements:
+        return []
+    if isinstance(elements, Element):
+        elements = [elements]
+    if isinstance(query, str):
+        return transform(select_raw("".join(map(str, elements)), query))
+    if query is Element:
+        return elements
+    results = []
+    for elem in elements:
+        if isinstance(elem, query):
+            results.append(elem)
+        if elem.children:
+            results.extend(select(elem.children, query))
+    return results
+
+
 class E:
     text = Text
     at = At
@@ -598,5 +625,7 @@ class E:
     link_button = Button.link
     input_button = Button.input
 
+    select = select
+
     def __new__(cls, *args, **kwargs):
         raise TypeError("E is not instantiable")