Python Cookbook-2.15 用类文件对象适配真实文件对象
任务
需要传递一个类似文件的对象(比如,调用urllib.urlopen 返回的结果)给一个函数或者方法,但这个函数或方法要求只接受真实的文件对象(比如,像marshalload 这样的函数)。
解决方案
为了过类型检查这一关,我们需要将类文件对象中的所有数据写入到磁盘中的一个临时文件。然后使用临时文件的(真实)文件对象。下面给出一个实现这个想法的函数:
import types, tempfile
CHUNK SIZE = 16*1024
def adapt_file(fileObj):
if isinstance(fileObj,file): return fileObj
tmpFileObj = tempfile.TemporaryFile
while True:
data = fileObj.read(CHUNK_SIE)
if not data: break
tmpFileObi.write(data)
fileObj.close()
tmpFileObj.seek(0)
return tmpFileObj
讨论
本节展示的其实是设计模式中的适配器(即Adapter,比如你想要X,我却给你Y以替换 X)的 Python 风格的实现。虽然设计模式通常被认为是一种面向对象的设计方式,因此一般需要用类来实现,但具体实现并没有什么限制。比如此例中,我们根本不需要引人任何新类,因为 adapt file 函数已经足够了。这里我们遵守奥卡姆剃刀原理(译者注:Occam’sRazor,奥卡姆是14世纪的一个逻辑学家和天主教修道士,奥卡姆剃刀原理即“如无必要,勿增实体”),不在没有必要的情况下引入任何实体。
当需要依赖一些底层的、要求精确类型的工具时,应该首先考虑适配,而不是类型检查。当获得一个适合的可以绕过类型检查的对象时,应该考虑将其配接成需要的对象。用这种方式,你的代码会更加灵活,也更具复用性。