如何为可以是 class A 或 class B 的参数编写类型提示？

Question

我想将 object 作为 function 的参数值。 anything_object对象都可以是 A class 或 B class。

class A:
    name: str

class B:
    phone: int

class ObjectInfo
    def __init__(self) -> None:
        pass
    
    def getObjectFieldName(self, anything_object: ?) -> list:
        return dir(anything_object)

Answer 1

您可以使用Union指定多种类型提示（对于两者，arguments 和返回类型）。 您必须从标准库typing中导入Union 。

from typing import Union

class A:
    def __init__(self, name) -> None:
        self.name = name

class B:
    def __init__(self, name) -> None:
        self.name = name 


class ObjectInfo:
    def __init__(self) -> None:
        pass 

    def method(self, object:Union[A, B]=None) -> list:
        return dir(object)

obj = ObjectInfo()
print(obj.method())

[外部链接]
更多关于类型提示： https://docs.python.org/3/library/typing.html