1. from flask import Flask, request, Response
  3. app = Flask(__name__)
  5. def sandbox_handler():
  6. """Handles requests for sandbox testing with header validation."""
  7. try:
  8. # Sanity checks for required headers
  9. if 'X-Sandbox-Id' not in request.headers:
  10. raise ValueError("Missing X-Sandbox-Id header")
  12. if 'X-Sandbox-Version' not in request.headers:
  13. raise ValueError("Missing X-Sandbox-Version header")
  15. # Basic data validation (example)
  16. try:
  17. sandbox_id = request.headers['X-Sandbox-Id']
  18. version = request.headers['X-Sandbox-Version']
  19. int(sandbox_id) # Check if sandbox_id is an integer
  20. int(version) # Check if version is an integer
  21. except ValueError:
  22. raise ValueError("X-Sandbox-Id and X-Sandbox-Version must be integers.")
  24. # Simulate a response with the validated headers
  25. response_body = "Sandbox test successful!"
  26. return Response(response_body, status=200, headers={'Content-Type': 'text/plain'})
  28. except ValueError as e:
  29. # Return an error response if validation fails
  30. return Response(f"Error: {str(e)}", status=400, headers={'Content-Type': 'text/plain'})
  31. except Exception as e:
  32. # Handle other exceptions
  33. return Response(f"Internal Server Error: {str(e)}", status=500, headers={'Content-Type': 'text/plain'})
  36. @app.route('/sandbox', methods=['GET'])
  37. def sandbox():
  38. """Routes requests to the sandbox handler."""
  39. return sandbox_handler()
  42. if __name__ == '__main__':
  43. app.run(debug=True)

Add your comment