logstack.py

import inspect
import logging
import re

# from https://github.com/baniuyao/Python-CLog/blob/master/CLog.py

class LogRecordWithStack(logging.LogRecord):
    def __init__(self, *args, **kargs):
        super(LogRecordWithStack, self).__init__(*args, **kargs)
        self.chain = self.get_meta_data()

    def get_file_name_in_full_path(self, file_path):
        return file_path.split('/')[-1]

    def get_meta_data(self):
        frames = inspect.stack()
        chain_list = []
        for i in range(0, len(frames)):
            _, file_path, _, func_name, _, _ = frames[i]
            file_name = self.get_file_name_in_full_path(file_path)
            try:
                args = re.findall('\((.*)\)', frames[i+1][-2][0])[0]
            except IndexError, e:
                func_name = LogRecordWithStack.get_class_from_frame(frames[2][0]).__name__
                args = ''
            current_chain = '%s:%s(%s)' % (file_name, func_name, args)
            chain_list.append(current_chain)
        chain_list.reverse()
        return ' --> '.join(chain_list[1:-6])

    @staticmethod
    def get_class_from_frame(fr):
        args, _, _, value_dict = inspect.getargvalues(fr)
        if len(args) and args[0] == 'self':
            instance = value_dict.get('self', None)
            if instance:
                return getattr(instance, '__class__', None)
        return None


logging.LogRecord = LogRecordWithStack
logging.basicConfig(level=logging.DEBUG,
                    format='%(asctime)s [%(chain)s] %(message)s')

def test(zzz):
    logging.debug("hahha %s" % zzz)

def main():
    test('aaa')

if __name__ == '__main__':
    main()