Python_解决脚本执行过程中，文件被多次读取的问题

今天在封装pandas过程中，发现封装脚本的执行耗时明显高于未封装的脚本

复盘问题

import time


class Demo:

    def mock_read_excel(self):
        print("读取文件")
        time.sleep(5)
        return "excel数据"

    def mock_use_1(self):
        self.mock_read_excel()
        return "指标1"

    def mock_use_2(self):
        self.mock_read_excel()
        return "指标2"

    def mock_use_3(self):
        self.mock_read_excel()
        return "指标3"


if __name__ == '__main__':
    start_time = time.time()
    demo = Demo()
    demo.mock_use_1()
    demo.mock_use_2()
    demo.mock_use_3()
    print("耗时", time.time() - start_time)

代码是模拟 读取文件方法被多次调用的情况，执行后，结果如下：

问题原因

每次调用 读取文件方法 都会去读取一次文件，所以导致封装的脚本耗时明显高于未封装的脚本

解决方法

python实例化有个特性，__init__方法中定义的实例属性如果赋值对象是方法(包括被 property装饰的方法)，在实例化时，会调用一次赋值的方法，指针会指向方法return的值，当属性被多次调用时，不会多次执行赋值的方法。

利用该特性 将读取excel的结果赋值给实例属性，多次调用该实例属性也不会再对excel进行读取操作。

import time


class Demo:

    def __init__(self):
        self.excel_info = self.mock_read_excel()
        print("实例化完成")

    def mock_read_excel(self):
        print("读取文件")
        time.sleep(5)
        return "excel数据"

    def mock_use_1(self):
        a = self.excel_info
        return "指标1"

    def mock_use_2(self):
        b = self.excel_info
        return "指标2"

    def mock_use_3(self):
        c = self.excel_info
        return "指标3"


if __name__ == '__main__':
    start_time = time.time()
    demo = Demo()
    demo.mock_use_1()
    demo.mock_use_2()
    demo.mock_use_3()
    print("耗时", time.time() - start_time)

通过执行结果可以看出问题得到解决。

总结

对于 执行比较耗时的方法 且方法的返回值是个不变的对象 都可以采用这种方法来解决问题。