logger.py

#!/usr/bin/env python
### Script provided by DataStax.

import urllib2
import os
import re
import shlex
import subprocess
import sys
import time

from exceptions import SystemExit

configfile = 'ami.log'

def appendLog(text):
    with open(configfile, "a") as f:
        f.write(text + "\n")
        print text

def exe(command, log=True, expectError=False):
    # Helper function to execute commands and print traces of the command and output for debugging/logging purposes
    process = subprocess.Popen(shlex.split(command), stderr=subprocess.PIPE, stdout=subprocess.PIPE)
    read = process.communicate()
    
    if log:
        # Print output on next line if it exists
        if len(read[0]) > 0:
            appendLog('[EXEC] ' + time.strftime("%m/%d/%y-%H:%M:%S", time.localtime()) + ' ' + command + ":\n" + read[0])
        elif len(read[1]) > 0:
            if expectError:
                appendLog('[EXEC:E] ' + time.strftime("%m/%d/%y-%H:%M:%S", time.localtime()) + ' ' + command + ":\n" + read[1])
            else:
                appendLog('[ERROR] ' + time.strftime("%m/%d/%y-%H:%M:%S", time.localtime()) + ' ' + command + ":\n" + read[1])
    
    if not log or (len(read[0]) == 0 and len(read[1]) == 0):
        appendLog('[EXEC] ' + time.strftime("%m/%d/%y-%H:%M:%S", time.localtime()) + ' ' + command)
    
    return read

def pipe(command1, command2, log=True):
    # Helper function to execute piping commands and print traces of the commands and output for debugging/logging purposes
    p1 = subprocess.Popen(shlex.split(command1), stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    p2 = subprocess.Popen(shlex.split(command2), stdin=p1.stdout, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    p1.stdout.close()  # Allow p1 to receive a SIGPIPE if p2 exits.
    read = p2.stdout.read()

    if not log:
        read = ""
    
    # Print output on next line if it exists
    if len(read) > 0:
        appendLog('[PIPE] ' + time.strftime("%m/%d/%y-%H:%M:%S", time.localtime()) + ' ' + command1 + ' | ' + command2 + ":\n" + read)
    else:
        appendLog('[PIPE] ' + time.strftime("%m/%d/%y-%H:%M:%S", time.localtime()) + ' ' + command1 + ' | ' + command2)

    output = p2.communicate()[0]
    
    if log:
        if read and len(read) > 0:
            appendLog('[PIPE] ' + time.strftime("%m/%d/%y-%H:%M:%S", time.localtime()) + ' ' + command1 + ' | ' + command2 + ":\n" + read)
 
        if output and len(output[0]) > 0:
            appendLog('[PIPE] ' + time.strftime("%m/%d/%y-%H:%M:%S", time.localtime()) + ' ' + command1 + ' | ' + command2 + ":\n" + output[0])
        if output and len(output[1] > 0):
            appendLog('[PIPE] ' + time.strftime("%m/%d/%y-%H:%M:%S", time.localtime()) + ' ' + command1 + ' | ' + command2 + ":\n" + output[1])

        return output

def debug(infotext):
    appendLog('[DEBUG] ' + str(infotext))

def info(infotext):
    appendLog('[INFO] ' + str(infotext))

def warn(infotext):
    appendLog('[WARN] ' + str(infotext))

def error(infotext):
    appendLog('[ERROR] ' + str(infotext))

def exception(filename):
    if type(sys.exc_info()[1]) == SystemExit:
        return
    
    appendLog("[ERROR] Exception seen in %s:" % filename)
    import traceback
    appendLog(traceback.format_exc())
    sys.exit(1)