Bug fix for wire color lookup

inventree · Aug 6, 2024 · ee1cbdb · ee1cbdb
1 parent 468a164
commit ee1cbdb
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/inventree_wireviz/processing.py b/inventree_wireviz/processing.py
@@ -407,16 +407,19 @@ def match_part(self, line: dict):
 
             if len(wire_data) >= 3:
                 color = wire_data[2]
-
-            wire_pn = f"{pn}-{color}"
+                wire_pn = f"{pn}-{color}"
+            else:
+                wire_pn = pn
 
             # Match wire_pn -> part.IPN
             results = Part.objects.filter(IPN=wire_pn)
+
             if results.count() == 1:
                 return results.first()
 
             # Match wire_pn -> part.name
             results = Part.objects.filter(name=wire_pn)
+
             if results.count() == 1:
                 return results.first()