wehub-resource-sync
35 lines
1.0 KiB
Python
35 lines
1.0 KiB
Python
from datetime import datetime, timezone
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
from pydantic_settings import SettingsConfigDict
|
|
|
|
from rdagent.core.conf import ExtendedBaseSettings
|
|
|
|
|
|
class LogSettings(ExtendedBaseSettings):
|
|
model_config = SettingsConfigDict(env_prefix="LOG_", protected_namespaces=())
|
|
|
|
trace_path: str = str(Path.cwd() / "log" / datetime.now(timezone.utc).strftime("%Y-%m-%d_%H-%M-%S-%f"))
|
|
|
|
format_console: str | None = None
|
|
""""If it is None, leave it as the default"""
|
|
|
|
ui_server_port: int | None = None
|
|
|
|
storages: dict[str, list[int | str]] = {}
|
|
|
|
def set_ui_server_port(self, port: int | None) -> None:
|
|
self.ui_server_port = port
|
|
if port is None:
|
|
self.storages.pop("rdagent.log.ui.storage.WebStorage", None)
|
|
return
|
|
|
|
self.storages["rdagent.log.ui.storage.WebStorage"] = [port, self.trace_path]
|
|
|
|
def model_post_init(self, _context: Any, /) -> None:
|
|
self.set_ui_server_port(self.ui_server_port)
|
|
|
|
|
|
LOG_SETTINGS = LogSettings()
|