bear-utils 0.9.8__py3-none-any.whl → 0.9.9__py3-none-any.whl

bear_utils/_internal/_version.py

@@ -1 +1 @@
-__version__ = "0.9.8"
+__version__ = "0.9.9"

bear_utils/events/events_class.py

@@ -15,14 +15,25 @@ Callback = Callable[..., Any]
 
 
 class Events:
-    """Simple wrapper exposing :mod:`events_module` functionality as methods."""
+    """Simple wrapper exposing :mod:`events_module` functionality as methods.
 
-    # Method names mirror functions from ``events_module`` for familiarity
+    Method names mirror functions from ``events_module`` for familiarity
+    """
 
-    def event_handler(self, event_name: str, func: Callback | None = None):
+    def event_handler(
+        self, event_name: str, func: Callback | None = None
+    ) -> Callable[[Callable[..., Any]], Callable[..., Any]] | Callable[..., Any]:
         """Register ``func`` as a handler for ``event_name``.
 
         Can be used as a decorator when ``func`` is omitted.
+
+        Args:
+            event_name (str): The name of the event to handle.
+            func (Callable, optional): The function to register as a handler. If omitted, returns
+                a decorator that can be used to register a handler function.
+
+        Returns:
+            Callable: If ``func`` is provided, returns the function itself. If ``func`` is None, returns a decorator that can be used to register a handler function.
         """
         if func is None:
             return _event_handler(event_name)

bear_utils/events/events_module.py

@@ -1,6 +1,5 @@
 """Event handling module for Bear Utils."""
 
-import asyncio
 from collections import defaultdict
 from collections.abc import Callable
 from functools import wraps
@@ -9,7 +8,7 @@ from typing import Any
 import weakref
 from weakref import WeakMethod, ref
 
-from bear_utils.extras._async_helpers import is_async_function
+from bear_utils.extras._async_helpers import AsyncResponseModel, create_async_task, is_async_function
 
 Callback = Callable[..., Any]
 
@@ -45,18 +44,20 @@ def set_handler(name: str, func: Callback) -> None:
         _event_registry[name].add(ref(func, _make_callback(name)))
 
 
-def dispatch_event(name: str, *args, **kwargs) -> Any | None:
+def dispatch_event(name: str, single: bool = False, *args, **kwargs) -> list[Any]:
     """Dispatch an event to all registered handlers."""
     results: list[Any] = []
     for func in _event_registry.get(name, []):
         if is_async_function(func):
-            result: Any = asyncio.run(func(*args, **kwargs))  # FIXME: This will crash if called from an async context
+            task: AsyncResponseModel = create_async_task(func(*args, **kwargs))
+            result = task.get_conditional_result(timeout=1)
+            results.append(result)
         else:
             result: Any = func(*args, **kwargs)
             results.append(result)
-    if not results:
-        return None
-    return results[0] if len(results) == 1 else results
+    if single and len(results) == 1:
+        return results[0]
+    return results
 
 
 def event_handler(event_name: str) -> Callable[[Callback], Callback]:

bear_utils/extras/_async_helpers.py

@@ -3,37 +3,65 @@ from asyncio import AbstractEventLoop, Task
 from collections.abc import Callable
 from contextlib import suppress
 import inspect
+import time
+from typing import Any
 
 from pydantic import BaseModel, Field
 
 
+def is_async_function(func: Callable) -> bool:
+    """Check if a function is asynchronous.
+
+    Args:
+        func (Callable): The function/method to check.
+
+    Returns:
+        bool: True if the function is asynchronous, False otherwise.
+    """
+    return inspect.iscoroutinefunction(func) or inspect.isasyncgenfunction(func) or inspect.isasyncgen(func)
+
+
 class AsyncResponseModel(BaseModel):
     """A model to handle asynchronous operations with a function and its arguments."""
 
     loop: AbstractEventLoop | None = Field(default=None, description="The event loop to run the function in.")
     task: Task | None = Field(default=None, description="The task created for the asynchronous function.")
     before_loop: bool = Field(default=False, description="If the function was called from a running loop.")
+    result: Any = Field(default=None, description="The result of the completed task.")
 
     model_config = {"arbitrary_types_allowed": True}
 
+    @property
+    def done(self) -> bool:
+        """Check if the task is completed."""
+        return self.task is not None and self.task.done()
+
+    def get_result(self) -> Any:
+        """Get the result of the completed task."""
+        if not self.task:
+            raise ValueError("No task available")
+        if not self.task.done():
+            raise ValueError("Task not completed yet")
+        if self.result is None:
+            self.result = self.task.result()
+        return self.result
+
+    def get_conditional_result(self, timeout: float = 10.0) -> Any:
+        if self.loop and self.task and not self.before_loop:
+            self.loop.run_until_complete(self.task)
+        start_time: float = time.monotonic()
+        while not self.done:
+            if time.monotonic() - start_time > timeout:
+                raise TimeoutError("Task timed out")
+            time.sleep(0.1)
+        return self.get_result()
+
     def conditional_run(self) -> None:
         """Run the event loop until the task is complete if not in a running loop."""
         if self.loop and self.task and not self.before_loop:
             self.loop.run_until_complete(self.task)
 
 
