Skip to content

Commit 03b46ab

Browse files
authored
Add raw_result to CommandResultItem (#235)
1 parent 4830dab commit 03b46ab

File tree

3 files changed

+6
-3
lines changed

3 files changed

+6
-3
lines changed

knack/cli.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -235,7 +235,7 @@ def invoke(self, args, initial_invocation_data=None, out_file=None):
235235
if cmd_result and cmd_result.result is not None:
236236
formatter = self.output.get_formatter(output_type)
237237
self.output.out(cmd_result, formatter=formatter, out_file=out_file)
238-
self.raise_event(EVENT_CLI_SUCCESSFUL_EXECUTE, result=cmd_result.result)
238+
self.raise_event(EVENT_CLI_SUCCESSFUL_EXECUTE, result=cmd_result)
239239
except KeyboardInterrupt as ex:
240240
exit_code = 1
241241
self.result = CommandResultItem(None, error=ex, exit_code=exit_code)

knack/invocation.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -230,4 +230,5 @@ def execute(self, args):
230230
return CommandResultItem(event_data['result'],
231231
exit_code=0,
232232
table_transformer=cmd_tbl[parsed_args.command].table_transformer,
233-
is_query_active=self.data['query_active'])
233+
is_query_active=self.data['query_active'],
234+
raw_result=cmd_result)

knack/util.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -35,12 +35,14 @@
3535

3636
class CommandResultItem(object): # pylint: disable=too-few-public-methods
3737
def __init__(self, result, table_transformer=None, is_query_active=False,
38-
exit_code=0, error=None):
38+
exit_code=0, error=None, raw_result=None):
3939
self.result = result
4040
self.error = error
4141
self.exit_code = exit_code
4242
self.table_transformer = table_transformer
4343
self.is_query_active = is_query_active
44+
# The result before applying query
45+
self.raw_result = raw_result
4446

4547

4648
class CLIError(Exception):

0 commit comments

Comments
 (0)