prediction-market-agent-tooling 0.22.0__py3-none-any.whl → 0.23.0__py3-none-any.whl

prediction_market_agent_tooling/loggers.py

@@ -1,13 +1,11 @@
+import logging
 import sys
-import typing as t
+import warnings
 from enum import Enum
 
 from loguru import logger
 from pydantic_settings import BaseSettings, SettingsConfigDict
 
-if t.TYPE_CHECKING:
-    from loguru import Logger
-
 
 class LogFormat(str, Enum):
     DEFAULT = "default"
@@ -22,33 +20,56 @@ class LogConfig(BaseSettings):
     LOG_FORMAT: LogFormat = LogFormat.DEFAULT
 
 
-GCP_LOG_FORMAT = "{level:<.1}{time:MMDD HH:mm:ss.SSSSSS} {process} {name}:{line}] {message} | {extra}"
+GCP_LOG_LOGURU_FORMAT = (
+    "{level:<.1}{time:MMDD HH:mm:ss} {process} {name}:{line}] {message}"
+)
+GCP_LOG_LOGGING_FORMAT, GCP_LOG_FORMAT_LOGGING_DATEFMT = (
+    "%(levelname).1s%(asctime)s %(process)d %(name)s:%(lineno)d] %(message)s"
+), "%m%d %H:%M:%S"
 
 
-def patch_logger(logger: "Logger") -> None:
+def patch_logger() -> None:
     config = LogConfig()
 
     if config.LOG_FORMAT == LogFormat.GCP:
-        format_ = GCP_LOG_FORMAT
+        format_loguru = GCP_LOG_LOGURU_FORMAT
+        format_logging = GCP_LOG_LOGGING_FORMAT
+        datefmt_logging = GCP_LOG_FORMAT_LOGGING_DATEFMT
 
     elif config.LOG_FORMAT == LogFormat.DEFAULT:
-        format_ = None
+        format_loguru, format_logging, datefmt_logging = None, None, None
 
     else:
         raise ValueError(f"Unknown log format: {config.LOG_FORMAT}")
 
-    if format_ is not None:
+    # Change built-in logging.
+    if format_logging is not None:
+        logging.basicConfig(
+            level=logging.DEBUG, format=format_logging, datefmt=datefmt_logging
+        )
+
+    # Change loguru.
+    if format_loguru is not None:
         logger.remove()
         logger.add(
             sys.stdout,
-            format=format_,
+            format=format_loguru,
             level="DEBUG",  # Can be the lowest level, higher ones will use by default this one.
             colorize=True,
         )
 
+    # Change warning formatting to a simpler one (no source code in a new line).
+    warnings.formatwarning = simple_warning_format
+    # Use logging module for warnings.
+    logging.captureWarnings(True)
+
     logger.info(f"Patched logger for {config.LOG_FORMAT.value} format.")
 
 
+def simple_warning_format(message, category, filename, lineno, line=None):  # type: ignore[no-untyped-def] # Not typed in the standard library neither.
+    return f"{category.__name__}: {message}"
+
+
 if not getattr(logger, "_patched", False):
-    patch_logger(logger)
+    patch_logger()
     logger._patched = True  # type: ignore[attr-defined] # Hacky way to store a flag on the logger object, to not patch it multiple times.

{prediction_market_agent_tooling-0.22.0.dist-info → prediction_market_agent_tooling-0.23.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prediction-market-agent-tooling
-Version: 0.22.0
+Version: 0.23.0
 Summary: Tools to benchmark, deploy and monitor prediction market agents.
 Author: Gnosis
 Requires-Python: >=3.10,<3.12

{prediction_market_agent_tooling-0.22.0.dist-info → prediction_market_agent_tooling-0.23.0.dist-info}/RECORD

@@ -19,7 +19,7 @@ prediction_market_agent_tooling/deploy/gcp/deploy.py,sha256=CYUgnfy-9XVk04kkxA_5
 prediction_market_agent_tooling/deploy/gcp/kubernetes_models.py,sha256=qYIHRxQLac3yxtZ8ChikiPG9O1aUQucHW0muTSm1nto,2627
 prediction_market_agent_tooling/deploy/gcp/utils.py,sha256=oyW0jgrUT2Tr49c7GlpcMsYNQjoCSOcWis3q-MmVAhU,6089
 prediction_market_agent_tooling/gtypes.py,sha256=xGSJXw12hzp8LwvQ956l01GiZMWd07MZTYqo8CXVeLY,2417
-prediction_market_agent_tooling/loggers.py,sha256=wIsV7DoB9XOWJKElSvP4ZeUqC0rArCMQS28wcZIYkww,1403
+prediction_market_agent_tooling/loggers.py,sha256=0znHrzxSbeBaDiB920EZC6a2TiF0tw80Sa_yoLwvo_w,2289
 prediction_market_agent_tooling/markets/agent_market.py,sha256=qeuIXEujziDu_gQPvMiHsGSejhkPQtHhO_E7l16NOL0,5997
 prediction_market_agent_tooling/markets/categorize.py,sha256=yTd-lDMPW4ESDSzmxeLLBuzLX0FggOF7Vbswh7295o0,941
 prediction_market_agent_tooling/markets/data_models.py,sha256=uODY3aoFp8YYeLAUcrzMk1yU8pIKsTLobB9xIEGTmKs,1170
@@ -65,8 +65,8 @@ prediction_market_agent_tooling/tools/safe.py,sha256=h0xOO0eNtitClf0fPkn-0oTc6A_
 prediction_market_agent_tooling/tools/singleton.py,sha256=CiIELUiI-OeS7U7eeHEt0rnVhtQGzwoUdAgn_7u_GBM,729
 prediction_market_agent_tooling/tools/utils.py,sha256=zkmwPi3YisgZDPCeNwaRbL8sInOYOkvFgFY4Q8PbEo4,5077
 prediction_market_agent_tooling/tools/web3_utils.py,sha256=cboATXNmEdn5RmPbVblHOwOdUMKBYrUK3GiI6i6Vzxo,9855
-prediction_market_agent_tooling-0.22.0.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.22.0.dist-info/METADATA,sha256=0hbWI60DDcdP3a4am9cGdGHx4J86DcH8gYmQ4oyAxfo,5563
-prediction_market_agent_tooling-0.22.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-prediction_market_agent_tooling-0.22.0.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.22.0.dist-info/RECORD,,
+prediction_market_agent_tooling-0.23.0.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.23.0.dist-info/METADATA,sha256=gkLdQmTB2vbEO3toPUFru7h7BqsIvLm50aXGbNDK1xM,5563
+prediction_market_agent_tooling-0.23.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+prediction_market_agent_tooling-0.23.0.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.23.0.dist-info/RECORD,,