traceback.py

from curtsies.fmtfuncs import *
import code
import traceback


class Interp(code.InteractiveInterpreter):
    def showtraceback(self):
        output_lines = []
        lines = ['asdf', 'asdfa.py', 'asdfasdf']
        for line in lines:
            if '.py' in line:
                output_lines.append(blue(line))
            else:
                output_lines.append(yellow(line))
        for line in output_lines:
            self.write(str(line))
            self.write('\n')


i = Interp()
i.runsource('asdf')