Initial Codes
This commit is contained in:
@@ -0,0 +1,340 @@
|
||||
"""
|
||||
Multi-tab SQL editor with syntax highlighting, line numbers, and run controls.
|
||||
"""
|
||||
import os
|
||||
from PyQt6.QtWidgets import (
|
||||
QWidget, QVBoxLayout, QHBoxLayout, QPlainTextEdit, QTextEdit,
|
||||
QTabWidget, QPushButton, QLabel, QSplitter, QTabBar, QSizePolicy,
|
||||
QFileDialog, QMessageBox, QToolButton, QComboBox,
|
||||
)
|
||||
from PyQt6.QtCore import Qt, QRect, QSize, pyqtSignal, QTimer
|
||||
from PyQt6.QtGui import (
|
||||
QColor, QPainter, QTextFormat, QFont, QKeySequence, QShortcut,
|
||||
QFontMetrics, QTextCursor,
|
||||
)
|
||||
|
||||
from app.ui.syntax_highlighter import SQLHighlighter
|
||||
from app.ui.results_panel import ResultsPanel
|
||||
from app.utils.worker import QueryWorker
|
||||
|
||||
|
||||
# ── Line-number gutter ────────────────────────────────────────────────────────
|
||||
|
||||
class LineNumberArea(QWidget):
|
||||
def __init__(self, editor):
|
||||
super().__init__(editor)
|
||||
self._editor = editor
|
||||
|
||||
def sizeHint(self) -> QSize:
|
||||
return QSize(self._editor.line_number_area_width(), 0)
|
||||
|
||||
def paintEvent(self, event):
|
||||
self._editor.line_number_area_paint_event(event)
|
||||
|
||||
|
||||
class CodeEditor(QPlainTextEdit):
|
||||
"""QPlainTextEdit with line numbers, current-line highlight, and tab→spaces."""
|
||||
|
||||
def __init__(self, parent=None):
|
||||
super().__init__(parent)
|
||||
self._line_area = LineNumberArea(self)
|
||||
|
||||
# Font
|
||||
font = QFont("Consolas", 13)
|
||||
font.setFixedPitch(True)
|
||||
self.setFont(font)
|
||||
self.setTabStopDistance(QFontMetrics(font).horizontalAdvance(" ") * 4)
|
||||
|
||||
# Connect signals
|
||||
self.blockCountChanged.connect(self._update_line_area_width)
|
||||
self.updateRequest.connect(self._update_line_area)
|
||||
self.cursorPositionChanged.connect(self._highlight_current_line)
|
||||
|
||||
self._update_line_area_width(0)
|
||||
self._highlight_current_line()
|
||||
|
||||
def line_number_area_width(self) -> int:
|
||||
digits = max(3, len(str(self.blockCount())))
|
||||
return 12 + self.fontMetrics().horizontalAdvance("9") * digits
|
||||
|
||||
def _update_line_area_width(self, _):
|
||||
self.setViewportMargins(self.line_number_area_width(), 0, 0, 0)
|
||||
|
||||
def _update_line_area(self, rect, dy):
|
||||
if dy:
|
||||
self._line_area.scroll(0, dy)
|
||||
else:
|
||||
self._line_area.update(0, rect.y(), self._line_area.width(), rect.height())
|
||||
if rect.contains(self.viewport().rect()):
|
||||
self._update_line_area_width(0)
|
||||
|
||||
def resizeEvent(self, event):
|
||||
super().resizeEvent(event)
|
||||
cr = self.contentsRect()
|
||||
self._line_area.setGeometry(
|
||||
QRect(cr.left(), cr.top(), self.line_number_area_width(), cr.height())
|
||||
)
|
||||
|
||||
def _highlight_current_line(self):
|
||||
extra = []
|
||||
if not self.isReadOnly():
|
||||
sel = QTextEdit.ExtraSelection()
|
||||
sel.format.setBackground(QColor("#2a2a3c"))
|
||||
sel.format.setProperty(QTextFormat.Property.FullWidthSelection, True)
|
||||
sel.cursor = self.textCursor()
|
||||
sel.cursor.clearSelection()
|
||||
extra.append(sel)
|
||||
self.setExtraSelections(extra)
|
||||
|
||||
def line_number_area_paint_event(self, event):
|
||||
painter = QPainter(self._line_area)
|
||||
painter.fillRect(event.rect(), QColor("#1a1a2e"))
|
||||
|
||||
block = self.firstVisibleBlock()
|
||||
number = block.blockNumber()
|
||||
top = round(self.blockBoundingGeometry(block).translated(
|
||||
self.contentOffset()).top())
|
||||
bottom = top + round(self.blockBoundingRect(block).height())
|
||||
|
||||
while block.isValid() and top <= event.rect().bottom():
|
||||
if block.isVisible() and bottom >= event.rect().top():
|
||||
painter.setPen(QColor("#45475a"))
|
||||
painter.drawText(
|
||||
0, top, self._line_area.width() - 6,
|
||||
self.fontMetrics().height(),
|
||||
Qt.AlignmentFlag.AlignRight, str(number + 1)
|
||||
)
|
||||
block = block.next()
|
||||
top = bottom
|
||||
bottom = top + round(self.blockBoundingRect(block).height())
|
||||
number += 1
|
||||
|
||||
def keyPressEvent(self, event):
|
||||
# Tab → 4 spaces
|
||||
if event.key() == Qt.Key.Key_Tab:
|
||||
cursor = self.textCursor()
|
||||
cursor.insertText(" ")
|
||||
return
|
||||
# Ctrl+/ → toggle comment
|
||||
if event.modifiers() == Qt.KeyboardModifier.ControlModifier \
|
||||
and event.key() == Qt.Key.Key_Slash:
|
||||
self._toggle_comment()
|
||||
return
|
||||
super().keyPressEvent(event)
|
||||
|
||||
def _toggle_comment(self):
|
||||
cursor = self.textCursor()
|
||||
start = cursor.selectionStart()
|
||||
end = cursor.selectionEnd()
|
||||
cursor.setPosition(start)
|
||||
cursor.movePosition(QTextCursor.MoveOperation.StartOfBlock)
|
||||
cursor.setPosition(end, QTextCursor.MoveMode.KeepAnchor)
|
||||
cursor.movePosition(QTextCursor.MoveOperation.EndOfBlock,
|
||||
QTextCursor.MoveMode.KeepAnchor)
|
||||
text = cursor.selectedText()
|
||||
lines = text.split("\u2029") # Qt paragraph separator
|
||||
if all(l.lstrip().startswith("--") for l in lines if l.strip()):
|
||||
new = [l.replace("--", "", 1) if l.lstrip().startswith("--") else l
|
||||
for l in lines]
|
||||
else:
|
||||
new = ["--" + l for l in lines]
|
||||
cursor.insertText("\u2029".join(new))
|
||||
|
||||
def selected_or_all(self) -> str:
|
||||
cursor = self.textCursor()
|
||||
text = cursor.selectedText().replace("\u2029", "\n")
|
||||
return text if text.strip() else self.toPlainText()
|
||||
|
||||
|
||||
# ── Single editor tab (editor + results splitter) ─────────────────────────────
|
||||
|
||||
class EditorTab(QWidget):
|
||||
status_message = pyqtSignal(str)
|
||||
|
||||
def __init__(self, driver, database: str = "", parent=None):
|
||||
super().__init__(parent)
|
||||
self._driver = driver
|
||||
self._database = database
|
||||
self._worker: QueryWorker | None = None
|
||||
self._build_ui()
|
||||
|
||||
def _build_ui(self):
|
||||
root = QVBoxLayout(self)
|
||||
root.setContentsMargins(0, 0, 0, 0)
|
||||
root.setSpacing(0)
|
||||
|
||||
# ── Editor toolbar ─────────────────────────────────────────────────────
|
||||
toolbar = QHBoxLayout()
|
||||
toolbar.setContentsMargins(6, 4, 6, 4)
|
||||
toolbar.setSpacing(4)
|
||||
|
||||
self._run_btn = QPushButton("▶ Run F5")
|
||||
self._run_btn.setObjectName("runBtn")
|
||||
self._run_btn.clicked.connect(self._run)
|
||||
|
||||
self._stop_btn = QPushButton("⏹ Stop")
|
||||
self._stop_btn.setObjectName("stopBtn")
|
||||
self._stop_btn.setEnabled(False)
|
||||
self._stop_btn.clicked.connect(self._stop)
|
||||
|
||||
self._explain_btn = QPushButton("🔎 Explain")
|
||||
self._explain_btn.clicked.connect(self._explain)
|
||||
|
||||
self._export_btn = QPushButton("📤 Export")
|
||||
self._export_btn.clicked.connect(self._export)
|
||||
|
||||
self._db_label = QLabel(f"DB: {self._database}" if self._database else "")
|
||||
self._db_label.setObjectName("dbLabel")
|
||||
|
||||
toolbar.addWidget(self._run_btn)
|
||||
toolbar.addWidget(self._stop_btn)
|
||||
toolbar.addWidget(self._explain_btn)
|
||||
toolbar.addWidget(self._export_btn)
|
||||
toolbar.addStretch()
|
||||
toolbar.addWidget(self._db_label)
|
||||
|
||||
root.addLayout(toolbar)
|
||||
|
||||
# ── Splitter: editor / results ──────────────────────────────────────
|
||||
self._splitter = QSplitter(Qt.Orientation.Vertical)
|
||||
self._splitter.setHandleWidth(3)
|
||||
|
||||
self._editor = CodeEditor()
|
||||
SQLHighlighter(self._editor.document())
|
||||
|
||||
self._results = ResultsPanel()
|
||||
self._results.status_message.connect(self.status_message)
|
||||
|
||||
self._splitter.addWidget(self._editor)
|
||||
self._splitter.addWidget(self._results)
|
||||
self._splitter.setSizes([400, 250])
|
||||
|
||||
root.addWidget(self._splitter, 1)
|
||||
|
||||
# ── Shortcuts ──────────────────────────────────────────────────────
|
||||
QShortcut(QKeySequence("F5"), self, self._run)
|
||||
QShortcut(QKeySequence("Ctrl+Return"), self, self._run)
|
||||
|
||||
# ── Run logic ─────────────────────────────────────────────────────────────
|
||||
|
||||
def _run(self):
|
||||
sql = self._editor.selected_or_all().strip()
|
||||
if not sql:
|
||||
return
|
||||
self._results.show_loading()
|
||||
self._run_btn.setEnabled(False)
|
||||
self._stop_btn.setEnabled(True)
|
||||
|
||||
is_script = ";" in sql[:-1] # multiple statements
|
||||
self._worker = QueryWorker(self._driver, sql, is_script=is_script)
|
||||
self._worker.finished.connect(self._on_result)
|
||||
self._worker.script_done.connect(self._on_script_done)
|
||||
self._worker.error.connect(self._on_error)
|
||||
self._worker.finished.connect(lambda *_: self._reset_buttons())
|
||||
self._worker.script_done.connect(lambda *_: self._reset_buttons())
|
||||
self._worker.error.connect(lambda *_: self._reset_buttons())
|
||||
self._worker.start()
|
||||
|
||||
def _stop(self):
|
||||
if self._worker and self._worker.isRunning():
|
||||
self._worker.terminate()
|
||||
self._reset_buttons()
|
||||
|
||||
def _reset_buttons(self):
|
||||
self._run_btn.setEnabled(True)
|
||||
self._stop_btn.setEnabled(False)
|
||||
|
||||
def _explain(self):
|
||||
sql = self._editor.selected_or_all().strip()
|
||||
if not sql:
|
||||
return
|
||||
# Prefer opening a full ExplainPanel in the main window workspace
|
||||
from app.main_window import MainWindow
|
||||
win = self.window()
|
||||
if isinstance(win, MainWindow):
|
||||
win.open_explain_tab(self._driver, self._database, sql)
|
||||
else:
|
||||
# Fallback: show raw EXPLAIN in the inline results panel
|
||||
try:
|
||||
cols, rows = self._driver.explain_query(sql)
|
||||
self._results.show_data(cols, rows, len(rows), 0)
|
||||
except Exception as e:
|
||||
self._results.show_error(str(e))
|
||||
|
||||
def _export(self):
|
||||
self._results.export_dialog()
|
||||
|
||||
def _on_result(self, cols, rows, cnt, elapsed):
|
||||
self._results.show_data(cols, rows, cnt, elapsed)
|
||||
|
||||
def _on_script_done(self, results: list):
|
||||
# Show the last SELECT result; messages for DML
|
||||
for cols, rows, cnt, msg in results:
|
||||
if cols:
|
||||
self._results.show_data(cols, rows, cnt, 0)
|
||||
else:
|
||||
self._results.show_message(msg)
|
||||
|
||||
def _on_error(self, msg: str):
|
||||
self._results.show_error(msg)
|
||||
|
||||
# ── Public ─────────────────────────────────────────────────────────────────
|
||||
|
||||
def set_sql(self, sql: str):
|
||||
self._editor.setPlainText(sql)
|
||||
|
||||
def get_sql(self) -> str:
|
||||
return self._editor.toPlainText()
|
||||
|
||||
|
||||
# ── Tabbed SQL editor container ───────────────────────────────────────────────
|
||||
|
||||
class SQLEditorWidget(QWidget):
|
||||
status_message = pyqtSignal(str)
|
||||
|
||||
def __init__(self, parent=None):
|
||||
super().__init__(parent)
|
||||
self._build_ui()
|
||||
|
||||
def _build_ui(self):
|
||||
root = QVBoxLayout(self)
|
||||
root.setContentsMargins(0, 0, 0, 0)
|
||||
|
||||
self._tabs = QTabWidget()
|
||||
self._tabs.setTabsClosable(True)
|
||||
self._tabs.setMovable(True)
|
||||
self._tabs.tabCloseRequested.connect(self._close_tab)
|
||||
|
||||
# New tab button
|
||||
new_btn = QToolButton()
|
||||
new_btn.setText("＋")
|
||||
new_btn.setToolTip("New SQL Tab")
|
||||
new_btn.clicked.connect(lambda: self.new_tab())
|
||||
self._tabs.setCornerWidget(new_btn, Qt.Corner.TopRightCorner)
|
||||
|
||||
root.addWidget(self._tabs)
|
||||
|
||||
def new_tab(self, driver=None, database: str = "",
|
||||
sql: str = "", title: str = None) -> EditorTab:
|
||||
tab = EditorTab(driver, database)
|
||||
tab.status_message.connect(self.status_message)
|
||||
if sql:
|
||||
tab.set_sql(sql)
|
||||
label = title or (f"Query — {database}" if database else "Query")
|
||||
idx = self._tabs.addTab(tab, label)
|
||||
self._tabs.setCurrentIndex(idx)
|
||||
return tab
|
||||
|
||||
def _close_tab(self, idx: int):
|
||||
if self._tabs.count() > 1:
|
||||
self._tabs.removeTab(idx)
|
||||
|
||||
def current_tab(self) -> EditorTab | None:
|
||||
w = self._tabs.currentWidget()
|
||||
return w if isinstance(w, EditorTab) else None
|
||||
|
||||
def open_sql_for(self, driver, database: str, sql: str = ""):
|
||||
tab = self.new_tab(driver, database, sql,
|
||||
title=f"SQL — {database}")
|
||||
if sql:
|
||||
tab.set_sql(sql)
|
||||
Reference in New Issue
Block a user