如何使用类型提示为参数指定多种类型？ [复制]

Posted 2023-02-24

【中文标题】如何使用类型提示为参数指定多种类型？ [复制]

【英文标题】：How do I specify multiple types for a parameter using type-hints? [duplicate]

【发布时间】：2018-07-20 10:00:38

【问题描述】：

我有一个 Python 函数，它接受 XML 数据作为 str。

为方便起见，该函数还会检查xml.etree.ElementTree.Element，并在必要时自动转换为str。

import xml.etree.ElementTree as ET

def post_xml(data: str):
    if type(data) is ET.Element:
        data = ET.tostring(data).decode()
    # ...

是否可以使用类型提示指定参数可以作为两种类型之一给出？

def post_xml(data: str or ET.Element):
    # ...

【参考方案1】：

你想要一个类型union：

from typing import Union

def post_xml(data: Union[str, ET.Element]):
    ...

【讨论】：

Pycharm的参数信息将Union[str, ET.Element]解释为Union[str, Element, Element]。