ddeutil-workflow 0.0.56__py3-none-any.whl → 0.0.58__py3-none-any.whl

ddeutil/workflow/utils.py

@@ -24,13 +24,45 @@ from .__types import DictData, Matrix
 
 T = TypeVar("T")
 UTC: Final[ZoneInfo] = ZoneInfo("UTC")
-NEWLINE: Final[str] = "\n\t| ...\t"
+MARK_NL: Final[str] = "||"
+
+
+def prepare_newline(msg: str) -> str:
+    """Prepare message that has multiple newline char.
+
+    :param msg: (str) A message that want to prepare.
+
+    :rtype: str
+    """
+    msg: str = msg.strip("\n").replace("\n", MARK_NL)
+    if MARK_NL not in msg:
+        return msg
+
+    msg_lines: list[str] = msg.split(MARK_NL)
+    msg_last: str = msg_lines[-1]
+    msg_body: str = (
+        "\n" + "\n".join(f" ... |  \t{s}" for s in msg_lines[1:-1])
+        if len(msg_lines) > 2
+        else ""
+    )
+    return msg_lines[0] + msg_body + f"\n ... ╰─ \t{msg_last}"
 
 
 def replace_sec(dt: datetime) -> datetime:
+    """Replace second and microsecond values to 0.
+
+    :param dt: A datetime object that want to replace.
+
+    :rtype: datetime
+    """
     return dt.replace(second=0, microsecond=0)
 
 
+def clear_tz(dt: datetime) -> datetime:
+    """Replace timezone info on an input datetime object to None."""
+    return dt.replace(tzinfo=None)
+
+
 def get_dt_now(tz: ZoneInfo | None = None, offset: float = 0.0) -> datetime:
     """Return the current datetime object.
 
@@ -40,7 +72,7 @@ def get_dt_now(tz: ZoneInfo | None = None, offset: float = 0.0) -> datetime:
     :rtype: datetime
     :return: The current datetime object that use an input timezone or UTC.
     """
-    return datetime.now(tz=(tz or UTC)) - timedelta(seconds=offset)
+    return datetime.now(tz=tz) - timedelta(seconds=offset)
 
 
 def get_d_now(
@@ -54,7 +86,7 @@ def get_d_now(
     :rtype: date
     :return: The current date object that use an input timezone or UTC.
     """
-    return (datetime.now(tz=(tz or UTC)) - timedelta(seconds=offset)).date()
+    return (datetime.now(tz=tz) - timedelta(seconds=offset)).date()
 
 
 def get_diff_sec(dt: datetime, offset: float = 0.0) -> int:
@@ -81,20 +113,19 @@ def reach_next_minute(dt: datetime, offset: float = 0.0) -> bool:
     :param offset: (float) An offset second value.
     """
     diff: float = (
-        replace_sec(dt) - replace_sec(get_dt_now(tz=dt.tzinfo, offset=offset))
+        replace_sec(clear_tz(dt)) - replace_sec(get_dt_now(offset=offset))
     ).total_seconds()
     if diff >= 60:
         return True
     elif diff >= 0:
         return False
-
     raise ValueError(
         "Check reach the next minute function should check a datetime that not "
         "less than the current date"
     )
 
 
-def wait_to_next_minute(
+def wait_until_next_minute(
     dt: datetime, second: float = 0
 ) -> None:  # pragma: no cov
     """Wait with sleep to the next minute with an offset second value."""