Skip to content

Commit 148ce30

Browse files
committed
✨ add feature for removing ANSI codes
1 parent 2a345ab commit 148ce30

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

markdown_generator/mdconverter_class.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -136,6 +136,11 @@ def _markdown_exporter(self) -> None:
136136

137137
self.script, self.resources = exporter.from_notebook_node(self.notebook_content)
138138
self._setup_image_processing()
139+
self._remove_ansi_escape_codes()
140+
141+
def _remove_ansi_escape_codes(self) -> None:
142+
ANSI_ESCAPE_PATTERN = re.compile(r"\x1B\[[0-?]*[ -/]*[@-~]")
143+
self.script = ANSI_ESCAPE_PATTERN.sub("", self.script)
139144

140145
def _setup_image_processing(self) -> None:
141146
"""Set up image processing"""
@@ -184,7 +189,12 @@ def _get_image_type(filename: str) -> str:
184189

185190
def _process_markdown_images_pattern(self) -> None:
186191
"""Handle Markdown image patterns"""
187-
pattern = r"!\[([^\]]+)\]\((\.\/assets\/[^)]+)\)"
192+
pattern = r"!\[([^\]]*)\]\((\.\/assets\/[^)]+)\)"
193+
for match in re.finditer(pattern, self.script):
194+
desc, old_path = match.groups()
195+
self._process_markdown_image(desc, old_path)
196+
197+
pattern = r"!\[([^\]]*)\]\((assets\/[^)]+)\)"
188198
for match in re.finditer(pattern, self.script):
189199
desc, old_path = match.groups()
190200
self._process_markdown_image(desc, old_path)

0 commit comments

Comments
 (0)