From ddb4e6319eb4b71eac7daa5a0864e728f73acb05 Mon Sep 17 00:00:00 2001 From: Carl Montanari Date: Sat, 4 Apr 2020 15:11:40 -0700 Subject: [PATCH] fix structured print for single command --- nornir_scrapli/functions/print_structured_result.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/nornir_scrapli/functions/print_structured_result.py b/nornir_scrapli/functions/print_structured_result.py index 992bf5b..dca21c5 100644 --- a/nornir_scrapli/functions/print_structured_result.py +++ b/nornir_scrapli/functions/print_structured_result.py @@ -5,6 +5,7 @@ from nornir.core.task import AggregatedResult, MultiResult, Result from nornir.plugins.functions.text import _print_result +from scrapli.response import Response LOCK = threading.Lock() @@ -37,6 +38,8 @@ def print_structured_result( updated_multi_result = MultiResult(result.name) for individual_result in multi_result: scrapli_responses = getattr(individual_result, "scrapli_response", None) + if isinstance(scrapli_responses, Response): + scrapli_responses = [scrapli_responses] if not scrapli_responses: updated_multi_result.append(individual_result) continue