refactor: use fast return if len(request) < 20

TionAPI · Jul 8, 2022 · 4282e4e · 4282e4e
1 parent aa6a012
commit 4282e4e
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/tion_btle/light_family.py b/tion_btle/light_family.py
@@ -107,16 +107,16 @@ def chunks(lst, n):
 
         if len(request) < 20:
             request[0] = self.SINGLE_PACKET_ID
-            result = [request]
-        else:
-            request[0] = self.FIRST_PACKET_ID
-            result = list(chunks(request, 20))
-
-            for i in range(1, len(result)):
-                if i == len(result)-1:
-                    result[i].insert(0, self.END_PACKET_ID)
-                else:
-                    result[i].insert(0, self.MIDDLE_PACKET_ID)
+            return [request]
+
+        request[0] = self.FIRST_PACKET_ID
+        result = list(chunks(request, 20))
+
+        for i in range(1, len(result)):
+            if i == len(result)-1:
+                result[i].insert(0, self.END_PACKET_ID)
+            else:
+                result[i].insert(0, self.MIDDLE_PACKET_ID)
 
         return result