From bda7d1755918969411c4bcd6d44a7e33666329f7 Mon Sep 17 00:00:00 2001 From: Marc Mance Date: Wed, 31 Jul 2024 13:04:39 -0400 Subject: [PATCH] added remove ansi escape --- collect.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/collect.py b/collect.py index 6db4b82..3e17b46 100644 --- a/collect.py +++ b/collect.py @@ -100,6 +100,12 @@ def convert_to_megabytes(data_size): return int(size * unit_map[unit]) +def remove_ansi_escape(text): + """Removes ANSI escape sequences from a string.""" + ansi_escape = re.compile(r"\x1B(?:[@-Z\\^[\]m])+") + return ansi_escape.sub("", text) + + def scaleMemstat(): command = ["top", "-n", "1"] @@ -108,8 +114,8 @@ def scaleMemstat(): if collect: mem = re.search("Mem :.*", collect).group() - mem = re.sub("^.*Mem..", "", mem.strip()) - mem = re.sub("[total|free|used|buff.cache|,]", "", mem.strip()).split() + mem = re.sub("Mem..", "", mem.strip()) + mem = re.sub("[total|free|used|buff|cache|,|/]", "", mem.strip()) breakpoint()