Styling for cells, insert row custom number of rows

Author: darginmathi

core/table_controller.py

@@ -105,6 +105,7 @@ def clear(self, model, selection):
         self.undo_stack.endMacro()
 
     def generate_test_cases(self, model):
+        test_case_generated = 0
         self.undo_stack.beginMacro("Generate Test Cases")
         for row in range(model.rowCount()):
             steps_performed_index = model.index(row, TestScenarioModel.STEPS_COL)
@@ -138,9 +139,11 @@ def generate_test_cases(self, model):
                     if steps_performed_text != old_steps_performed:
                         command_obj = EditCellCommand(model, steps_performed_index, steps_performed_text, old_steps_performed)
                         self.undo_stack.push(command_obj)
+                        test_case_generated += 1
 
                     old_expected_result = model.data(expected_result_index, Qt.ItemDataRole.EditRole)
                     if expected_result_text != old_expected_result:
                         command_obj = EditCellCommand(model, expected_result_index, expected_result_text, old_expected_result)
                         self.undo_stack.push(command_obj)
         self.undo_stack.endMacro()
+        return test_case_generated

ui/delegates.py

@@ -186,32 +186,72 @@ def _get_objects(self):
         return self.cached_objects
 
     def paint(self, painter, option, index):
-        if index.row() == 0:
-            super().paint(painter, option, index)
-            return
-
+        model = index.model()
+        current_row = index.row()
+        current_col = index.column()
+        total_rows = model.rowCount()
         option = QStyleOptionViewItem(option)
+        is_special_row = False
+
+        if isinstance(model, TestScenarioModel):
+            if (current_row == 0 or
+                current_row == 1 or
+                current_row == total_rows - 1 or
+                current_row == total_rows - 2):
+                option.font.setBold(True)
+                is_special_row = True
+            elif current_col in [TestScenarioModel.COMMAND_COL, TestScenarioModel.DESC_COL, TestScenarioModel.SKIP_COL]:
+                option.font.setBold(True)
+            if current_col == TestScenarioModel.STEPS_COL:
+                value = model.data(index, Qt.ItemDataRole.DisplayRole)
+                if value in ["Scenario Started", "Scenario Ended"]:
+                    option.font.setBold(True)
+
+        elif isinstance(model, ObjRepoModel):
+            if (current_row == 0 or
+                current_row == total_rows - 1):
+                option.font.setBold(True)
+                is_special_row = True
 
         final_color = option.palette.color(QPalette.ColorRole.Text)
 
-        model = index.model()
-        if isinstance(model, TestScenarioModel):
-            col = index.column()
+        if not is_special_row and isinstance(model, TestScenarioModel):
             value = index.data(Qt.ItemDataRole.DisplayRole)
 
             if value:
-                if col == TestScenarioModel.COMMAND_COL:
+                if current_col == TestScenarioModel.COMMAND_COL:
                     if value not in self.commands:
                         final_color = QColor("#E57373")
 
-                elif TestScenarioModel.DATA1_COL <= col <= TestScenarioModel.DATA5_COL:
+                elif TestScenarioModel.DATA1_COL <= current_col <= TestScenarioModel.DATA5_COL:
                     objects = self._get_objects()
                     if value not in objects:
                         final_color = QColor("#64B5F6")
 
         option.palette.setColor(QPalette.ColorRole.Text, final_color)
         option.palette.setColor(QPalette.ColorRole.HighlightedText, final_color)
 
+        # if not is_special_row and isinstance(model, TestScenarioModel):
+        #     command_index = model.index(current_row, TestScenarioModel.COMMAND_COL)
+        #     command_value = model.data(command_index, Qt.ItemDataRole.DisplayRole)
+
+        #     if command_value == "StartScenario":
+        #         painter.save()
+        #         pen = painter.pen()
+        #         pen.setColor(QColor("#B0B0B0"))
+        #         pen.setWidth(1)
+        #         painter.setPen(pen)
+        #         painter.drawLine(option.rect.topLeft(), option.rect.topRight())
+        #         painter.restore()
+        #     if command_value == "EndScenario":
+        #         painter.save()
+        #         pen = painter.pen()
+        #         pen.setColor(QColor("#B0B0B0"))
+        #         pen.setWidth(1)
+        #         painter.setPen(pen)
+        #         painter.drawLine(option.rect.bottomLeft(), option.rect.bottomRight())
+        #         painter.restore()
+
         super().paint(painter, option, index)
 
 

ui/menu.py

@@ -78,7 +78,13 @@ def generate_test_cases_action(self):
         if current_tab and hasattr(current_tab, 'controller'):
             current_table = self.main.get_current_table()
             if current_table:
-                current_tab.controller.generate_test_cases(current_table.model)
+                num_generated = current_tab.controller.generate_test_cases(current_table.model)
+                if num_generated > 0:
+                    message = f"Generated {num_generated} test case(s)."
+                    self.main.show_status_message(message, "success", 5000)
+                else:
+                    message = f"No empty fields found to generate."
+                    self.main.show_status_message(message, "info", 5000)
 
     def auto_adjust_cells(self):
         table = self.main.get_current_table()

ui/styles.py

@@ -60,7 +60,7 @@ def get_stylesheet(theme_name="dark"):
     return f"""
     /* Global */
     QWidget {{
-        font-family: "Segoe UI", "Calibri", "Arial";
+        font-family "Courier New", "Calibri", "Arial", sans-serif;
         font-size: 16px;
         color: {colors["TEXT_MUTED"]};
         background-color: {colors["BG_DARK"]};

ui/table.py

@@ -1,4 +1,4 @@
-from PyQt6.QtWidgets import QWidget, QVBoxLayout, QTableView, QHeaderView, QMenu
+from PyQt6.QtWidgets import QWidget, QVBoxLayout, QTableView, QHeaderView, QMenu, QInputDialog
 from PyQt6.QtCore import Qt, pyqtSignal, QItemSelection, QModelIndex
 from PyQt6.QtGui import QAction, QKeySequence, QShortcut
 
@@ -128,18 +128,30 @@ def clear_cells(self):
             self.clearRequested.emit(selection)
 
     def insert_rows(self):
-        current_index = self.table.currentIndex()
         selection = self.table.selectionModel().selection()
-        if not selection.isEmpty():
-            rows = sorted(set(index.row() for index in selection.indexes()))
-            position = rows[0]
-            rows_to_insert = len(rows)
-        elif current_index.isValid():
-            position = current_index.row()
-            rows_to_insert = 1
-        else:
+
+        rows = sorted(set(index.row() for index in selection.indexes()))
+        num_rows = len(rows)
+        if num_rows < 1:
             return
-        self.insertRowsRequested.emit(position, rows_to_insert)
+        position = rows[0]
+
+        if num_rows > 1:
+            self.insertRowsRequested.emit(position, num_rows)
+        if num_rows == 1:
+            num, ok = QInputDialog.getInt(
+                self,
+                "Insert Rows",
+                "No. of Rows:",
+                1,
+                1,
+                100
+            )
+            if ok and num > 0:
+                rows_to_insert = num
+                self.insertRowsRequested.emit(position, rows_to_insert)
+            else:
+                return
 
     def delete_rows(self):
         selection = self.table.selectionModel().selection()