From 85207c104952d991b107aa6d2fd7329b4a8f915a Mon Sep 17 00:00:00 2001 From: thomas-bc Date: Tue, 21 Mar 2023 09:59:57 -0700 Subject: [PATCH] Handle whitespaces --- src/fprime/util/code_formatter.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/fprime/util/code_formatter.py b/src/fprime/util/code_formatter.py index 9d5ada23..de40beb8 100644 --- a/src/fprime/util/code_formatter.py +++ b/src/fprime/util/code_formatter.py @@ -97,9 +97,9 @@ def _preprocess_files(self) -> None: with open(filepath, "r") as file: content = file.read() # Replace the strings in the file content - content = re.sub("PROTECTED:", PROTECTED_PRE_PATTERN, content) - content = re.sub("PRIVATE:", PRIVATE_PRE_PATTERN, content) - content = re.sub("STATIC:", STATIC_PRE_PATTERN, content) + content = re.sub("PROTECTED[\s]*:", PROTECTED_PRE_PATTERN, content) + content = re.sub("PRIVATE[\s]*:", PRIVATE_PRE_PATTERN, content) + content = re.sub("STATIC[\s]*:", STATIC_PRE_PATTERN, content) # Write the file out to the same location, seemingly in-place with open(filepath, "w") as file: file.write(content)