flatten pydantic models

main.py

class Ipva(BaseModel):
    veiculo: Veiculo
    status: str = Field(alias='STATUS')
    ano: Optional[str] = Field(None, alias='ANO')
    valor: Optional[str] = Field(None, alias='VALOR')
    juros: Optional[str] = Field(None, alias='JUROS')
    multa: Optional[str] = Field(None, alias='MULTA')
    correcao: Optional[str] = Field(None, alias='CORRECAO')
    tse: Optional[str] = Field(None, alias='TSE')
    desconto: Optional[str] = Field(None, alias='DESCONTO')
    total: Optional[str] = Field(None, alias='TOTAL')
    parcelaspagas: Optional[str] = Field(None, alias='PARCELAS_PAGAS')
    totalparcelas: Optional[str] = Field(None, alias='TOTAL_PARCELAS')
    situacao: Optional[str] = Field(None, alias='SITUACAO')

    def flatten(self, use_alias_as_key: bool = False) -> dict:
        'Retorna o modelo em formato de dicionário achatando os dados do veículo'
        ipva_data = self.model_dump(by_alias=use_alias_as_key)
        veiculo_data = ipva_data.pop('veiculo')
        ipva_data.update(veiculo_data)
        return ipva_data