Add by alias flag to addon to dict (#13)

Author: mdegat01

aiohasupervisor/models/addons.py

@@ -6,6 +6,7 @@
 from typing import Any
 
 from mashumaro import field_options
+from mashumaro.config import TO_DICT_ADD_BY_ALIAS_FLAG, BaseConfig
 
 from .base import DEFAULT, Options, Request, RequestConfig, ResponseData
 
@@ -151,6 +152,11 @@ class AddonInfoStoreExtFields(ABC):
         metadata=field_options(alias="hassio_role"),
     )
 
+    class Config(BaseConfig):
+        """Mashumaro config options."""
+
+        code_generation_options = [TO_DICT_ADD_BY_ALIAS_FLAG]  # noqa: RUF012
+
 
 @dataclass(frozen=True)
 class AddonInfoStoreExtInstalledBaseFields(ABC):

tests/test_addons.py

@@ -12,8 +12,11 @@
     AddonState,
     AddonsUninstall,
     Capability,
+    InstalledAddonComplete,
+    StoreAddonComplete,
     SupervisorRole,
 )
+from aiohasupervisor.models.base import Response
 
 from . import load_fixture
 from .const import SUPERVISOR_URL
@@ -234,3 +237,20 @@ async def test_addons_stats(
     assert stats.cpu_percent == 0
     assert stats.memory_usage == 24588288
     assert stats.network_rx == 1717120021
+
+
+async def test_addons_serialize_by_alias() -> None:
+    """Test serializing addons by alias."""
+    response = Response.from_json(load_fixture("store_addon_info.json"))
+    store_addon_info = StoreAddonComplete.from_dict(response.data)
+
+    assert store_addon_info.supervisor_api is False
+    assert (store_addon_info.to_dict())["supervisor_api"] is False
+    assert (store_addon_info.to_dict(by_alias=True))["hassio_api"] is False
+
+    response = Response.from_json(load_fixture("addons_info.json"))
+    addon_info = InstalledAddonComplete.from_dict(response.data)
+
+    assert addon_info.supervisor_api is True
+    assert (addon_info.to_dict())["supervisor_api"] is True
+    assert (addon_info.to_dict(by_alias=True))["hassio_api"] is True