Colored diff output with Python

Say you are generating a colored diff output with the standard difflib Python package:

diff = difflib.ndiff(file1_lines, file2_lines)

Now, I'll show you how to write a simple color_diff function that you can use to color your diff like this:

diff = difflib.ndiff(file1_lines, file2_lines)
diff = color_diff(diff)

Here is the code:

    from colorama import Fore, Back, Style, init
except ImportError:  # fallback so that the imported classes always exist
    class ColorFallback():
        __getattr__ = lambda self, name: ''
    Fore = Back = Style = ColorFallback()

def color_diff(diff):
    for line in diff:
        if line.startswith('+'):
            yield Fore.GREEN + line + Fore.RESET
        elif line.startswith('-'):
            yield Fore.RED + line + Fore.RESET
        elif line.startswith('^'):
            yield Fore.BLUE + line + Fore.RESET
            yield line