pybot/logger.py

import datetime
import inspect
import os
import sys
import time


class Logger:
  CRITICAL, WARNING, INFO = list(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")
      if sys.version_info > (3, 0, 0):
        f.write(str(datetime.datetime.now()) \
                + " (" + inspect.getframeinfo(inspect.stack()[1][0]).filename \
                + ":"  + str(inspect.getframeinfo(inspect.stack()[1][0]).lineno) \
                + " <"  + inspect.stack()[1][3] + ">) " + str(Logger.levels[level]) \
                + ": " + line + '\n')
      else:
        f.write(str(datetime.datetime.now())+ " (" + inspect.stack()[1][3] + ") " + str(Logger.levels[level]) + ": " + line + '\n')
    except IOError as e:
      print("IO Error!")
      print(e)
    except OSError:
      print("probably permission denied.")