Warn user when they issue sudo rm -rf, passes all other sudo commands otherwise

sudo_rm.py

#!/usr/bin/env python

import os
import sys

# Warns user before issuing deadly commands with sudo.
#
# add alias to ~/.bashrc like so:
# alias sudo="/home/users/sudo_rm.py"
# then reload bashrc: 
# source ~/.bashrc

class alertColor:
    OKGREEN  = '\033[92m'
    WARNYEL  = '\033[93m'
    WARNRED  = '\033[91m'
    ENDCOLOR = '\033[0m'

if sys.argv[1] == "rm":
    if sys.argv[2] == "-rf":
        print(alertColor.WARNYEL + "WARNING: you are about to execute " + alertColor.OKGREEN + "rm -rf" + alertColor.WARNYEL + " as sudo!" + alertColor.ENDCOLOR)
        warnprompt = raw_input(alertColor.WARNYEL + "ARE YOU SURE you want to " + str(" ".join(sys.argv[1:])) + "(y/n) ??? " + alertColor.ENDCOLOR)
        if warnprompt == "y":
            cmd = "sudo " + (" ".join(sys.argv[1:]))
            os.system(cmd)
            sys.exit(1)
        else:
            print("Not executing command...")
            sys.exit(1)

cmd = "sudo " + (" ".join(sys.argv[1:]))
os.system(cmd)