pybot/logger.py

import datetime
import time
import inspect
import sys
import os
class Logger:
  CRITICAL, WARNING, INFO = range(3)
  levels = ['CRITICAL', 'WARNING', 'INFO']

  def write(self, level, line, nick=None, location=None):
    """
    Write out to the logfile of either default location or otherwise specified.
    Includes calling class in its logged line.

    Args:
    level: enumerated thing from the logger class.
    line: string. thing to write out to the logger.
    nick: string. determines filename.
    location: where to write the logfile out to.

    Returns: 
    nothing.
      
    """
    if location is not None:
      l = location
    else:
      l = "~/"
    if nick is not None:
      n = nick
    else:
      n = 'pylog'

    try:
      if not os.path.exists(os.path.expanduser(l + '/logs/')):
        os.makedirs(os.path.expanduser(l + '/logs'))
      f = open(os.path.expanduser(l + '/logs/' + n + '-'+str(time.strftime("%m-%d-%Y")))+'.log', "a")
      f.write(str(datetime.datetime.now())+ " (" + inspect.stack()[1][3] + ") " + str(Logger.levels[level]) + ": " + line + '\n')
    except IOError, e:
      print "IO Error!"
      print e
    except OSError:
      print "probably permission denied."
      pass