debug.py

#!/usr/bin/python -tt
# -*- coding: utf-8 -*-

import traceback
import os
import inspect

from globdata import config

def debugPath(root,path):
  printStack()
  vp = "root"
  vl = 0
  NORM = '\033[0;37;40m' # normal gray
  VALID = '\033[32;40m' # green
  INVALID = '\033[31;40m' # red
  if path == str(path):
    path = [path] # prevents string from being processed as a long list
  if root and root.keys():
    for i in range(len(path)):
#      print "%s %s" % (path[i],root.keys())
      x = root.get(path[i],None)
      if x is not None:
        vl += 1
        vp += " > %s" % VALID + path[i] + NORM
      else:
#        printPretty(root)
        vp += " X %s" % INVALID + path[i] + NORM
      root = root.get(path[i],{})
    print "%d levels: %s" % (vl,vp)
  else:
    say("Root not found!")

def lineno():
    """Returns the current line number in our program."""
    return inspect.currentframe().f_back.f_lineno

def printStack(length = 3,callonly = False):
  stack = traceback.extract_stack()
  start = -2 - length
  for item in stack[start:-2]: # give three lines, trimming call to this function and stack = extract_stack
    p,f = os.path.split(item[0])
    ln = item[1]
    func = item[2]
    cmd = item[3]
    if callonly:
      print "* %s:" % cmd
    else:
      BOLD = ""
      NORM = ""
      if config['termcolors']:
        BOLD = "\033[1;36;40m"
        NORM = "\033[0;37;40m"
      print "* At line %s%d%s in %s%s%s:%s():\n\t%s" % (BOLD,ln,NORM,BOLD,f,NORM,func,cmd)

def printPretty(string,**kwargs):
  """This function prints Python variable/string in colored format in bash/ANSI
  terminal. It makes some broad assumptions that won't always work, particularly,
  that you won't pass it a string containing special Python characters. If you do,
  it will simply put that character in its special color, and you may become confused.
  """
  quiet = True
  xml = False
  length = 1
  stack = True
  for key in kwargs:
    if config['debug'] > 3: print "%s:%s" % (key,kwargs[key])
    if key == "quiet": quiet = kwargs[key]
    if key == "xml": xml = kwargs[key]
    if key == "length": length = kwargs[key]
    if key == "stack": stack = kwargs[key]
  if stack: printStack(length,quiet) # Say what we're printing
  string = str(string) # Treat this as a string, even if it's not, which it usually won't be
  NORM = '\033[0;37;40m' # normal gray
  BRACE = '\033[32;40m' # green
  LIST = '\033[35;40m' # purple
  ANGLES = '\033[31;40m' # red
  COLON = '\033[36;40m' # cyan
  EQ = '\033[1;34;40m' # bold blue
  COMMA = '\033[1;33;40m' # bold yellow
  if not config['termcolors']:
    NORM = ""
    BRACE = ""
    LIST = ""
    ANGLES = ""
    COLON = ""
    EQ = ""
    COMMA = ""
  print NORM,
  output = ""
  depth = 0
  un = 2
  b = ""
  closetag = False
  bump = False
  if string == "{}":
    string = "(empty dictionary)"
  elif string == "":
    string = "(empty string)"
  elif string == "[]":
    string = "(empty list)"
  elif string == "()":
    string = "(empty tuple)"
  for c in string:
    if c == '{':
      depth += un
      output += "%s\n%s " % (BRACE + c + NORM,pad(depth))
    elif c == '}':
      output += "\n%s%s" % (pad(depth),BRACE + c + NORM)
      depth -= un
    elif c == '[':
      depth += un
      output += "%s" % (LIST + c + NORM)
    elif c == ']':
      depth -= un
      output += "%s" % (LIST + c + NORM)
    elif c == ',' and b in ['}',']']:
      output += "%s\n%s" % (c,pad(depth))
    elif c == ',':
      output += "%s" % COMMA + c + NORM
    elif c == '<':
      if not xml: output += "%s" % ANGLES + c + NORM
      elif closetag:
        depth -= un
        bump = True
        closetag = False
      else:
        depth += un
    elif b == '<' and xml:
      x = ""
      if c != '/':
        x = "\n%s" % pad(depth)
      else:
        closetag = True
        if bump:
          x = "\n%s" % pad(depth)
          bump = False
      output += "%s" % x + ANGLES + '<' + NORM + c
    elif c == '>':
      if b == '/': depth -= un
      output += "%s" % ANGLES + c + NORM
    elif c == ':':
      output += "%s" % COLON + c + NORM
    elif c == '=':
      output += "%s" % EQ + c + NORM
    else:
      output += c
    b = str(c)
  print output

def showConfig():
  print "Config:\n-------"
  for key in config.keys():
    value = config[key]
    if value == str(value):
      value = "\'%s\'" % value
    printPretty("'%s':\t%s" % (key,value),stack=False)
  print "-------"

def pad(depth):
  sp = ""
  for i in range(depth):
    sp += ' '
  return sp