data() for CheckStateRole should not be cast to int

Author: SteampunkIslande

cutevariant/gui/widgets/choice_button.py

@@ -51,7 +51,7 @@ def data(self, index: QModelIndex, role: Qt.ItemDataRole):
             return QIcon(self._data[index.row()]["icon"])
 
         if role == Qt.CheckStateRole:
-            return int(Qt.Checked) if self._data[index.row()]["checked"] else int(Qt.Unchecked)
+            return Qt.Checked if self._data[index.row()]["checked"] else Qt.Unchecked
 
         if role == Qt.ToolTipRole:
             return self._data[index.row()]["description"]

cutevariant/gui/widgets/choice_widget.py

@@ -28,7 +28,7 @@ def data(self, index: QModelIndex, role: Qt.ItemDataRole):
             return QIcon(self._data[index.row()]["icon"])
 
         if role == Qt.CheckStateRole:
-            return int(Qt.Checked) if self._data[index.row()]["checked"] else int(Qt.Unchecked)
+            return Qt.Checked if self._data[index.row()]["checked"] else Qt.Unchecked
 
         if role == Qt.ToolTipRole:
             return self._data[index.row()]["description"]

cutevariant/gui/widgets/groupby_widget.py

@@ -140,7 +140,7 @@ def data(self, index: QModelIndex, role: int):
                 return int(Qt.AlignmentFlag(Qt.AlignRight | Qt.AlignVCenter))
 
         if role == Qt.CheckStateRole and index.column() == 0 and self.is_checkable:
-            return int(Qt.Checked if self._raw_data[index.row()]["checked"] else Qt.Unchecked)
+            return Qt.Checked if self._raw_data[index.row()]["checked"] else Qt.Unchecked
 
     def clear(self):
         self._set_raw_data([])

cutevariant/gui/widgets/import_widget.py

@@ -126,7 +126,7 @@ def data(self, index: QModelIndex, role: int):
 
         if role == Qt.CheckStateRole:
             if index.column() == self.NAME_COL:
-                return int(Qt.Checked) if item["enabled"] else int(Qt.Unchecked)
+                return Qt.Checked if item["enabled"] else Qt.Unchecked
 
         if role == Qt.ToolTipRole:
             return item["description"]
@@ -553,7 +553,7 @@ def run(self):
                     self.conn,
                     reader,
                     pedfile=self.pedfile,
-                    project = self.project,
+                    project=self.project,
                     import_id=import_id,
                     ignored_fields=self.ignored_fields,
                     indexed_fields=self.indexed_fields,