gozerbot.utils.trace

trace related functions

gozerbot.utils.trace.calledfrom(frame)

return plugin name from which this function is called.

gozerbot.utils.trace.callstack(frame)

return the call stackframe in a list.

gozerbot.utils.trace.whichmodule(depth=1)

return in which module this function is called.

CODE

# gozerbot/utils/trace.py
#
#

""" trace related functions """

__status__ = "seen"

basic imports

import sys
import os

calledfrom function

def calledfrom(frame):
    """ return plugin name from which this function is called. """
    try:
        plugname = frame.f_back.f_code.co_filename
        name = plugname.split(os.sep)[-1][:-3]
        if name == '__init__': name = plugname.split(os.sep)[-2]
        if name == 'generic':
            plugname = frame.f_back.f_back.f_code.co_filename
            name = plugname.split(os.sep)[-1][:-3]
    except AttributeError: name = None
    del frame
    return name

callstack function

def callstack(frame):
    """ return the call stackframe in a list. """
    result = []
    loopframe = frame
    while 1:
        try:
            plugname = loopframe.f_back.f_code.co_filename
            result.append("%s:%s" % (plugname.split(os.sep)[-1][:-3], loopframe.f_back.f_lineno))
            loopframe = loopframe.f_back
        except: break
    del frame
    return result

whichmodule function

def whichmodule(depth=1):
    """ return in which module this function is called. """
    try:
        frame = sys._getframe(depth)
        plugfile = frame.f_back.f_code.co_filename[:-3].split('/')
        lineno = frame.f_back.f_lineno
        mod = []
        for i in plugfile[::-1]:
            mod.append(i)
            if i == 'gozerbot' or i == 'gplugs': break
            if i == 'generic':
                plugname = frame.f_back.f_back.f_code.co_filename
                name = plugname.split(os.sep)[-1][:-3]
        modstr = '.'.join(mod[::-1]) + ':' + str(lineno)
    except AttributeError: modstr = None
    del frame
    return modstr