Improve part list

datamodel.py

@@ -0,0 +1,164 @@
+"""Implementation of the Datamodel for the parts list with natural sort."""
+
+import logging
+import re
+
+import wx.dataview as dv
+
+from .helpers import loadIconScaled
+
+BOM_COL = 6
+POS_COL = 7
+SIDE_COL = 9
+
+
+class PartListDataModel(dv.PyDataViewModel):
+    """Datamodel for use with the DataViewCtrl of the mainwindow."""
+
+    def __init__(self, scale_factor):
+        super().__init__()
+        self.data = []
+        self.bom_pos_icons = [
+            loadIconScaled(
+                "mdi-check-color.png",
+                scale_factor,
+            ),
+            loadIconScaled(
+                "mdi-close-color.png",
+                scale_factor,
+            ),
+        ]
+        self.side_icons = [
+            loadIconScaled(
+                "TOP.png",
+                scale_factor,
+            ),
+            loadIconScaled(
+                "BOT.png",
+                scale_factor,
+            ),
+        ]
+        self.logger = logging.getLogger(__name__)
+
+    @staticmethod
+    def natural_sort_key(s):
+        """Return a tuple that can be used for natural sorting."""
+        return [
+            int(text) if text.isdigit() else text.lower()
+            for text in re.split("([0-9]+)", s)
+        ]
+
+    def GetColumnCount(self):
+        """Get number of columns."""
+        return 10
+
+    def GetColumnType(self, col):
+        """Get type of each column."""
+        columntypes = (
+            "string",
+            "string",
+            "string",
+            "string",
+            "string",
+            "string",
+            "wxDataViewIconText",
+            "wxDataViewIconText",
+            "string",
+            "wxDataViewIconText",
+        )
+        return columntypes[col]
+
+    def GetChildren(self, parent, children):
+        """Get child items of a parent."""
+        if not parent:
+            for row in self.data:
+                children.append(self.ObjectToItem(row))
+            return len(self.data)
+        return 0
+
+    def IsContainer(self, item):
+        """Check if tem is a container."""
+        return not item
+
+    def GetParent(self, item):
+        """Get parent item."""
+        return dv.NullDataViewItem
+
+    def GetValue(self, item, col):
+        """Get value of an item."""
+        row = self.ItemToObject(item)
+        if col in [BOM_COL, POS_COL, SIDE_COL]:
+            icon = row[col]
+            return dv.DataViewIconText("", icon)
+        return row[col]
+
+    def SetValue(self, value, item, col):
+        """Set value of an item."""
+        row = self.ItemToObject(item)
+        if col in [BOM_COL, POS_COL, SIDE_COL]:
+            return False
+        row[col] = value
+        return True
+
+    def Compare(self, item1, item2, column, ascending):
+        """Override to implement natural sorting."""
+        val1 = self.GetValue(item1, column)
+        val2 = self.GetValue(item2, column)
+
+        key1 = self.natural_sort_key(val1)
+        key2 = self.natural_sort_key(val2)
+
+        if ascending:
+            return (key1 > key2) - (key1 < key2)
+        else:
+            return (key2 > key1) - (key2 < key1)
+
+    def find_index(self, ref):
+        """Get the index of a part within the data list by its reference."""
+        try:
+            return self.data.index([x for x in self.data if x[0] == ref].pop())
+        except (ValueError, IndexError):
+            return None
+
+    def get_bom_pos_icon(self, state: str):
+        """Get an icon for a state."""
+        return self.bom_pos_icons[int(state)]
+
+    def get_side_icon(self, side: str):
+        """Get The side for a layer number."""
+        return self.side_icons[0] if side == "0" else self.side_icons[1]
+
+    def AddEntry(self, data: list):
+        """Add a new entry to the data model."""
+        data[BOM_COL] = self.get_bom_pos_icon(data[BOM_COL])
+        data[POS_COL] = self.get_bom_pos_icon(data[POS_COL])
+        data[SIDE_COL] = self.get_side_icon(data[SIDE_COL])
+        self.data.append(data)
+        self.ItemAdded(dv.NullDataViewItem, self.ObjectToItem(data))
+
+    def UpdateEntry(self, data: list):
+        """Update an entry in the data model."""
+        if (index := self.find_index(data[0])) is None:
+            return
+        item = self.data[index]
+        for i in range(0, len(data)):
+            if i in [BOM_COL, POS_COL]:
+                item[i] = self.get_bom_pos_icon(data[i])
+            elif i == SIDE_COL:
+                item[i] = self.get_side_icon(data[i])
+            else:
+                item[i] = data[i]
+        self.ItemChanged(self.ObjectToItem(item))
+
+    def RemoveEntry(self, ref: str):
+        """Remove an entry from the data model."""
+        if (index := self.find_index(ref)) is None:
+            return
+        item = self.ObjectToItem(self.data[index])
+        self.data.remove(self.data[index])
+        self.ItemDeleted(dv.NullDataViewItem, item)
+
+    def RemoveAll(self):
+        """Remove all entries from the data model."""
+        self.data.clear()
+        self.Cleared()