Merge pull request #167 from nickknissen/feat/latest-transaction-sensor

Feat/latest transaction sensor

Author: nickknissen

.vscode/launch.json

@@ -8,10 +8,12 @@
       // Debug by attaching to local Home Assistant server using Remote Python Debugger.
       // See https://www.home-assistant.io/integrations/debugpy/
       "name": "Home Assistant: Attach Local",
-      "type": "python",
+      "type": "debugpy",
       "request": "attach",
-      "port": 5678,
-      "host": "localhost",
+      "connect": {
+        "port": 5678,
+        "host": "localhost",
+      },
       "pathMappings": [
         {
           "localRoot": "${workspaceFolder}",
@@ -23,10 +25,12 @@
       // Debug by attaching to remote Home Assistant server using Remote Python Debugger.
       // See https://www.home-assistant.io/integrations/debugpy/
       "name": "Home Assistant: Attach Remote",
-      "type": "python",
+      "type": "debugpy",
       "request": "attach",
-      "port": 5678,
-      "host": "homeassistant.local",
+      "connect": {
+        "port": 5678,
+        "host": "homeassistant.local",
+      },
       "pathMappings": [
         {
           "localRoot": "${workspaceFolder}",

custom_components/monta/const.py

@@ -12,6 +12,7 @@
 
 ATTR_CHARGE_POINTS = "charge_points"
 ATTR_WALLET = "wallet"
+ATTR_TRANSACTIONS = "transactions"
 
 PREEMPTIVE_REFRESH_TTL_IN_SECONDS = 300
 STORAGE_KEY = "monta_auth"

custom_components/monta/coordinator.py

@@ -10,10 +10,9 @@
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
 
 from .api import MontaApiClient, MontaApiClientAuthenticationError, MontaApiClientError
-from .const import ATTR_CHARGE_POINTS, ATTR_WALLET, DOMAIN, LOGGER
+from .const import ATTR_CHARGE_POINTS, ATTR_TRANSACTIONS, ATTR_WALLET, DOMAIN, LOGGER
 
 
-# https://developers.home-assistant.io/docs/integration_fetching_data#coordinated-single-api-poll-for-data-for-all-entities
 class MontaDataUpdateCoordinator(DataUpdateCoordinator):
     """Class to manage fetching data from the API."""
 
@@ -37,8 +36,15 @@ async def _async_update_data(self):
                 charge_points[charge_point_id][
                     "charges"
                 ] = await self.client.async_get_charges(charge_point_id)
+
             personal_wallet = await self.client.async_get_personal_wallet()
-            return {ATTR_CHARGE_POINTS: charge_points, ATTR_WALLET: personal_wallet}
+            wallet_transactions = await self.client.async_get_wallet_transactions()
+
+            return {
+                ATTR_CHARGE_POINTS: charge_points,
+                ATTR_WALLET: personal_wallet,
+                ATTR_TRANSACTIONS: wallet_transactions,
+            }
         except MontaApiClientAuthenticationError as exception:
             raise ConfigEntryAuthFailed(exception) from exception
         except MontaApiClientError as exception:

custom_components/monta/sensor.py

@@ -27,6 +27,7 @@
 
 from .const import (
     ATTR_CHARGE_POINTS,
+    ATTR_TRANSACTIONS,
     ATTR_WALLET,
     ATTRIBUTION,
     DOMAIN,
@@ -178,6 +179,16 @@ def _parse_date(chargedate: str):
     ),
 )
 
+TRANSACTION_ENTITY_DESCRIPTIONS: tuple[MontaSensorEntityDescription, ...] = (
+    MontaSensorEntityDescription(  # pylint: disable=unexpected-keyword-arg
+        key="monta-latest-wallet-transactions",
+        name="Monta - Latest Wallet Transactions",
+        icon="mdi:wallet-outline",
+        value_fn=lambda data: data,
+        extra_state_attributes_fn=None,
+    ),
+)
+
 
 async def async_setup_entry(
     hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
@@ -197,16 +208,19 @@ async def async_setup_entry(
                 for description in CHARGE_POINT_ENTITY_DESCRIPTIONS
             ]
         )
+
     async_add_entities(
         [
-            MontaWalletSensor(
-                coordinator,
-                entry,
-                description,
-            )
+            MontaWalletSensor(coordinator, entry, description)
             for description in WALLET_ENTITY_DESCRIPTIONS
         ]
     )
+    async_add_entities(
+        [
+            MontaTransactionsSensor(coordinator, entry, description)
+            for description in TRANSACTION_ENTITY_DESCRIPTIONS
+        ]
+    )
 
 
 class MontaChargePointSensor(MontaEntity, SensorEntity):
@@ -276,10 +290,7 @@ def __init__(
     @property
     def native_unit_of_measurement(self) -> str | None:
         """Return the unit of measurement of the sensor."""
-        if (
-            self.entity_description.key == "monta-wallet-amount"
-            and self.coordinator.data
-        ):
+        if self.coordinator.data:
             wallet_data = self.coordinator.data.get(ATTR_WALLET, {})
             if wallet_currency := wallet_data.get("currency"):
                 return wallet_currency.get("identifier", "").upper()
@@ -302,3 +313,54 @@ def extra_state_attributes(self) -> dict[str, str] | None:
                 self.coordinator.data[ATTR_WALLET]
             )
         return None
+
+
+class MontaTransactionsSensor(
+    CoordinatorEntity[MontaDataUpdateCoordinator], SensorEntity
+):
+    """monta Sensor class."""
+
+    _attr_attribution = ATTRIBUTION
+    _attr_has_entity_name = True
+
+    def __init__(
+        self,
+        coordinator: MontaDataUpdateCoordinator,
+        _: ConfigEntry,
+        entity_description: SensorEntityDescription,
+    ) -> None:
+        """Initialize the sensor class."""
+        super().__init__(coordinator)
+
+        self.entity_description = entity_description
+        self._attr_unique_id = generate_entity_id(
+            ENTITY_ID_FORMAT,
+            f"monta_{snake_case(entity_description.key)}",
+            "monta_latest_transactions",
+        )
+
+    @property
+    def native_value(self) -> StateType:
+        """Return the state."""
+        return self.entity_description.value_fn(
+            self.coordinator.data[ATTR_TRANSACTIONS][0]["state"]
+        )
+
+    @property
+    def extra_attributes(self) -> str:
+        """Return extra attributes for the sensor."""
+        return None
+
+    @property
+    def extra_state_attributes(self) -> dict[str, str] | None:
+        """Converts the dates to correct home assitant format."""
+        attributes = {}
+
+        if data := self.coordinator.data.get(ATTR_TRANSACTIONS, []):
+            for transaction in data:
+                for key in WALLET_DATE_KEYS:
+                    if key in transaction:
+                        transaction[key] = _parse_date(transaction[key])
+            attributes["transactions"] = data
+
+        return attributes