policyengine-api-simulation-client 0.20251210.56__py3-none-any.whl

policyengine_api_simulation_client/models/cliff_impact.py

@@ -0,0 +1,75 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+if TYPE_CHECKING:
+    from ..models.cliff_impact_in_simulation import CliffImpactInSimulation
+
+
+T = TypeVar("T", bound="CliffImpact")
+
+
+@_attrs_define
+class CliffImpact:
+    """
+    Attributes:
+        baseline (CliffImpactInSimulation):
+        reform (CliffImpactInSimulation):
+    """
+
+    baseline: CliffImpactInSimulation
+    reform: CliffImpactInSimulation
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        baseline = self.baseline.to_dict()
+
+        reform = self.reform.to_dict()
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "baseline": baseline,
+                "reform": reform,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.cliff_impact_in_simulation import CliffImpactInSimulation
+
+        d = dict(src_dict)
+        baseline = CliffImpactInSimulation.from_dict(d.pop("baseline"))
+
+        reform = CliffImpactInSimulation.from_dict(d.pop("reform"))
+
+        cliff_impact = cls(
+            baseline=baseline,
+            reform=reform,
+        )
+
+        cliff_impact.additional_properties = d
+        return cliff_impact
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

policyengine_api_simulation_client/models/cliff_impact_in_simulation.py

@@ -0,0 +1,69 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="CliffImpactInSimulation")
+
+
+@_attrs_define
+class CliffImpactInSimulation:
+    """
+    Attributes:
+        cliff_gap (float):
+        cliff_share (float):
+    """
+
+    cliff_gap: float
+    cliff_share: float
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        cliff_gap = self.cliff_gap
+
+        cliff_share = self.cliff_share
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "cliff_gap": cliff_gap,
+                "cliff_share": cliff_share,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        cliff_gap = d.pop("cliff_gap")
+
+        cliff_share = d.pop("cliff_share")
+
+        cliff_impact_in_simulation = cls(
+            cliff_gap=cliff_gap,
+            cliff_share=cliff_share,
+        )
+
+        cliff_impact_in_simulation.additional_properties = d
+        return cliff_impact_in_simulation
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

policyengine_api_simulation_client/models/decile_impact.py

@@ -0,0 +1,77 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+if TYPE_CHECKING:
+    from ..models.decile_impact_average import DecileImpactAverage
+    from ..models.decile_impact_relative import DecileImpactRelative
+
+
+T = TypeVar("T", bound="DecileImpact")
+
+
+@_attrs_define
+class DecileImpact:
+    """
+    Attributes:
+        relative (DecileImpactRelative):
+        average (DecileImpactAverage):
+    """
+
+    relative: DecileImpactRelative
+    average: DecileImpactAverage
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        relative = self.relative.to_dict()
+
+        average = self.average.to_dict()
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "relative": relative,
+                "average": average,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.decile_impact_average import DecileImpactAverage
+        from ..models.decile_impact_relative import DecileImpactRelative
+
+        d = dict(src_dict)
+        relative = DecileImpactRelative.from_dict(d.pop("relative"))
+
+        average = DecileImpactAverage.from_dict(d.pop("average"))
+
+        decile_impact = cls(
+            relative=relative,
+            average=average,
+        )
+
+        decile_impact.additional_properties = d
+        return decile_impact
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

policyengine_api_simulation_client/models/decile_impact_average.py

@@ -0,0 +1,46 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="DecileImpactAverage")
+
+
+@_attrs_define
+class DecileImpactAverage:
+    """ """
+
+    additional_properties: dict[str, float] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        decile_impact_average = cls()
+
+        decile_impact_average.additional_properties = d
+        return decile_impact_average
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> float:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: float) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

policyengine_api_simulation_client/models/decile_impact_relative.py

