File size: 1,054 Bytes
4303959 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
from __future__ import annotations
import os
from typing import Any, Dict
def _flag(name: str) -> bool:
val = os.getenv(name, "0").lower()
return val in ("1", "true", "yes")
def debug_enabled() -> bool:
return _flag("NSA_DEBUG_LOG")
_COUNTS: Dict[str, int] = {}
def log(tag: str, **fields: Any) -> None:
if not debug_enabled():
return
limit_env = os.getenv("NSA_LOG_LIMIT")
if limit_env is not None:
try:
limit = int(limit_env)
except Exception:
limit = 0
if limit > 0:
cnt = _COUNTS.get(tag, 0)
if cnt >= limit:
return
_COUNTS[tag] = cnt + 1
parts = [f"{k}={_safe(v)}" for k, v in fields.items()]
print(f"NSA-LOG {tag} " + " ".join(parts))
def _safe(v: Any) -> str:
try:
if isinstance(v, int | float | str):
return str(v)
if hasattr(v, "shape"):
return str(tuple(int(x) for x in v.shape))
return str(v)
except Exception:
return "<unrepr>"
|