Python：如何找到打印语句所在的代码位置

一、背景

前两天在研究一个项目时，发现在控制台总是会输出一些不重要的信息，影响体验。于是，就想着在代码里面直接把这些打印信息的地方注释掉。

二、方案一

如果打印的信息（全部或者部分信息，可以根据信息内容推测然后多尝试）是在代码里面写死的字面量，直接通过 IDE 的查找功能就能定位代码位置，然后注释掉即可。

如果这个方案行不通，继续往下看！

三、方案二

如果打印的信息是动态的，即不是在编码阶段产生，而是在编译或运行阶段产生，此时无法通过 方案一 定位代码的位置。那么我们可以这样：

1.在程序入口处添加如下代码：

# 追踪print打印位置
old_f = sys.stdout
class F:
    def write(self, x):
        old_f.write(x.replace("\n", " [%s]\n" % str(traceback.extract_stack())))
sys.stdout = F()

然后运行程序，此时控制台的每一行输出后面都会打印出对应的堆栈信息，我们只需要挨个查看其中的每个文件路径即可（比如 D:\aaa\bbb…）

如果你觉得每一行输出都打印堆栈看起来太多，那么可以这样优化一下：

# 追踪print打印位置
old_f = sys.stdout
class F:
    def write(self, x):
        if x == "你要找的输出信息内容"
            old_f.write(x.replace("\n", " [%s]\n" % str(traceback.extract_stack())))
sys.stdout = F()

这样就只会打印你想找的那一行输出的堆栈信息了。

四、参考资料

https://stackoverflow.com/questions/4883789/adding-a-datetime-stamp-to-python-print