ddeutil-workflow 0.0.63__py3-none-any.whl → 0.0.65__py3-none-any.whl

ddeutil/workflow/utils.py

@@ -6,10 +6,12 @@
 """Utility function model."""
 from __future__ import annotations
 
+import asyncio
 import stat
 import time
 from collections.abc import Iterator
 from datetime import date, datetime, timedelta
+from functools import wraps
 from hashlib import md5
 from inspect import isfunction
 from itertools import chain, islice, product
@@ -28,6 +30,16 @@ UTC: Final[ZoneInfo] = ZoneInfo("UTC")
 MARK_NEWLINE: Final[str] = "||"
 
 
+def to_train(camel: str) -> str:
+    """Convert camel case string to train case.
+
+    :param camel: (str) A camel case string that want to convert.
+
+    :rtype: str
+    """
+    return "".join("-" + i if i.isupper() else i for i in camel).lstrip("-")
+
+
 def prepare_newline(msg: str) -> str:
     """Prepare message that has multiple newline char.
 
@@ -293,11 +305,13 @@ def cut_id(run_id: str, *, num: int = 6) -> str:
 
 
 @overload
-def dump_all(value: BaseModel, by_alias: bool = False) -> DictData: ...
+def dump_all(
+    value: BaseModel, by_alias: bool = False
+) -> DictData: ...  # pragma: no cov
 
 
 @overload
-def dump_all(value: T, by_alias: bool = False) -> T: ...
+def dump_all(value: T, by_alias: bool = False) -> T: ...  # pragma: no cov
 
 
 def dump_all(
@@ -311,3 +325,24 @@ def dump_all(
     elif isinstance(value, BaseModel):
         return value.model_dump(by_alias=by_alias)
     return value
+
+
+def awaitable(func):
+    """Dynamic function to async or not depend on the called statement."""
+
+    @wraps(func)
+    async def async_wrapper(*args, **kwargs):
+        return func(*args, **kwargs)
+
+    @wraps(func)
+    def sync_wrapper(*args, **kwargs):
+        return func(*args, **kwargs)
+
+    def dispatch(*args, **kwargs):
+        try:
+            asyncio.get_running_loop()
+            return async_wrapper(*args, **kwargs)
+        except RuntimeError:
+            return sync_wrapper(*args, **kwargs)
+
+    return dispatch