Remove logging directory from aitbc-core src to avoid shadowing Python's standard logging module
Some checks failed
Integration Tests / test-service-integration (push) Successful in 1m24s
Package Tests / Python package - aitbc-agent-sdk (push) Successful in 11s
Package Tests / Python package - aitbc-core (push) Failing after 9s
Package Tests / Python package - aitbc-crypto (push) Successful in 9s
Package Tests / Python package - aitbc-sdk (push) Successful in 12s
Package Tests / JavaScript package - aitbc-sdk-js (push) Successful in 16s
Package Tests / JavaScript package - aitbc-token (push) Has been cancelled
Python Tests / test-python (push) Successful in 30s
Security Scanning / security-scan (push) Successful in 37s

This commit is contained in:
aitbc
2026-04-25 20:51:49 +02:00
parent 23ee227fd3
commit e2d2d7ed43

View File

@@ -1,108 +0,0 @@
from __future__ import annotations
import logging
import sys
from datetime import datetime
from typing import Any, Optional
import json
class StructuredLogFormatter(logging.Formatter):
"""Custom JSON formatter for structured logging."""
RESERVED = {
"name", "msg", "args", "levelname", "levelno", "pathname", "filename",
"module", "exc_info", "exc_text", "stack_info", "lineno", "funcName",
"created", "msecs", "relativeCreated", "thread", "threadName",
"processName", "process"
}
def __init__(self, service_name: str, env: str = "production"):
super().__init__()
self.service_name = service_name
self.env = env
def format(self, record: logging.LogRecord) -> str:
payload: dict[str, Any] = {
"timestamp": datetime.fromtimestamp(record.created).isoformat() + "Z",
"level": record.levelname,
"service": self.service_name,
"env": self.env,
"logger": record.name,
"message": record.getMessage(),
}
# Add exception info if present
if record.exc_info:
ei = record.exc_info
# In Python 3.12+, exc_info can be True to indicate lazy capture; resolve it now.
if ei is True:
ei = sys.exc_info()
if ei:
payload["exception"] = self.formatException(ei)
# Add stack info if present
if record.stack_info:
payload["stack"] = self.formatStack(record.stack_info)
# Add extra fields passed in the record
for key, value in record.__dict__.items():
if key not in self.RESERVED and not key.startswith("_"):
# Make sure value is JSON serializable
try:
json.dumps(value)
payload[key] = value
except (TypeError, ValueError):
payload[key] = str(value)
return json.dumps(payload)
def setup_logger(
name: str,
service_name: str,
env: str = "production",
level: int | str = logging.INFO,
log_file: Optional[str] = None
) -> logging.Logger:
"""Setup a logger with structured JSON formatting."""
logger = logging.getLogger(name)
logger.setLevel(level)
# Don't propagate to root logger
logger.propagate = False
# Remove existing handlers to avoid duplicates
for handler in list(logger.handlers):
logger.removeHandler(handler)
formatter = StructuredLogFormatter(service_name=service_name, env=env)
# Console handler
console_handler = logging.StreamHandler(sys.stdout)
console_handler.setFormatter(formatter)
logger.addHandler(console_handler)
# File handler (optional)
if log_file:
file_handler = logging.FileHandler(log_file)
file_handler.setFormatter(formatter)
logger.addHandler(file_handler)
return logger
def get_logger(name: str) -> logging.Logger:
"""Get an existing logger or create a generic one if it doesn't exist."""
return logging.getLogger(name)
def get_audit_logger(service_name: str) -> logging.Logger:
"""Get a dedicated logger for security/audit events."""
logger = logging.getLogger(f"{service_name}.audit")
if not logger.handlers:
logger = setup_logger(
name=f"{service_name}.audit",
service_name=service_name,
level=logging.INFO
)
return logger