From bfad33c3efeaff60f0f3b36ed858fb27f89239a7 Mon Sep 17 00:00:00 2001 From: Laurent Pinchart Date: Fri, 3 Jul 2020 01:43:57 +0300 Subject: utils: raspberrypi: ctt: json_pretty_print: Collapse newlines Simplify the newline skipping logic by simply collapsing newlines. If a newline has been output, all subsequent newlines will be skipped until the next non-newline character is output. Signed-off-by: Laurent Pinchart Reviewed-by: David Plowman Tested-by: David Plowman --- utils/raspberrypi/ctt/ctt_pretty_print_json.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) (limited to 'utils/raspberrypi/ctt/ctt_pretty_print_json.py') diff --git a/utils/raspberrypi/ctt/ctt_pretty_print_json.py b/utils/raspberrypi/ctt/ctt_pretty_print_json.py index f1c61347..f7fdb651 100644 --- a/utils/raspberrypi/ctt/ctt_pretty_print_json.py +++ b/utils/raspberrypi/ctt/ctt_pretty_print_json.py @@ -23,19 +23,21 @@ class JSONPrettyPrinter(object): self.fout = fout def newline(self): - self.fout.write('\n') - self.state["need_indent"] = True + if not self.state["skipnewline"]: + self.fout.write('\n') + self.state["need_indent"] = True + self.state["skipnewline"] = True def write(self, c): if self.state["need_indent"]: self.fout.write(' ' * self.state["indent"] * 4) self.state["need_indent"] = False self.fout.write(c) + self.state["skipnewline"] = False def process_char(self, c): if c == '{': - if not self.state["skipnewline"]: - self.newline() + self.newline() self.write(c) self.state["indent"] += 1 self.newline() @@ -76,7 +78,6 @@ class JSONPrettyPrinter(object): pass else: self.write(c) - self.state["skipnewline"] = (c == '[') def print(self, string): for c in string: -- cgit v1.2.1