Coverage for sel_tools/file_export/file_content_remover.py: 100%

1"""Remove block(s) within delimiters defined in config file from file or multilinestring.""" 

2 

3import re 

4from pathlib import Path 

5 

6from sel_tools.file_export.config import EXPORT_BEGIN, EXPORT_END 

7from sel_tools.utils.files import FileVisitor, is_cpp 

8 

9 

10class SolutionsRemoverVisitor(FileVisitor): 

11 """Remove. 

12 

13 - block(s) within delimiters defined in config file from file 

14 - clang tidy comments 

15 """ 

16 

17 def visit_file(self, file: Path) -> None: 

18 text_with_reduced_content = remove_lines_within_limiters_from_string(file.read_text()) 

19 if is_cpp(file): 

20 text_with_reduced_content = remove_clang_tidy_comment_lines(text_with_reduced_content) 

21 file.write_text(text_with_reduced_content) 

22 

23 

24def remove_lines_within_limiters_from_string(multiline_string: str) -> str: 

25 """Remove block(s) within delimiters defined in config file from multiline string.""" 

26 minimal_length_between_markers_pattern = r"\n[\S ]*" + EXPORT_BEGIN + r".*?[\S ]*" + EXPORT_END 

27 return re.sub(minimal_length_between_markers_pattern, "", multiline_string, flags=re.DOTALL) 

28 

29 

30def remove_clang_tidy_comment_lines(multiline_string: str) -> str: 

31 """Remove clang-tidy comments from multiline string.""" 

32 clang_tidy_nolint_end_of_line = " // NOLINT.*?\n" 

33 multiline_string = re.sub(clang_tidy_nolint_end_of_line, "\n", multiline_string, flags=re.DOTALL) 

34 

35 clang_tidy_nolint_next_line = "// NOLINTNEXTLINE.*?\n" 

36 return re.sub(clang_tidy_nolint_next_line, "", multiline_string, flags=re.DOTALL)