fix(plant): null ultimaAtualizacao

sunweg/api.py

@@ -1,4 +1,5 @@
 """API Helper."""
+
 import json
 from datetime import datetime
 from typing import Any
@@ -190,7 +191,9 @@ def plant(self, id: int, retry=True) -> Plant | None:
                 total_carbon_saving=result["reduz_carbono_total_number"],
                 last_update=datetime.strptime(
                     result["ultimaAtualizacao"], "%Y-%m-%d %H:%M:%S"
-                ),
+                )
+                if result["ultimaAtualizacao"] is not None
+                else None,
             )
 
             plant.inverters.extend(

sunweg/plant.py

@@ -1,4 +1,5 @@
 """Sunweg API plant."""
+
 from datetime import datetime
 
 from .device import Inverter
@@ -19,7 +20,7 @@ def __init__(
         today_energy_metric: str,
         total_energy: float,
         total_carbon_saving: float,
-        last_update: datetime,
+        last_update: datetime | None,
     ) -> None:
         """
         Initialize Plant.
@@ -45,7 +46,7 @@ def __init__(
         :param total_carbon_saving: total of CO2 saved
         :type total_carbon_saving: float
         :param last_update: when the data was updated
-        :type last_update: datetime
+        :type last_update: datetime | None
         """
         self._id = id
         self._name = name
@@ -161,12 +162,12 @@ def total_carbon_saving(self) -> float:
         return self._total_carbon_saving
 
     @property
-    def last_update(self) -> datetime:
+    def last_update(self) -> datetime | None:
         """
         Get when the plant data was updated.
 
         :return: when the plant data was updated
-        :rtype: datetime
+        :rtype: datetime | None
         """
         return self._last_update
 

sunweg/util.py

@@ -1,4 +1,5 @@
 """Sunweg API util."""
+
 from datetime import date
 from enum import Enum
 
@@ -10,12 +11,14 @@ class Status(Enum):
     WARN = 2
     ERROR = 1
 
+
 class ProductionStats:
     """Energy production statistics"""
-    def __init__(self, date:date, production:float, prognostic:float) -> None:
+
+    def __init__(self, date: date, production: float, prognostic: float) -> None:
         """
         Initialize energy production statistics.
-        
+
         :param date: statistics date
         :type date: date
         :param production: statistics production in kWh
@@ -35,12 +38,12 @@ def date(self) -> date:
     def production(self) -> float:
         """Get energy production in kWh."""
         return self._production
-    
+
     @property
     def prognostic(self) -> float:
         """Get expected energy production in kWh."""
         return self._prognostic
-    
+
     def __str__(self) -> str:
         """Cast Phase to str."""
-        return str(self.__class__) + ": " + str(self.__dict__)
+        return str(self.__class__) + ": " + str(self.__dict__)

tests/responses/plant_success_alt_response.json

@@ -0,0 +1,27 @@
+{
+  "success": true,
+  "usinas": {
+    "id": 16925,
+    "nome": "Plant Name",
+    "inversores": [
+      {
+        "id": 21255,
+        "nome": "Inverter Name",
+        "descricao": "Inverter Description",
+        "esn": "1234ABC",
+        "situacao": 1,
+        "tensaoca": 242,
+        "temperatura": 80
+      }
+    ]
+  },
+  "ultimaAtualizacao": null,
+  "AcumuladoPotencia": "25,23 kW",
+  "energiaGeradaHoje": "1,23 kWh",
+  "energiaacumuladanumber": "23.20",
+  "taxaPerformance": 1.48,
+  "KWHporkWp": "1,2",
+  "PerformanceRate": 0.1,
+  "reduz_carbono_total_number": 0.012296,
+  "economia": 12.786912
+}

tests/test_api.py

@@ -1,4 +1,5 @@
 """Test sunweg.api."""
+
 from datetime import date, datetime
 from os import path
 import os
@@ -191,6 +192,44 @@ def test_plant_success(self) -> None:
                 assert inverter.total_energy_metric == ""
                 assert not inverter.is_complete
 
+    def test_plant_success_alt(self) -> None:
+        """Test plant success."""
+        with patch(
+            "requests.Session.get",
+            return_value=self.responses["plant_success_alt_response.json"],
+        ), patch("sunweg.api.APIHelper.inverter", return_value=INVERTER_MOCK):
+            api = APIHelper("[email protected]", "password")
+            plant = api.plant(16925)
+            assert plant is not None
+            assert plant.id == 16925
+            assert plant.name == "Plant Name"
+            assert plant.total_power == 25.23
+            assert plant.last_update is None
+            assert plant.kwh_per_kwp == 1.2
+            assert plant.performance_rate == 1.48
+            assert plant.saving == 12.786912
+            assert plant.today_energy == 1.23
+            assert plant.today_energy_metric == "kWh"
+            assert plant.total_carbon_saving == 0.012296
+            assert plant.total_energy == 23.2
+            assert plant.__str__().startswith("<class 'sunweg.plant.Plant'>")
+            assert len(plant.inverters) == 1
+            for inverter in plant.inverters:
+                assert inverter.id == 21255
+                assert inverter.name == "Inverter Name"
+                assert inverter.frequency == 0
+                assert inverter.power == 0.0
+                assert inverter.power_metric == ""
+                assert inverter.power_factor == 0.0
+                assert inverter.sn == "1234ABC"
+                assert inverter.status == Status.ERROR
+                assert inverter.temperature == 80
+                assert inverter.today_energy == 0.0
+                assert inverter.today_energy_metric == ""
+                assert inverter.total_energy == 0.0
+                assert inverter.total_energy_metric == ""
+                assert not inverter.is_complete
+
     def test_plant_401(self) -> None:
         """Test plant with expired token."""
         with patch(