fix timeout when alternate IPN is used

see sparkmicro#163
T0jan · Jun 23, 2023 · 47c45f2 · 47c45f2
1 parent 322f40f
commit 47c45f2
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/kintree/database/inventree_api.py b/kintree/database/inventree_api.py
@@ -168,14 +168,14 @@ def get_part_from_ipn(part_ipn='') -> int:
     ''' Get Part ID from Part IPN '''
     global inventree_api
 
-    parts = Part.list(inventree_api)
-
-    for part in parts:
-        if part.IPN == part_ipn:
-            return part
-
-    # No part found
-    return None
+    parts = Part.list(inventree_api, IPN=part_ipn)
+
+    if not parts:
+        # No part found
+        return None
+    else:
+        # parts should have only one entry
+        return parts[0]
 
 
 def fetch_part(part_id='', part_ipn='') -> int: