Bring back result_callback

Author: jonavellecuerdo

tim/cli.py

@@ -63,13 +63,14 @@ def main(ctx: click.Context, url: str, *, verbose: bool) -> None:
     ctx.obj["CLIENT"] = tim_os.configure_opensearch_client(url)
     logger.info("OpenSearch client configured for endpoint '%s'", url)
 
-    def _log_command_elapsed_time() -> None:
-        elapsed_time = perf_counter() - ctx.obj["start_time"]
-        logger.info(
-            "Total time to complete process: %s", str(timedelta(seconds=elapsed_time))
-        )
 
-    ctx.call_on_close(_log_command_elapsed_time)
+@main.result_callback()
+@click.pass_context
+def log_process_time(ctx: click.Context, _result: object, **_kwargs: dict) -> None:
+    elapsed_time = perf_counter() - ctx.obj["START_TIME"]
+    logger.info(
+        "Total time to complete process: %s", str(timedelta(seconds=elapsed_time))
+    )
 
 
 # Cluster commands