Python Cookbook-2.15 用类文件对象适配真实文件对象

我不会编程5552025-03-03 15:22

任务

需要传递一个类似文件的对象(比如，调用urllib.urlopen 返回的结果)给一个函数或者方法，但这个函数或方法要求只接受真实的文件对象(比如，像marshalload 这样的函数)。

解决方案

为了过类型检查这一关，我们需要将类文件对象中的所有数据写入到磁盘中的一个临时文件。然后使用临时文件的(真实)文件对象。下面给出一个实现这个想法的函数:

python 复制代码

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世纪的一个逻辑学家和天主教修道士，奥卡姆剃刀原理即"如无必要，勿增实体")，不在没有必要的情况下引入任何实体。

当需要依赖一些底层的、要求精确类型的工具时，应该首先考虑适配，而不是类型检查。当获得一个适合的可以绕过类型检查的对象时，应该考虑将其配接成需要的对象。用这种方式，你的代码会更加灵活，也更具复用性。