@@ -0,0 +1,46 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="DecileImpactRelative")
+
+
+@_attrs_define
+class DecileImpactRelative:
+    """ """
+
+    additional_properties: dict[str, float] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        decile_impact_relative = cls()
+
+        decile_impact_relative.additional_properties = d
+        return decile_impact_relative
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> float:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: float) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

policyengine_api_simulation_client/models/economy_comparison.py

@@ -0,0 +1,338 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, TypeVar, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.budgetary_impact import BudgetaryImpact
+    from ..models.cliff_impact import CliffImpact
+    from ..models.decile_impact import DecileImpact
+    from ..models.economy_comparison_detailed_budget_type_0 import EconomyComparisonDetailedBudgetType0
+    from ..models.inequality_impact import InequalityImpact
+    from ..models.intra_decile_impact import IntraDecileImpact
+    from ..models.intra_wealth_decile_impact_with_values import IntraWealthDecileImpactWithValues
+    from ..models.labor_supply_response import LaborSupplyResponse
+    from ..models.poverty_gender_breakdown import PovertyGenderBreakdown
+    from ..models.poverty_impact import PovertyImpact
+    from ..models.poverty_racial_breakdown_with_values import PovertyRacialBreakdownWithValues
+    from ..models.uk_constituency_breakdown_with_values import UKConstituencyBreakdownWithValues
+    from ..models.wealth_decile_impact_with_values import WealthDecileImpactWithValues
+
+
+T = TypeVar("T", bound="EconomyComparison")
+
+
+@_attrs_define
+class EconomyComparison:
+    """
+    Attributes:
+        budget (BudgetaryImpact):
+        detailed_budget (EconomyComparisonDetailedBudgetType0 | None):
+        decile (DecileImpact):
+        inequality (InequalityImpact):
+        poverty (PovertyImpact):
+        poverty_by_gender (PovertyGenderBreakdown):
+        poverty_by_race (None | PovertyRacialBreakdownWithValues):
+        intra_decile (IntraDecileImpact):
+        wealth_decile (None | WealthDecileImpactWithValues):
+        intra_wealth_decile (IntraWealthDecileImpactWithValues | None):
+        labor_supply_response (LaborSupplyResponse):
+        constituency_impact (None | UKConstituencyBreakdownWithValues):
+        cliff_impact (CliffImpact | None):
+        model_version (None | str | Unset):
+        data_version (None | str | Unset):
+    """
+
+    budget: BudgetaryImpact
+    detailed_budget: EconomyComparisonDetailedBudgetType0 | None
+    decile: DecileImpact
+    inequality: InequalityImpact
+    poverty: PovertyImpact
+    poverty_by_gender: PovertyGenderBreakdown
+    poverty_by_race: None | PovertyRacialBreakdownWithValues
+    intra_decile: IntraDecileImpact
+    wealth_decile: None | WealthDecileImpactWithValues
+    intra_wealth_decile: IntraWealthDecileImpactWithValues | None
+    labor_supply_response: LaborSupplyResponse
+    constituency_impact: None | UKConstituencyBreakdownWithValues
+    cliff_impact: CliffImpact | None
+    model_version: None | str | Unset = UNSET
+    data_version: None | str | Unset = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        from ..models.cliff_impact import CliffImpact
+        from ..models.economy_comparison_detailed_budget_type_0 import EconomyComparisonDetailedBudgetType0
+        from ..models.intra_wealth_decile_impact_with_values import IntraWealthDecileImpactWithValues
+        from ..models.poverty_racial_breakdown_with_values import PovertyRacialBreakdownWithValues
+        from ..models.uk_constituency_breakdown_with_values import UKConstituencyBreakdownWithValues
+        from ..models.wealth_decile_impact_with_values import WealthDecileImpactWithValues
+
+        budget = self.budget.to_dict()
+
+        detailed_budget: dict[str, Any] | None
+        if isinstance(self.detailed_budget, EconomyComparisonDetailedBudgetType0):
+            detailed_budget = self.detailed_budget.to_dict()
+        else:
+            detailed_budget = self.detailed_budget
+
+        decile = self.decile.to_dict()
+
+        inequality = self.inequality.to_dict()
+
+        poverty = self.poverty.to_dict()
+
+        poverty_by_gender = self.poverty_by_gender.to_dict()
+
+        poverty_by_race: dict[str, Any] | None
+        if isinstance(self.poverty_by_race, PovertyRacialBreakdownWithValues):
+            poverty_by_race = self.poverty_by_race.to_dict()
+        else:
+            poverty_by_race = self.poverty_by_race
+
+        intra_decile = self.intra_decile.to_dict()
+
+        wealth_decile: dict[str, Any] | None
+        if isinstance(self.wealth_decile, WealthDecileImpactWithValues):
+            wealth_decile = self.wealth_decile.to_dict()
+        else:
+            wealth_decile = self.wealth_decile
+
+        intra_wealth_decile: dict[str, Any] | None
+        if isinstance(self.intra_wealth_decile, IntraWealthDecileImpactWithValues):
+            intra_wealth_decile = self.intra_wealth_decile.to_dict()
+        else:
+            intra_wealth_decile = self.intra_wealth_decile
+
+        labor_supply_response = self.labor_supply_response.to_dict()
+
+        constituency_impact: dict[str, Any] | None
+        if isinstance(self.constituency_impact, UKConstituencyBreakdownWithValues):
+            constituency_impact = self.constituency_impact.to_dict()
+        else:
+            constituency_impact = self.constituency_impact
+
+        cliff_impact: dict[str, Any] | None
+        if isinstance(self.cliff_impact, CliffImpact):
+            cliff_impact = self.cliff_impact.to_dict()
+        else:
+            cliff_impact = self.cliff_impact
+
+        model_version: None | str | Unset
+        if isinstance(self.model_version, Unset):
+            model_version = UNSET
+        else:
+            model_version = self.model_version
+
+        data_version: None | str | Unset
+        if isinstance(self.data_version, Unset):
+            data_version = UNSET
+        else:
+            data_version = self.data_version
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "budget": budget,
+                "detailed_budget": detailed_budget,
+                "decile": decile,
+                "inequality": inequality,
+                "poverty": poverty,
+                "poverty_by_gender": poverty_by_gender,
+                "poverty_by_race": poverty_by_race,
+                "intra_decile": intra_decile,
+                "wealth_decile": wealth_decile,
+                "intra_wealth_decile": intra_wealth_decile,
+                "labor_supply_response": labor_supply_response,
+                "constituency_impact": constituency_impact,
+                "cliff_impact": cliff_impact,
+            }
+        )
+        if model_version is not UNSET:
+            field_dict["model_version"] = model_version
+        if data_version is not UNSET:
+            field_dict["data_version"] = data_version
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.budgetary_impact import BudgetaryImpact
+        from ..models.cliff_impact import CliffImpact
+        from ..models.decile_impact import DecileImpact
+        from ..models.economy_comparison_detailed_budget_type_0 import EconomyComparisonDetailedBudgetType0
+        from ..models.inequality_impact import InequalityImpact
+        from ..models.intra_decile_impact import IntraDecileImpact
+        from ..models.intra_wealth_decile_impact_with_values import IntraWealthDecileImpactWithValues
+        from ..models.labor_supply_response import LaborSupplyResponse
+        from ..models.poverty_gender_breakdown import PovertyGenderBreakdown
+        from ..models.poverty_impact import PovertyImpact
+        from ..models.poverty_racial_breakdown_with_values import PovertyRacialBreakdownWithValues
+        from ..models.uk_constituency_breakdown_with_values import UKConstituencyBreakdownWithValues
+        from ..models.wealth_decile_impact_with_values import WealthDecileImpactWithValues
+
+        d = dict(src_dict)
+        budget = BudgetaryImpact.from_dict(d.pop("budget"))
+
+        def _parse_detailed_budget(data: object) -> EconomyComparisonDetailedBudgetType0 | None:
+            if data is None:
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                detailed_budget_type_0 = EconomyComparisonDetailedBudgetType0.from_dict(data)
+
+                return detailed_budget_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(EconomyComparisonDetailedBudgetType0 | None, data)
+
+        detailed_budget = _parse_detailed_budget(d.pop("detailed_budget"))
+
+        decile = DecileImpact.from_dict(d.pop("decile"))
+
+        inequality = InequalityImpact.from_dict(d.pop("inequality"))
+
+        poverty = PovertyImpact.from_dict(d.pop("poverty"))
+
+        poverty_by_gender = PovertyGenderBreakdown.from_dict(d.pop("poverty_by_gender"))
+
+        def _parse_poverty_by_race(data: object) -> None | PovertyRacialBreakdownWithValues:
+            if data is None:
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                poverty_by_race_type_0 = PovertyRacialBreakdownWithValues.from_dict(data)
+
+                return poverty_by_race_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(None | PovertyRacialBreakdownWithValues, data)
+
+        poverty_by_race = _parse_poverty_by_race(d.pop("poverty_by_race"))
+
+        intra_decile = IntraDecileImpact.from_dict(d.pop("intra_decile"))
+
+        def _parse_wealth_decile(data: object) -> None | WealthDecileImpactWithValues:
+            if data is None:
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                wealth_decile_type_0 = WealthDecileImpactWithValues.from_dict(data)
+
+                return wealth_decile_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(None | WealthDecileImpactWithValues, data)
+
+        wealth_decile = _parse_wealth_decile(d.pop("wealth_decile"))
+
+        def _parse_intra_wealth_decile(data: object) -> IntraWealthDecileImpactWithValues | None:
+            if data is None:
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                intra_wealth_decile_type_0 = IntraWealthDecileImpactWithValues.from_dict(data)
+
+                return intra_wealth_decile_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(IntraWealthDecileImpactWithValues | None, data)
+
+        intra_wealth_decile = _parse_intra_wealth_decile(d.pop("intra_wealth_decile"))
+
+        labor_supply_response = LaborSupplyResponse.from_dict(d.pop("labor_supply_response"))
+
+        def _parse_constituency_impact(data: object) -> None | UKConstituencyBreakdownWithValues:
+            if data is None:
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                constituency_impact_type_0 = UKConstituencyBreakdownWithValues.from_dict(data)
+
+                return constituency_impact_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(None | UKConstituencyBreakdownWithValues, data)
+
+        constituency_impact = _parse_constituency_impact(d.pop("constituency_impact"))
+
+        def _parse_cliff_impact(data: object) -> CliffImpact | None:
+            if data is None:
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                cliff_impact_type_0 = CliffImpact.from_dict(data)
+
+                return cliff_impact_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(CliffImpact | None, data)
+
+        cliff_impact = _parse_cliff_impact(d.pop("cliff_impact"))
+
+        def _parse_model_version(data: object) -> None | str | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(None | str | Unset, data)
+
+        model_version = _parse_model_version(d.pop("model_version", UNSET))
+
+        def _parse_data_version(data: object) -> None | str | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(None | str | Unset, data)
+
+        data_version = _parse_data_version(d.pop("data_version", UNSET))
+
+        economy_comparison = cls(
+            budget=budget,
+            detailed_budget=detailed_budget,
+            decile=decile,
+            inequality=inequality,
+            poverty=poverty,
+            poverty_by_gender=poverty_by_gender,
+            poverty_by_race=poverty_by_race,
+            intra_decile=intra_decile,
+            wealth_decile=wealth_decile,
+            intra_wealth_decile=intra_wealth_decile,
+            labor_supply_response=labor_supply_response,
+            constituency_impact=constituency_impact,
+            cliff_impact=cliff_impact,
+            model_version=model_version,
+            data_version=data_version,
+        )
+
+        economy_comparison.additional_properties = d
+        return economy_comparison
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties