color_gcc.py 文件源码

def format(self, rec):
        frame = sys._getframe()
        while frame:
            func = frame.f_code.co_name
            if func == 'exec_command':
                cmd = frame.f_locals['cmd']
                if isinstance(cmd, list) and ('gcc' in cmd[0] or 'g++' in cmd[0]):
                    lines = []
                    for line in rec.msg.splitlines():
                        if 'warning: ' in line:
                            lines.append(self.colors.YELLOW + line)
                        elif 'error: ' in line:
                            lines.append(self.colors.RED + line)
                        elif 'note: ' in line:
                            lines.append(self.colors.CYAN + line)
                        else:
                            lines.append(line)
                    rec.msg = "\n".join(lines)
            frame = frame.f_back
        return Logs.formatter.format(self, rec)