高效判断Python UnionType类型中是否包含特定类型
Python的类型提示系统中，UnionType 代表一个联合类型，可包含多个类型。本文探讨如何有效判断UnionType 是否包含特定类型，尤其是在函数参数类型检查的应用场景。
问题：在检查函数参数类型时，遇到UnionType（例如str | None），无法直接判断其是否包含特定类型，例如str。parameter.annotation 的类型为 types.UnionType，无法使用 in 操作符或迭代访问子类型。
解决方案：主要有两种方法：
立即学习“Python免费学习笔记（深入）”；
方法一：使用 typing.get_args() 函数
typing.get_args() 函数可获取泛型类型的参数。对于 UnionType，它返回一个包含所有子类型的元组。因此，可以使用 str in typing.get_args(parameter.annotation) 来判断 UnionType 是否包含 str 类型。 这种方法直接明了，效率高。
方法二：使用 isinstance() 函数
isinstance() 函数用于判断一个对象是否属于某个类型。我们可以传入一个 str 类型的实例（例如空字符串 ""），并检查它是否属于 parameter.annotation 表示的 UnionType。如果属于，则说明 UnionType 包含 str 类型。 这种方法更依赖于类型的实例化，在处理自定义类型时需要谨慎，确保自定义类型与 isinstance() 函数的判断逻辑兼容。
方法选择:
两种方法都能有效解决问题。typing.get_args() 方法更直接，也更推荐使用，因为它直接检查子类型，而不会受到自定义类型实现的影响。isinstance() 方法则需要考虑自定义类型的兼容性，相对来说不够直接。 在大多数情况下，typing.get_args() 方法是首选。
