pyroflow 0.2.0__tar.gz → 0.2.1__tar.gz

{pyroflow-0.2.0 → pyroflow-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyroflow
-Version: 0.2.0
+Version: 0.2.1
 Summary: Conversation-oriented Pyrogram extension with per-update listeners and multi-session coordination
 Project-URL: Homepage, https://github.com/eeeob/pyroflow
 Project-URL: Bug Tracker, https://github.com/eeeob/pyroflow/issues

{pyroflow-0.2.0 → pyroflow-0.2.1}/pyroflow/__init__.py

@@ -1,3 +1,5 @@
+from .__meta__ import __version__
+
 from .listener_coordinator import *
 from .update_coordinated import *
 from .update_coordinator import *
@@ -12,3 +14,4 @@ from .types import Message, CallbackQuery
 
 
 
+

pyroflow-0.2.1/pyroflow/__meta__.py

@@ -0,0 +1 @@
+__version__ = "0.2.1"

{pyroflow-0.2.0 → pyroflow-0.2.1}/pyroflow/enums.py

@@ -6,8 +6,10 @@ else:
     from enum import Enum
     
     class StrEnum(str, Enum):
-        def _generate_next_value_(self, *args):
-            return self.lower()
+        def _generate_next_value_(name, *args):
+            return name.lower()
+        def __str__(self):
+            return str(self.value)
 
 from enum import IntEnum, auto
 

{pyroflow-0.2.0 → pyroflow-0.2.1}/pyroflow/utils/async_tools.py

@@ -3,7 +3,7 @@ from concurrent.futures import ThreadPoolExecutor
 
 from .typings import NestedContainer, MaybeAwaitable, _True, _False, _P, _T
 from .validate_tools import is_exception, iscoroutinefunction_wrapped
-from .iter_tools import flat_cont
+from .iter_tools import iter_flat_cont
 
 import asyncio
 import functools
@@ -103,9 +103,9 @@ async def gather_helper(
 
     caller_stack = traceback.extract_stack()[:-1]
 
-    results = await asyncio.gather(*flat_cont(coros), return_exceptions=return_exc)
+    results = await asyncio.gather(*iter_flat_cont(coros), return_exceptions=return_exc)
 
-    if log_exc:
+    if log_exc and return_exc:
         for i, r in enumerate(results):
             if is_exception(r):
                 _log_exc("error in gather_helper", caller_stack, r, index=i)
@@ -159,10 +159,11 @@ async def safe_await(
     caller_stack = traceback.extract_stack()[:-1]
 
     results = []
-    flat_coros = flat_cont(coros)
-    is_multi = len(flat_coros) > 1
+    is_multi = False
 
-    for i, coro in enumerate(flat_coros):
+    for i, coro in enumerate(iter_flat_cont(coros)):
+        if not is_multi and i > 0:
+            is_multi = True
         try:
             result = await coro
         except Exception as e:

pyroflow-0.2.1/pyroflow/utils/iter_tools.py

@@ -0,0 +1,37 @@
+from typing import List, Generator, Any, overload
+
+
+from .typings import NestedContainer, _T
+from .validate_tools import is_container
+
+
+
+
+@overload
+def iter_flat_cont(*containers: None) -> Generator[Any, None, None]: ...
+@overload
+def iter_flat_cont(*containers: NestedContainer[None]) -> Generator[Any, None, None]: ...
+@overload
+def iter_flat_cont(*containers: NestedContainer[_T]) -> Generator[_T, None, None]: ...
+def iter_flat_cont(*containers):
+    for item in containers:
+        if is_container(item):
+            yield from iter_flat_cont(*item)
+        elif item is not None:
+            yield item
+
+@overload
+def flat_cont(*containers: None) -> List: ...
+@overload
+def flat_cont(*containers: NestedContainer[None]) -> List: ...
+@overload
+def flat_cont(*containers: NestedContainer[_T]) -> List[_T]: ...
+def flat_cont(*containers):
+    return list(iter_flat_cont(*containers))
+
+
+__all__ = (
+    "iter_flat_cont", 
+    "flat_cont", 
+    
+)

{pyroflow-0.2.0 → pyroflow-0.2.1}/pyroflow/utils/validate_tools.py

@@ -1,4 +1,4 @@
-from typing import Any, Union
+from typing import Any, Union, get_args
 
 import sys
 import inspect
@@ -11,13 +11,16 @@ else:
 
 from .typings import Container, NotContainer, _T
 
+_CONTAINER_TYPES = get_args(Container)
+_NOT_CONTAINER_TYPES = get_args(NotContainer)
+
 def is_exception(obj: Any) -> TypeIs[BaseException]:
     return isinstance(obj, BaseException)
 
 def is_container(obj: Any) -> TypeIs[Union['Container[_T]', Any]]:
-    return isinstance(obj, Container) and not isinstance(obj, NotContainer)
+    return isinstance(obj, _CONTAINER_TYPES) and not isinstance(obj, _NOT_CONTAINER_TYPES)
 
-def iscoroutinefunction_wrapped(f):
+def iscoroutinefunction_wrapped(f) -> bool:
     is_coro = False
 
     def _stop(func):

pyroflow-0.2.0/pyroflow/__meta__.py

@@ -1 +0,0 @@
-__version__ = "0.2.0"

pyroflow-0.2.0/pyroflow/utils/iter_tools.py

@@ -1,32 +0,0 @@
-from typing import List, overload, Any, Generator
-
-from .typings import NestedContainer, _T
-from .validate_tools import is_container
-
-
-@overload
-def flat_cont(*containers: None) -> List: ...
-@overload
-def flat_cont(*containers: NestedContainer[None]) -> List: ...
-@overload
-def flat_cont(*containers: NestedContainer[_T]) -> List[_T]: ...
-def flat_cont(*containers):
-
-    def _flat_generator(item: Any) -> Generator[Any, None, None]:
-        if is_container(item):
-            for i in item:
-                yield from _flat_generator(i)
-        elif item is not None:
-            yield item
-
-    result = []
-    
-    for item in containers:
-        result.extend(_flat_generator(item))
-        
-    return result
-
-
-__all__ = (
-    "flat_cont", 
-)