Better way to handle no bytes on socket.

Author: BiffoBear

adafruit_wiznet5k/adafruit_wiznet5k.py

@@ -848,19 +848,20 @@ def read_udp(self, socket_num: int, length: int) -> Tuple[int, bytes]:
         bytes_on_socket, bytes_read = 0, b""
         # Parse the UDP packet header.
         header_length, self._pbuff[:8] = self.socket_read(socket_num, 8)
-        if not header_length:
-            return bytes_on_socket, bytes_read
-        if header_length != 8:
-            raise ValueError("Invalid UDP header.")
-        data_length = self._chip_parse_udp_header(socket_num)
-        # Read the UDP packet data.
-        if data_length:
-            if data_length <= length:
-                bytes_on_socket, bytes_read = self.socket_read(socket_num, data_length)
-            else:
-                bytes_on_socket, bytes_read = self.socket_read(socket_num, length)
-                # just consume the rest, it is lost to the higher layers
-                self.socket_read(socket_num, data_length - length)
+        if header_length:
+            if header_length != 8:
+                raise ValueError("Invalid UDP header.")
+            data_length = self._chip_parse_udp_header(socket_num)
+            # Read the UDP packet data.
+            if data_length:
+                if data_length <= length:
+                    bytes_on_socket, bytes_read = self.socket_read(
+                        socket_num, data_length
+                    )
+                else:
+                    bytes_on_socket, bytes_read = self.socket_read(socket_num, length)
+                    # just consume the rest, it is lost to the higher layers
+                    self.socket_read(socket_num, data_length - length)
         return bytes_on_socket, bytes_read
 
     def socket_write(