Add colouring to Cassandra nodetool status output

ntcolour.py

#! /usr/bin/python

import sys, re

# To use run 
#
# $ nodetool status | ./ntcolour.py
# 
# or to refresh forever use:
#
# $ while true; do status="$(nodetool status)"; clear; date; echo "$status" | ./ntcolour.py; sleep 10; done

red = '\033[97;41m{0}\033[0m'
green = "\033[30;42m{0}\033[0m"
yellow = '\033[30;103m{0}\033[0m'

status = sys.stdin.read().strip()
for line in status.split("\n"):

  #DC
  dc =  re.match("Datacenter\: (.+)", line)
  if dc:
    print "\033[1m{0}\033[m".format(dc.group(1))
    continue;

  #column headings
  if re.match("^--.+", line):
    print line

  # up normal
  if re.match("^UN .+", line):
    print green.format(line)
    continue

  # up normal/joining/leaving/moving
  if re.match("^U[NLJM] .+", line):
    print yellow.format(line)
    continue

  # down normal/joining/leaving/moving
  if re.match("^D[NLJM] .+", line):
    print red.format(line)
    continue