callback test

cray/modules/rrs/cli.py

@@ -1,7 +1,7 @@
 #
 #  MIT License
 #
-#  (C) Copyright 2020-2023 Hewlett Packard Enterprise Development LP
+#  (C) Copyright 2020-2024 Hewlett Packard Enterprise Development LP
 #
 #  Permission is hereby granted, free of charge, to any person obtaining a
 #  copy of this software and associated documentation files (the "Software"),
@@ -24,5 +24,31 @@
 """ rrs """
 # pylint: disable=invalid-name
 from cray.generator import generate
+import click 
 
-cli = generate(__file__)
+# def print_response(resp):
+#     # Intercept the response by printing it.
+#     print(resp)
+#     print("I am printing")
+#     click.echo(resp)
+#     click.echo("I am using click to echo")
+#     # Return the response unchanged (or modify if needed).
+#     return resp
+
+def _rrs_format_response(cb):
+
+    def _cb(ctx, param, value):
+        data = value.read()
+        print(data)
+        print("I am printing")
+        click.echo(data)
+        click.echo("I am using click to echo")
+        if cb:
+            return cb(ctx, param, data)
+        return data
+
+    return _cb
+
+
+# Pass the callback to generate
+cli = generate(__file__, callback=_rrs_format_response)