update

utils/logger.py

@@ -5,6 +5,7 @@ import logging.handlers
 import os
 from pathlib import Path
 import sys
+from zoneinfo import ZoneInfo
 
 
 def setup_logging(
@@ -25,6 +26,7 @@ def setup_logging(
     microseconds: int = 3,  # 0, 3, or 6
     include_pid: bool = True,
     include_thread: bool = True,
+    time_zone_name: str = "UTC",
 ) -> None:
     """
     Script-friendly logging:
@@ -106,6 +108,7 @@ def setup_logging(
             include_name=include_name,
             include_pid=include_pid,
             include_thread=include_thread,
+            time_zone_name=time_zone_name,
         )
     else:
         formatter = _TextFormatter(
@@ -116,14 +119,15 @@ def setup_logging(
             include_name=include_name,
             include_pid=include_pid,
             include_thread=include_thread,
+            time_zone_name=time_zone_name,
         )
 
     # ---- Console handler ----
     if use_rich and not json_logs:
         console = RichHandler(
             level=console_level,
-            rich_tracebacks=True,
-            tracebacks_show_locals=True,
+            rich_tracebacks=False,
+            tracebacks_show_locals=False,
             show_time=False,  # YOU already handle time
             show_level=False,  # YOU already handle level
             show_path=False,  # avoid noise
@@ -181,7 +185,7 @@ def _level_from_env(val: str | None, *, default: int) -> int:
     return getattr(logging, name, default)
 
 
-def _format_timestamp(*, calendar: str, microseconds: int) -> str:
+def _format_timestamp(*, calendar: str, microseconds: int, time_zone_name: str) -> str:
     """
     microseconds:
       - 0: YYYY-mm-dd HH:MM:SS
@@ -191,7 +195,7 @@ def _format_timestamp(*, calendar: str, microseconds: int) -> str:
     from datetime import datetime
 
     if calendar == "gregorian":
-        now = datetime.now()
+        now = datetime.now(tz=ZoneInfo(time_zone_name))
         base = now.strftime("%Y-%m-%d %H:%M:%S")
         if microseconds == 0:
             return base
@@ -205,7 +209,7 @@ def _format_timestamp(*, calendar: str, microseconds: int) -> str:
         import jdatetime
     except Exception:
         # fallback
-        now = datetime.now()
+        now = datetime.now(tz=ZoneInfo(time_zone_name))
         base = now.strftime("%Y-%m-%d %H:%M:%S")
         if microseconds == 0:
             return base + " (gregorian-fallback)"
@@ -214,7 +218,7 @@ def _format_timestamp(*, calendar: str, microseconds: int) -> str:
             return f"{base}.{us // 1000:03d} (gregorian-fallback)"
         return f"{base}.{us:06d} (gregorian-fallback)"
 
-    jnow = jdatetime.datetime.now()
+    jnow = jdatetime.datetime.now(tz=ZoneInfo(time_zone_name))
     base = jnow.strftime("%Y-%m-%d %H:%M:%S")
     if microseconds == 0:
         return base
@@ -246,6 +250,7 @@ class _TextFormatter(logging.Formatter):
         include_name: bool,
         include_pid: bool,
         include_thread: bool,
+        time_zone_name: str,
     ):
         super().__init__()
         self.app_name = app_name
@@ -255,9 +260,10 @@ class _TextFormatter(logging.Formatter):
         self.include_name = include_name
         self.include_pid = include_pid
         self.include_thread = include_thread
+        self.time_zone_name = time_zone_name
 
     def format(self, record: logging.LogRecord) -> str:
-        ts = _format_timestamp(calendar=self.calendar, microseconds=self.microseconds)
+        ts = _format_timestamp(calendar=self.calendar, microseconds=self.microseconds, time_zone_name=self.time_zone_name)
 
         # best practice: keep a stable, grep-friendly prefix
         prefix_parts = [ts, record.levelname]
@@ -289,6 +295,7 @@ class _JsonFormatter(logging.Formatter):
         include_name: bool,
         include_pid: bool,
         include_thread: bool,
+        time_zone_name: str,
     ):
         super().__init__()
         self.app_name = app_name
@@ -298,12 +305,15 @@ class _JsonFormatter(logging.Formatter):
         self.include_name = include_name
         self.include_pid = include_pid
         self.include_thread = include_thread
+        self.time_zone_name = time_zone_name
 
     def format(self, record: logging.LogRecord) -> str:
         import json
 
         payload: dict[str, object] = {
-            "ts": _format_timestamp(calendar=self.calendar, microseconds=self.microseconds),
+            "ts": _format_timestamp(
+                calendar=self.calendar, microseconds=self.microseconds, time_zone_name=self.time_zone_name
+            ),
             "level": record.levelname,
             "msg": record.getMessage(),
         }
@@ -329,8 +339,9 @@ if __name__ == "__main__":
         include_name=True,
         include_pid=False,
         include_thread=False,
-        console_level=logging.DEBUG,
+        console_level=logging.INFO,
         use_rich=True,
+        time_zone_name="Asia/Tehran",
     )
     logging.getLogger("elasticsearch").setLevel(logging.WARNING)
     logging.getLogger("elastic_transport").setLevel(logging.WARNING)
@@ -338,6 +349,7 @@ if __name__ == "__main__":
     logging.getLogger("tortoise.db_client").setLevel(logging.WARNING)
     logging.getLogger("asyncio").setLevel(logging.WARNING)
     logging.getLogger("tortoise").setLevel(logging.WARNING)
+    logging.getLogger("httpx").setLevel(logging.WARNING)
     log = logging.getLogger(__name__)
     log.info("script_started")
     # time.sleep(2)