如何使用类型提示为参数指定多个类型?

时间:2018-02-09 15:29:52

标签: python python-3.x arguments type-hinting

我有一个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 个答案:

答案 0 :(得分:10)

您想要一个类型union

from typing import Union

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