在python

Question

有没有办法缓存python文件句柄？我有一个函数，它将netCDF文件路径作为输入，打开它，从netCDF文件中提取一些数据并关闭它。它被调用了很多次，每次打开文件的开销都很高。

如何通过缓存文件句柄来加快速度？也许有一个python库可以做到这一点

Answer 1

是的，您可以使用以下python库：

• dill（必填）
• python-memcached（可选）

让我们按照这个例子。你有两个文件：

# save.py - it puts deserialized file handler object to memcached
import dill
import memcache            


mc = memcache.Client(['127.0.0.1:11211'], debug=0)
file_handler = open('data.txt', 'r')
mc.set("file_handler", dill.dumps(file_handler))
print 'saved!'   

和

# read_from_file.py - it gets deserialized file handler object from memcached, 
#                     then serializes it and read lines from it
import dill
import memcache


mc = memcache.Client(['127.0.0.1:11211'], debug=0)
file_handler = dill.loads(mc.get("file_handler"))
print file_handler.readlines() 

现在如果你跑：

python save.py
python read_from_file.py

你可以得到你想要的东西。

为什么会有效？

因为您没有关闭文件（file_handler.close()），所以对象仍然存在于内存中（由于weakref而未被垃圾回收）并且您可以使用它。即使在不同的过程中。

<强>解决方案

import dill
import memcache


mc = memcache.Client(['127.0.0.1:11211'], debug=0)
serialized = mc.get("file_handler")
if serialized:
    file_handler = dill.loads(serialized)
else:
    file_handler = open('data.txt', 'r')
    mc.set("file_handler", dill.dumps(file_handler))
print file_handler.readlines() 

Answer 2

这个怎么样？

filehandle = None
def get_filehandle(filename):
    if filehandle is None or filehandle.closed():
        filehandle = open(filename, "r")
    return filehandle

您可能希望将其封装到一个类中，以防止其他代码弄乱filehandle变量。