summaryrefslogtreecommitdiff
path: root/lib/python/qmk/comment_remover.py
blob: 45a25257f8f54b34551e8340a9ea274b87872f70 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
"""Removes C/C++ style comments from text.

Gratefully adapted from https://stackoverflow.com/a/241506
"""
import re

comment_pattern = re.compile(r'//.*?$|/\*.*?\*/|\'(?:\\.|[^\\\'])*\'|"(?:\\.|[^\\"])*"', re.DOTALL | re.MULTILINE)


def _comment_stripper(match):
    """Removes C/C++ style comments from a regex match.
    """
    s = match.group(0)
    return ' ' if s.startswith('/') else s


def comment_remover(text):
    """Remove C/C++ style comments from text.
    """
    return re.sub(comment_pattern, _comment_stripper, text)