Add more data grabs

Author: rickboy

custom_components/sigenergy_cloud/api.py

@@ -18,6 +18,7 @@
     DEFAULT_REGION,
     ENDPOINT_AUTH,
     ENDPOINT_CURRENT_LOCAL_WEATHER,
+    ENDPOINT_ENERGY_CUSTOM,
     ENDPOINT_ENERGY_FLOW,
     ENDPOINT_MODE_CURRENT,
     ENDPOINT_MODE_SET,
@@ -26,6 +27,7 @@
     ENDPOINT_SMART_LOADS,
     ENDPOINT_STATION_HOME,
     ENDPOINT_SYSTEM_DEVICES,
+    ENDPOINT_TARIFF_SOC_DAY,
     OAUTH_CLIENT_ID,
     OAUTH_CLIENT_SECRET,
 )
@@ -322,6 +324,46 @@ async def get_current_local_weather(self, station_sn_code: str) -> dict[str, Any
             params={"stationSnCode": station_sn_code},
         )
 
+    async def get_energy_custom(
+        self,
+        station_id: str,
+        start_date: str,
+        end_date: str,
+        *,
+        date_flag: int = 1,
+        resource_ids: str = "energy_card",
+    ) -> dict[str, Any]:
+        """Get custom energy statistics for a station and date range."""
+        return await self._request(
+            "GET",
+            ENDPOINT_ENERGY_CUSTOM,
+            params={
+                "stationId": station_id,
+                "startDate": start_date,
+                "endDate": end_date,
+                "dateFlag": date_flag,
+                "resourceIds": resource_ids,
+            },
+        )
+
+    async def get_tariff_soc_day(
+        self,
+        station_id: str,
+        dt: str,
+        *,
+        need_prediction: bool = False,
+    ) -> dict[str, Any]:
+        """Get daily tariff and state-of-charge details for a station."""
+        return await self._request(
+            "GET",
+            ENDPOINT_TARIFF_SOC_DAY,
+            params={
+                "stationId": station_id,
+                "dt": dt,
+                "needPrediction": str(need_prediction).lower(),
+            },
+        )
+
     # ------------------------------------------------------------------
     # Convenience: fetch all data in one coordinator cycle
     # ------------------------------------------------------------------

custom_components/sigenergy_cloud/const.py

@@ -44,6 +44,8 @@
 ENDPOINT_REALTIME_CONSUMPTION = (
     "data-process/sigen/station/statistics/real-time-consumption"
 )
+ENDPOINT_ENERGY_CUSTOM = "data-process/sigen/station/statistics/v1/energy/custom"
+ENDPOINT_TARIFF_SOC_DAY = "data-process/sigen/station/statistics/tariff-soc/day"
 
 # Operational modes
 OPERATIONAL_MODES: dict[int, str] = {

tests/test_api.py

@@ -361,6 +361,60 @@ async def test_get_current_local_weather(self, mock_session):
         call_kwargs = mock_session.request.call_args[1]
         assert call_kwargs["params"] == {"stationSnCode": "102026031800194"}
 
+    @pytest.mark.asyncio
+    async def test_get_energy_custom(self, mock_session):
+        client = await self._make_authed_client(mock_session)
+
+        energy_data = {"summary": {"pvDayNrg": 15.58}}
+        api_resp = make_mock_response(
+            200,
+            json_data={"code": 0, "data": energy_data, "message": "success"},
+        )
+        mock_session.request.return_value = api_resp
+
+        result = await client.get_energy_custom(
+            "102026031800194",
+            "20260402",
+            "20260402",
+            date_flag=1,
+            resource_ids="energy_card",
+        )
+
+        assert result == energy_data
+        call_kwargs = mock_session.request.call_args[1]
+        assert call_kwargs["params"] == {
+            "stationId": "102026031800194",
+            "startDate": "20260402",
+            "endDate": "20260402",
+            "dateFlag": 1,
+            "resourceIds": "energy_card",
+        }
+
+    @pytest.mark.asyncio
+    async def test_get_tariff_soc_day(self, mock_session):
+        client = await self._make_authed_client(mock_session)
+
+        tariff_data = {"soc": [], "tariff": []}
+        api_resp = make_mock_response(
+            200,
+            json_data={"code": 0, "data": tariff_data, "message": "success"},
+        )
+        mock_session.request.return_value = api_resp
+
+        result = await client.get_tariff_soc_day(
+            "102026031800194",
+            "20260402",
+            need_prediction=False,
+        )
+
+        assert result == tariff_data
+        call_kwargs = mock_session.request.call_args[1]
+        assert call_kwargs["params"] == {
+            "stationId": "102026031800194",
+            "dt": "20260402",
+            "needPrediction": "false",
+        }
+
     @pytest.mark.asyncio
     async def test_set_operational_mode(self, mock_session):
         client = await self._make_authed_client(mock_session)