CRAI nitpicks

Author: bouwew

plugwise/helper.py

@@ -159,8 +159,8 @@ def _all_appliances(self) -> None:
     def _get_p1_smartmeter_info(self) -> None:
         """For P1 collect the connected SmartMeter info from the Home/building location.
 
-        Note: For P1, the entity_id for the gateway and smartmeter are
-        switched to maintain backward compatibility with existing implementations.
+        Note: For P1, the entity_id for the gateway and smartmeter are switched to maintain
+        backward compatibility. For Anna P1, the smartmeter uses the home location_id directly.
         """
         appl = Munch()
         locator = MODULE_LOCATOR
@@ -327,9 +327,10 @@ def _get_measurement_data(self, entity_id: str) -> GwEntityData:
         data: GwEntityData = {"binary_sensors": {}, "sensors": {}, "switches": {}}
         # Get P1 smartmeter data from LOCATIONS
         entity = self.gw_entities[entity_id]
-        dev_class = entity.get("dev_class")
         smile_is_power = self.smile.type == "power"
-        if (smile_is_power or self.smile.anna_p1) and dev_class == "smartmeter":
+        if (smile_is_power or self.smile.anna_p1) and entity.get(
+            "dev_class"
+        ) == "smartmeter":
             data.update(self._power_data_from_location())
 
         if smile_is_power and not self.smile.anna_p1: