1. import argparse
  2. import sys
  4. def restore_cli_args(args):
  5. """Restores CLI arguments from a dictionary, with defensive checks."""
  7. # Create a new parser with the same arguments as the original.
  8. parser = argparse.ArgumentParser(formatter_class=argparse.RawTextHelpFormatter)
  10. # Add the arguments from the dictionary.
  11. for name, value in args.items():
  12. if isinstance(value, str):
  13. parser.add_argument(name, type=str, help=value) # Default to string type
  14. elif isinstance(value, int):
  15. parser.add_argument(name, type=int, help=str(value)) # Convert int to string for help
  16. elif isinstance(value, float):
  17. parser.add_argument(name, type=float, help=str(value))# Convert float to string for help
  18. elif isinstance(value, bool):
  19. parser.add_argument(name, action='store_true' if value else 'store_false', help=str(value))
  20. else:
  21. print(f"Warning: Unsupported type for argument '{name}'. Skipping.")
  22. continue # Skip unsupported types
  24. # Parse the arguments.
  25. try:
  26. restored_args = parser.parse_args()
  27. return vars(restored_args) # Return dictionary of arguments
  28. except SystemExit:
  29. print("Error restoring arguments. Check the argument values.")
  30. return None # Indicate failure
  31. except Exception as e:
  32. print(f"An unexpected error occurred: {e}")
  33. return None
  37. if __name__ == '__main__':
  38. # Example usage:
  39. # Define the arguments to restore
  40. cli_args = {
  41. 'input_file': 'data.txt',
  42. 'output_dir': 'output',
  43. 'verbose': True,
  44. 'threshold': 0.5
  45. }
  47. # Restore the CLI arguments
  48. restored_args = restore_cli_args(cli_args)
  50. if restored_args:
  51. print("CLI arguments restored:")
  52. print(restored_args)

Add your comment