1. def sanitize_collection(data):
  2. """
  3. Sanitizes a collection (list, tuple, dict, or set) by removing potentially harmful characters
  4. and ensuring data types are consistent for testing.
  5. """
  7. if isinstance(data, list):
  8. sanitized_list = []
  9. for item in data:
  10. if isinstance(item, str):
  11. item = item.strip() # Remove leading/trailing whitespace
  12. item = ''.join(c for c in item if c.isalnum() or c in ['.', '_', ' ', '-']) #allow alphanumeric, ., _, space, -
  13. elif isinstance(item, (int, float)):
  14. pass # Keep numbers as is
  15. else:
  16. item = str(item) #Convert other types to string
  17. sanitized_list.append(item)
  18. return sanitized_list
  20. elif isinstance(data, tuple):
  21. return tuple(sanitize_collection(item) for item in data)
  23. elif isinstance(data, dict):
  24. sanitized_dict = {}
  25. for key, value in data.items():
  26. sanitized_key = sanitize_collection(str(key)) #Sanitize keys
  27. sanitized_dict[sanitized_key] = sanitize_collection(value)
  28. return sanitized_dict
  30. elif isinstance(data, set):
  31. return set(sanitize_collection(item) for item in data)
  32. else:
  33. return data # Return as is if not a supported collection type

Add your comment