在调试时自动输出变量名

来源:互联网 发布:买电脑配件 知乎 编辑:程序博客网 时间:2024/04/30 08:31
在Python中,最好的debug方式是什么?毫无疑问,是print,但是print有时候显得有点笨拙,比如考虑下面的代码:
def f(value):
    a 
= value
    
print a

f(
3)

我们通过print输出a的值,方便debug,但是如果你的代码里有十几个变量需要这样的方式来查看值,那你一定会搞不清谁对应谁,所以,我们还需要什么呢?毫无疑问,变量的名字!

最简单的方法是,在输出变量的同时,输出变量名,比如将“print a"改为"print 'a : ', a”。每一个都要这样写,是不是很不爽?有没有一种方法,可以自动侦测变量名?

有,当然有,实际上,Python代码中所有的信息在编译之后都会被存储起来,这样,你不仅能获得Python运行时的动态信息,而且还能获得Python代码中的静态信息,变量名不就是一静态信息么。只需要利用Python的frame对象和code对象,就能完成一切魔法。

下面在输出变量值时自动获取变量名的一种解决方案:

import sys
import dis
import StringIO

def parse_bytecodes(bufvalue):
    var_names 
= []
    bytecodes 
= bufvalue.split(' ')
    length 
= len(bytecodes)
    var_count 
= 0
    
for i in range(length-1-1-1):
        bytecode 
= bytecodes[i]
       
        
if var_count != 0:
            var_names.append(bytecode.split()[
-1][1:-1])
            var_count 
-= 1
           
        
if '-->' in bytecode:
            var_count 
= int(bytecode.split()[-1])
   
    var_names.reverse()
    
return var_names
   
def print_name(*args):
    
#replace standard stdout with a StringIO object
    buf = StringIO.StringIO()
    old_stdout 
= sys.stdout
    sys.stdout 
= buf
   
    
#call dis.disco to get bytecodes
    f = sys._getframe(1)
    dis.disco(f.f_code, f.f_lasti)
   
    
#restore standard stdout
    sys.stdout = old_stdout   
   
    
#process bytecode
    var_names = parse_bytecodes(buf.getvalue())
    buf.close()
   
    
#print var name and var value
    for i, var_name in enumerate(var_names):
        
print '<%s, %s>' % (var_name, args[i])

class A(object):
    
def __init__(self):
        
pass
       
    
def __str__(self):
        
return 'i am a'
       
def g():
    a 
= 1
    b 
= 2
    c 
= 3
    d 
= A()
    print_name(a, b, c, d)
   
g()

输出的结果为:
<a, 1>
<b, 2>
<c, 3>
<d, i am a>

当然,这种方法并不是万能的,因为在Python中,并不是所有的对象都会对应名字,因为Python有时可以通过位置引用对象,所以名字就不是必须的了,如果将函数g中最后一行改为print_name(a, b, c, A()),那么输出的结果就变为:
<b, 1>
<c, 2>
<A, 3>
<, i am a>