traceback test

code.py

from pprint import pprint
from cStringIO import StringIO
stdout = StringIO()

def test():

    try:
        
        a
    
    except:
        
        import sys, traceback
        
        tb = sys.exc_info()[2]
        pprint(traceback.extract_tb(tb), stream=stdout)

        print stdout.getvalue()
        
        
test()

main.py

def main():

    _ = {}
    code = ''

    args = ''
    argv = {}
    
    funcname = 'self'
    
    try:
        
        print code
        
        with open('code.py', 'r') as fp:
        
            buf = []
            head = True
    
            code = 'def '+funcname+'('+args+'):\n'
            
            for line in fp.readlines():
                
                code += '    ' + line
    
        exec code in _, _

        ret = _[funcname](**argv)

    except:
        
        from pprint import pprint
        import traceback
        
        mesg = traceback.format_exc()
        
        
        print mesg

main()