sqlshell 0.2.2__py3-none-any.whl → 0.3.0__py3-none-any.whl

sqlshell/table_list.py

@@ -37,7 +37,7 @@ class DraggableTablesList(QTreeWidget):
         self.setHeaderHidden(True)
         self.setColumnCount(1)
         self.setIndentation(15)  # Smaller indentation for a cleaner look
-        self.setSelectionMode(QTreeWidget.SelectionMode.SingleSelection)
+        self.setSelectionMode(QTreeWidget.SelectionMode.ExtendedSelection)
         self.setExpandsOnDoubleClick(False)  # Handle double-clicks manually
         
         # Apply custom styling
@@ -120,6 +120,95 @@ class DraggableTablesList(QTreeWidget):
     
     def startDrag(self, supportedActions):
         """Override startDrag to customize the drag data."""
+        # Check for multiple selected items
+        selected_items = self.selectedItems()
+        if len(selected_items) > 1:
+            # Only support dragging multiple items to the editor (not for folder management)
+            # Filter out folder items
+            table_items = [item for item in selected_items if not self.is_folder_item(item)]
+            
+            if not table_items:
+                return
+            
+            # Extract table names
+            table_names = [self.get_table_name_from_item(item) for item in table_items]
+            table_names = [name for name in table_names if name]  # Remove None values
+            
+            if not table_names:
+                return
+                
+            # Create mime data with comma-separated table names
+            mime_data = QMimeData()
+            mime_data.setText(", ".join(table_names))
+            
+            # Create drag object
+            drag = QDrag(self)
+            drag.setMimeData(mime_data)
+            
+            # Create a visually appealing drag pixmap
+            font = self.font()
+            font.setBold(True)
+            metrics = self.fontMetrics()
+            
+            # Build a preview label with limited number of tables
+            display_names = table_names[:3]
+            if len(table_names) > 3:
+                display_text = f"{', '.join(display_names)} (+{len(table_names) - 3} more)"
+            else:
+                display_text = ", ".join(display_names)
+                
+            text_width = metrics.horizontalAdvance(display_text)
+            text_height = metrics.height()
+            
+            # Make the pixmap large enough for the text plus padding and a small icon
+            padding = 10
+            pixmap = QPixmap(text_width + padding * 2 + 16, text_height + padding)
+            pixmap.fill(Qt.GlobalColor.transparent)
+            
+            # Begin painting
+            painter = QPainter(pixmap)
+            painter.setRenderHint(QPainter.RenderHint.Antialiasing)
+            
+            # Draw a nice rounded rectangle background
+            bg_color = QColor(44, 62, 80, 220)  # Dark blue with transparency
+            painter.setBrush(QBrush(bg_color))
+            painter.setPen(Qt.PenStyle.NoPen)
+            painter.drawRoundedRect(0, 0, pixmap.width(), pixmap.height(), 5, 5)
+            
+            # Draw text
+            painter.setPen(Qt.GlobalColor.white)
+            painter.setFont(font)
+            painter.drawText(int(padding + 16), int(text_height + (padding / 2) - 2), display_text)
+            
+            # Draw a small database icon (simulated)
+            icon_x = padding / 2
+            icon_y = (pixmap.height() - 12) / 2
+            
+            # Draw a simple database icon as a blue circle with lines
+            table_icon_color = QColor("#3498DB")
+            painter.setBrush(QBrush(table_icon_color))
+            painter.setPen(Qt.GlobalColor.white)
+            painter.drawEllipse(int(icon_x), int(icon_y), 12, 12)
+            
+            # Draw "table" lines inside the circle
+            painter.setPen(Qt.GlobalColor.white)
+            painter.drawLine(int(icon_x + 3), int(icon_y + 4), int(icon_x + 9), int(icon_y + 4))
+            painter.drawLine(int(icon_x + 3), int(icon_y + 6), int(icon_x + 9), int(icon_y + 6))
+            painter.drawLine(int(icon_x + 3), int(icon_y + 8), int(icon_x + 9), int(icon_y + 8))
+            
+            painter.end()
+            
+            # Set the drag pixmap
+            drag.setPixmap(pixmap)
+            
+            # Set hotspot to be at the top-left corner of the text
+            drag.setHotSpot(QPoint(padding, pixmap.height() // 2))
+            
+            # Execute drag operation - only allow copy action for multiple tables
+            drag.exec(Qt.DropAction.CopyAction)
+            return
+            
+        # Single item drag (original functionality)
         item = self.currentItem()
         if not item:
             return

sqlshell/ui/filter_header.py

@@ -1,5 +1,5 @@
 from PyQt6.QtWidgets import (QHeaderView, QMenu, QCheckBox, QWidgetAction, 
-                           QWidget, QVBoxLayout, QLineEdit, QHBoxLayout, QPushButton, QTableWidget)
+                           QWidget, QVBoxLayout, QLineEdit, QHBoxLayout, QPushButton, QTableWidget, QMessageBox)
 from PyQt6.QtCore import Qt, QRect, QPoint
 from PyQt6.QtGui import QColor, QFont, QPolygon, QPainterPath, QBrush
 
@@ -88,6 +88,17 @@ class FilterHeader(QHeaderView):
         # Add sort actions
         sort_asc_action = context_menu.addAction("Sort Ascending")
         sort_desc_action = context_menu.addAction("Sort Descending")
+        context_menu.addSeparator()
+        
+        # Add count rows action
+        count_rows_action = context_menu.addAction("Count Rows")
+        
+        # Add explain column action
+        explain_action = context_menu.addAction("Explain Column")
+        
+        # Add encode text action
+        encode_action = context_menu.addAction("Encode Text")
+        
         context_menu.addSeparator()
         filter_action = context_menu.addAction("Filter...")
         
@@ -127,6 +138,30 @@ class FilterHeader(QHeaderView):
             self.show_filter_menu(logical_index)
         elif action == toggle_bar_action:
             self.toggle_bar_chart(logical_index)
+        elif action == explain_action:
+            # Call the explain_column method on the main window
+            if self.main_window and hasattr(self.main_window, "explain_column"):
+                # Get the column name from the table (if it has a current dataframe)
+                current_tab = self.main_window.get_current_tab()
+                if current_tab and hasattr(current_tab, "current_df") and current_tab.current_df is not None:
+                    if logical_index < len(current_tab.current_df.columns):
+                        column_name = current_tab.current_df.columns[logical_index]
+                        self.main_window.explain_column(column_name)
+        elif action == encode_action:
+            # Call the encode_text method on the main window
+            if self.main_window and hasattr(self.main_window, "encode_text"):
+                # Get the column name from the table (if it has a current dataframe)
+                current_tab = self.main_window.get_current_tab()
+                if current_tab and hasattr(current_tab, "current_df") and current_tab.current_df is not None:
+                    if logical_index < len(current_tab.current_df.columns):
+                        column_name = current_tab.current_df.columns[logical_index]
+                        self.main_window.encode_text(column_name)
+        elif action == count_rows_action:
+            # Get the current tab and show row count
+            current_tab = self.main_window.get_current_tab()
+            if current_tab and hasattr(current_tab, "current_df") and current_tab.current_df is not None:
+                row_count = len(current_tab.current_df)
+                QMessageBox.information(self, "Row Count", f"Total rows: {row_count:,}")
 
     def set_main_window(self, window):
         """Set the reference to the main window"""