Merge pull request #27 from vyos-contrib/feat/architecture-and-quality-improvements

feat: v0.4.0 - Architecture refactor, bug fixes, and quality improvem…

1 files changed, 47 insertions, 0 deletions

diff --git a/pyvyos/utils/json.py b/pyvyos/utils/json.py
new file mode 100644
index 0000000..6d9a7c0
--- /dev/null
+++ b/pyvyos/utils/json.py
@@ -0,0 +1,47 @@
+"""JSON utilities with security (redaction) support."""
+
+import json
+from typing import Any, Dict, List
+
+
+def redact_key(data: Dict[str, Any], keys: List[str] = None) -> Dict[str, Any]:
+    """
+    Redact sensitive keys from a dictionary.
+    
+    Args:
+        data: Dictionary to redact
+        keys: List of keys to redact (default: ["key", "apikey", "password"])
+    
+    Returns:
+        Copy of data with specified keys redacted as "***REDACTED***"
+    """
+    if keys is None:
+        keys = ["key", "apikey", "password"]
+    
+    if not isinstance(data, dict):
+        return data
+    
+    redacted = data.copy()
+    for key in keys:
+        if key in redacted:
+            redacted[key] = "***REDACTED***"
+    
+    return redacted
+
+
+def safe_dumps(obj: Any, redact_keys: List[str] = None) -> str:
+    """
+    Safely serialize object to JSON with key redaction.
+    
+    Args:
+        obj: Object to serialize
+        redact_keys: Keys to redact if obj is a dict
+    
+    Returns:
+        JSON string with sensitive data redacted
+    """
+    if isinstance(obj, dict):
+        obj = redact_key(obj, redact_keys)
+    
+    return json.dumps(obj, default=str)
+