1. import json
  2. from typing import Any, Callable, Dict, List, Optional, Union
  5. def serialize_cli_args(args: Dict[str, Any], schema: Dict[str, str]) -> str:
  6. """Serializes CLI arguments to JSON with schema validation and defensive checks.
  8. Args:
  9. args: Dictionary of CLI arguments (e.g., {"--name": "John", "--age": 30}).
  10. schema: Dictionary defining the expected type of each argument.
  11. (e.g., {"--name": "string", "--age": "integer"})
  13. Returns:
  14. JSON string representation of the validated arguments.
  15. Raises ValueError if validation fails.
  16. """
  18. validated_args: Dict[str, Any] = {}
  19. for arg_name, arg_type in schema.items():
  20. if arg_name in args:
  21. value = args[arg_name]
  23. if arg_type == "string":
  24. if not isinstance(value, str):
  25. raise ValueError(f"Argument '{arg_name}' must be a string, got {type(value)}")
  26. validated_args[arg_name] = value
  27. elif arg_type == "integer":
  28. if not isinstance(value, int):
  29. try:
  30. validated_args[arg_name] = int(value) #attempt conversion
  31. except (ValueError, TypeError):
  32. raise ValueError(f"Argument '{arg_name}' must be an integer, got {type(value)}")
  33. else:
  34. validated_args[arg_name] = value
  35. elif arg_type == "float":
  36. if not isinstance(value, (int, float)):
  37. try:
  38. validated_args[arg_name] = float(value) #attempt conversion
  39. except (ValueError, TypeError):
  40. raise ValueError(f"Argument '{arg_name}' must be a float, got {type(value)}")
  41. else:
  42. validated_args[arg_name] = value
  43. elif arg_type == "boolean":
  44. if not isinstance(value, bool):
  45. if isinstance(value, str):
  46. value = value.lower()
  47. if value in ("true", "1", "yes"):
  48. validated_args[arg_name] = True
  49. elif value in ("false", "0", "no"):
  50. validated_args[arg_name] = False
  51. else:
  52. raise ValueError(f"Argument '{arg_name}' must be a boolean, got {type(value)}")
  53. else:
  54. raise ValueError(f"Argument '{arg_name}' must be a boolean, got {type(value)}")
  55. else:
  56. validated_args[arg_name] = value
  57. elif arg_type == "list":
  58. if not isinstance(value, list):
  59. raise ValueError(f"Argument '{arg_name}' must be a list, got {type(value)}")
  60. validated_args[arg_name] = value
  61. else:
  62. raise ValueError(f"Unsupported argument type: {arg_type} for argument '{arg_name}'")
  63. elif arg_name in schema:
  64. raise ValueError(f"Missing required argument: '{arg_name}'")
  66. return json.dumps(validated_args, indent=2)

Add your comment