Use walrus operator

Author: brefra

plugwise_usb/network/__init__.py

@@ -223,8 +223,7 @@ async def node_awake_message(self, response: PlugwiseResponse) -> bool:
                 await self._notify_node_event_subscribers(NodeEvent.AWAKE, mac)
             self._awake_discovery[mac] = response.timestamp
             return True
-        address = self._register.network_address(mac)
-        if address is None:
+        if (address := self._register.network_address(mac)) is None:
             if self._register.scan_completed:
                 return True
             _LOGGER.debug(
@@ -540,7 +539,7 @@ async def allow_join_requests(self, state: bool) -> None:
         """Enable or disable Plugwise network."""
         request = CirclePlusAllowJoiningRequest(self._controller.send, state)
         response = await request.send()
-        if response is None:
+        if (response := await request.send()) is None:
             raise NodeError("No response to get notifications for join request.")
         if response.response_type != NodeResponseType.JOIN_ACCEPTED:
             raise MessageError(

plugwise_usb/nodes/circle.py

@@ -185,7 +185,7 @@ async def calibration_update(self) -> bool:
         )
         request = EnergyCalibrationRequest(self._send, self._mac_in_bytes)
         calibration_response = await request.send()
-        if calibration_response is None:
+        if (calibration_response := await request.send()) is None:
             _LOGGER.warning(
                 "Retrieving energy calibration information for %s failed",
                 self.name,
@@ -494,7 +494,7 @@ async def energy_log_update(self, address: int | None) -> bool:
         )
         request = CircleEnergyLogsRequest(self._send, self._mac_in_bytes, address)
         response = await request.send()
-        if response is None:
+        if (response := await request.send()) is None:
             _LOGGER.debug(
                 "Retrieving of energy log at address %s for node %s failed",
                 str(address),
@@ -529,7 +529,7 @@ async def energy_log_update(self, address: int | None) -> bool:
     async def _energy_log_records_load_from_cache(self) -> bool:
         """Load energy_log_record from cache."""
         cache_data = self._get_cache(CACHE_ENERGY_COLLECTION)
-        if cache_data is None:
+        if (cache_data := self._get_cache(CACHE_ENERGY_COLLECTION)) is None:
             _LOGGER.debug(
                 "Failed to restore energy log records from cache for node %s", self.name
             )
@@ -730,8 +730,7 @@ async def clock_synchronize(self) -> bool:
             self._node_protocols.max,
         )
         node_response: NodeResponse | None = await set_clock_request.send()
-
-        if node_response is None:
+        if (node_response := await set_clock_request.send()) is None:
             _LOGGER.warning(
                 "Failed to (re)set the internal clock of %s",
                 self.name,

plugwise_usb/nodes/circle_plus.py

@@ -84,7 +84,7 @@ async def clock_synchronize(self) -> bool:
             self._send, self._mac_in_bytes
         )
         clock_response = await clock_request.send()
-        if clock_response is None:
+        if (clock_response := await clock_request.send()) is None:
             _LOGGER.debug(
                 "No response for async_realtime_clock_synchronize() for %s", self.mac
             )

plugwise_usb/nodes/sed.py

@@ -685,7 +685,7 @@ async def sed_configure( #pylint: disable=too-many-arguments
             sleep_duration,
         )
         response = await request.send()
-        if response is None:
+        if (response := await request.send()) is None:
             self._new_battery_config = BatteryConfig()
             _LOGGER.warning(
                 "No response from %s to configure sleep settings request", self.name