Highlight the provided (regexp) arg in the output

highlight.py

#!/usr/bin/env python

import sys
import re

r = re.compile("(?P<subject>" + sys.argv[1] + ")")

for line in sys.stdin:
    print r.sub("\x1B[1;32m\g<subject>\x1B[0m", line)