-def is_async_function(func: Callable) -> bool:
-    """Check if a function is asynchronous.
-
-    Args:
-        func (Callable): The function/method to check.
-
-    Returns:
-        bool: True if the function is asynchronous, False otherwise.
-    """
-    return inspect.iscoroutinefunction(func) or inspect.isasyncgenfunction(func) or inspect.isasyncgen(func)
-
-
 def in_async_loop() -> bool:
     """Check if the current context is already in an async loop.
 

{bear_utils-0.9.8.dist-info → bear_utils-0.9.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bear-utils
-Version: 0.9.8
+Version: 0.9.9
 Summary: Various utilities for Bear programmers, including a rich logging utility, a disk cache, and a SQLite database wrapper amongst other things.
 Author-email: chaz <bright.lid5647@fastmail.com>
 Requires-Python: >=3.12

{bear_utils-0.9.8.dist-info → bear_utils-0.9.9.dist-info}/RECORD

@@ -1,7 +1,7 @@
 bear_utils/__init__.py,sha256=Cin66XUC7cwuJ7ePZwgfdDnwFFPX_CHXOI3dMBWuyY8,1515
 bear_utils/__main__.py,sha256=-FlPquBlI1Tg2RoeX6d0Z8jTAiMFnJ0V06ZeRyiq58k,355
 bear_utils/_internal/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-bear_utils/_internal/_version.py,sha256=GI3Cr15wC6XhdahtMoSOf62d2gYve5G_PiR_xx_RGKc,22
+bear_utils/_internal/_version.py,sha256=nhsA3KKA-CXSYpbzuChuLyxpDepY_-JffnUNClcYEaU,22
 bear_utils/_internal/cli.py,sha256=GWVpdNxtb54EwiJPp4lync7M-hjo8hqVoVB7RFItXaA,4457
 bear_utils/_internal/debug.py,sha256=mpjGkoAHutSmkyPuHk8QA2xCI0ttPkqs8tQtpjFcVqw,5977
 bear_utils/ai/__init__.py,sha256=Q5P1KpSYS6iMt3vRbmasdWU5Oy5UkXfOGGyDI7Qy3Po,747
@@ -36,10 +36,10 @@ bear_utils/constants/time_related.py,sha256=Mykb27THqC-K0HFrbbrdkTNJUKhzBEOfmPJ0
 bear_utils/database/__init__.py,sha256=dS_HHJKESpsI6BFDLVx055o0GCJV9CMYOQVuHs7EfgY,192
 bear_utils/database/_db_manager.py,sha256=bbg5zG06DcOvd37oafLIqc480BwF9ZW9qWF3ldZHR8o,3805
 bear_utils/events/__init__.py,sha256=EFqmuzhaEYK9kjkGlrM7bjdjPwFEDbKn6RjJKfIBEJY,414
-bear_utils/events/events_class.py,sha256=vPDjWrbut8L3TFn7byyYFZpWYM5ADIqtW2Aeh-qtKfQ,1632
-bear_utils/events/events_module.py,sha256=DkQsDZ5WzM1MH1Msg7mRny40a17Jl31CXbpw4-pICxc,2349
+bear_utils/events/events_class.py,sha256=Rzvw-TuX8Tpaf_D0A50fwKiV73BLCKpBTH6Dp08eg3I,2161
+bear_utils/events/events_module.py,sha256=zXl3MB5VtITBCvCRHMRwfCrBkx8GkcH8k1aeB2w9Tnc,2436
 bear_utils/extras/__init__.py,sha256=0mz3ZYOkOcW_tUHSx2pV2mtCfyuXwgPe9SYXEqlGHsg,755
-bear_utils/extras/_async_helpers.py,sha256=4buULZZkR0n2z13Q8_FK59dMchj0Mup03YBaqSCbveQ,2291
+bear_utils/extras/_async_helpers.py,sha256=iTLIJlME5Ifeqy6Y_0kNWAihc2oHCN57U6gehov1aDg,3340
 bear_utils/extras/_tools.py,sha256=cY5XxtUgeJxomeLO_g0X0xX_vEhMrvJkVd4I3-psjaw,6068
 bear_utils/extras/_zapper.py,sha256=K41SlZzvfjiKr04E6IMPxC94XBsKIkvwOgLRIWfYgGA,15830
 bear_utils/extras/platform_utils.py,sha256=Ai7ow7S-_cKb5zFwFh8dkC8xmbMJFy-0_-w3NCERdEw,1362
@@ -102,6 +102,6 @@ bear_utils/monitoring/__init__.py,sha256=9DKNIWTp_voLnaWgiP-wJ-o_N0hYixo-MzjUmg8
 bear_utils/monitoring/_common.py,sha256=LYQFxgTP9fk0cH71IQTuGwBYYPWCqHP_mMRNecoD76M,657
 bear_utils/monitoring/host_monitor.py,sha256=e0TYRJw9iDj5Ga6y3ck1TBFEeH42Cax5mQYaNU8yams,13241
 bear_utils/time/__init__.py,sha256=VctjJG17SyEHAFXytI1sZrOrq7zm3hVenIDOJFdaMN0,1424
-bear_utils-0.9.8.dist-info/METADATA,sha256=vhZfMgtDoFqM3oebfv3bbDz4ye0_hx_yj6Z0e8tULBc,8800
-bear_utils-0.9.8.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-bear_utils-0.9.8.dist-info/RECORD,,
+bear_utils-0.9.9.dist-info/METADATA,sha256=P8iMvX60OVd18V56xwJL7dzZNaqFVxsmfvVIRzLM_30,8800
+bear_utils-0.9.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+bear_utils-0.9.9.dist-info/RECORD,,