set_columns, handle case with zero columns while setting a new db (#337)

European-XFEL · Sep 17, 2024 · c36f975 · c36f975
1 parent 687ffde
commit c36f975
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/damnit/gui/table.py b/damnit/gui/table.py
@@ -214,9 +214,15 @@ def set_columns(self, columns, statuses):
                                                   QtWidgets.QSizePolicy.Minimum)
 
         # Remove the static columns
-        columns, statuses = map(list, zip(*[x for x in zip(columns, statuses)
-                                            if x[0] not in static_columns]))
-        self.add_new_columns(columns, statuses)
+        new_columns = []
+        new_statuses = []
+        for col, status in zip(columns, statuses):
+            if col in static_columns:
+                continue
+            new_columns.append(col)
+            new_statuses.append(status)
+
+        self.add_new_columns(new_columns, new_statuses)
 
     def get_column_states(self):
         column_states = { }