接受python函数中的不同类型？

Question

我有一个Python函数，它在XML文件上做了很多重要的工作。

使用此函数时，我需要两个选项：将XML文件的名称传递给它，或者将其传递给预先解析的ElementTree实例。

我希望函数能够确定它在变量中给出的内容。

例：

def doLotsOfXmlStuff(xmlData):
    if (xmlData != # if xmlData is not ET instance):
        xmlData = ET.parse(xmlData)
    # do a bunch of stuff
    return stuff

调用此函数的应用程序可能只需要调用一次，或者可能需要多次调用它。 多次调用它并每次解析XML都是非常低效和不必要的。 创建一个完整的类只是为了包装这个函数似乎有点矫枉过正，最终需要一些代码重构。 例如：

ourResults = doLotsOfXmlStuff(myObject)

必须成为：

xmlObject = XMLProcessingObjectThatHasOneFunction("data.xml")
ourResult = xmlObject.doLotsOfXmlStuff()

如果我必须在许多小文件上运行它，每次都会创建一个类，这似乎效率低下。

是否有一种简单的方法可以简单地检测变量的类型？ 我知道很多Pythoners会说“你不应该检查”，但这里有一个很好的例子。

在其他强类型语言中，我可以通过方法重载来实现这一点，但这显然不是Pythonic的方式......

Answer 1

这是一种相当普通的模式（例如，接受文件对象或路径的Python函数 ）。 只需使用isinstance ：

def doLotsOfXmlStuff(xmlData):
    if not isinstance(xmlData, ET):
        xmlData = ET.parse(xmlData)
    ...

如果你需要进行清理（例如关闭文件），那么递归调用你的函数就可以了：

def doLotsOfXmlStuff(xmlData):
    if not isinstance(xmlData, ET):
        xmlData = ET.parse(xmlData)
        ret = doLotsOfXmlStuff(xmlData)
        ... # cleanup (or use a context manager)
        return ret
    ...

Answer 2

“鸭子打字”的原则是你不应该太在意对象的具体类型，而应该检查是否支持你感兴趣的API。

换句话说，如果通过xmlData参数传递给您的函数的对象包含一些指示已解析的ElementTree的方法或属性，那么您只需使用这些方法或属性...如果它没有必要的属性那么您然后可以自由地通过一些解析。

那么你想要使用结果ET的函数/方法/属性吗？ 您可以使用hasattr()来检查它。 或者，您可以通过try: ... except AttributeError: block来调用任何此类功能。

我个人认为if not hasattr(...):有点清洁。 （如果它没有我想要的属性，那么将名称重新绑定到已经准备，解析的内容，无论我需要什么）。

这种方法优于isinstance()因为它允许您的功能用户将引用传递给自己的类中的对象，这些对象通过组合而不是继承来扩展ET。 换句话说，如果我在我自己的类中包装一个类似ET的对象，并公开必要的功能，那么我应该能够将引用传递给你的函数，让你只需将我的对象看作是一个“鸭子”，即使它不是鸭子的后代。 如果您需要羽毛，帐单和蹼足，那么只需检查其中一个并尝试使用其余部分。 我可能是一个装有鸭子的黑匣子，我可能已经提供了洞，脚，鸭嘴和羽毛都可以穿过。

Answer 3

您可以使用isinstance来确定变量的类型。

Answer 4

您是否可以尝试使用if语句来检查类型并确定从那里运行的内容？

if type(xmlData).__name__=='ElementTree':
    #do stuff
else: 
    #do some other stuff

Answer 5

我想你可以比较数据类型：

if (xmlData.dtype==something):
    call Function1
else:
    call Function2