model_template.py

from typing import List, Union

from pydantic_xml import BaseXmlModel, attr, element


# [model-start]
class VehicleTemplate(BaseXmlModel):
    drivers: int = attr()
    title: str = attr()
    engine: str = element()


class Car(VehicleTemplate, tag='car'):
    class Config:
        fields = {
            "title": {"alias": "make"},
            "engine": {"alias": "motor"},
        }


class Airplane(VehicleTemplate, tag='airplane'):
    class Config:
        fields = {
            "drivers": {"alias": "pilots"},
            "title": {"alias": "model"},
        }


class Vehicles(BaseXmlModel, tag='vehicles'):
    items: List[Union[Car, Airplane]] = element()
# [model-end]


# [xml-start]
xml_doc = '''
<vehicles>
    <car make="Ford Mustang" drivers="1">
        <motor>Coyote V8</motor>
    </car>
    <airplane model="Bombardier Global 7500" pilots="2">
        <engine>General Electric Passport</engine>
    </airplane>
    <car make="Audi A8" drivers="1">
        <motor>V8</motor>
    </car>
</vehicles>
'''  # [xml-end]

# [json-start]
json_doc = '''
{
    "items": [
        {
            "make": "Ford Mustang",
            "drivers": 1,
            "motor": "Coyote V8"
        },
        {
            "model": "Bombardier Global 7500",
            "pilots": 2,
            "engine": "General Electric Passport"
        },
        {
            "make": "Audi A8",
            "drivers": 1,
            "motor": "V8"
        }
    ]
}
'''  # [json-end]

vehicles = Vehicles.from_xml(xml_doc)
assert vehicles == Vehicles.parse_raw(json_doc)