我有一个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):
# ...
答案 0 :(得分:10)
您想要一个类型union:
from typing import Union
def post_xml(data: Union[str, ET.Element]):
...