65 lines
2.1 KiB
Python
65 lines
2.1 KiB
Python
"""焦点管理器"""
|
|
|
|
from textual.app import App
|
|
|
|
|
|
class FocusHandler:
|
|
"""焦点管理器"""
|
|
|
|
# 焦点循环顺序
|
|
FOCUS_RING = [
|
|
"input-textarea", # 左侧输入框
|
|
"api-key-input", # 右侧配置区 API Key
|
|
"model-input", # 右侧配置区 Model
|
|
"base-url-input", # 右侧配置区 Base URL
|
|
"cypher-textarea", # 右侧 Cypher 查询框
|
|
]
|
|
|
|
# 焦点名称映射
|
|
FOCUS_NAMES = {
|
|
"input-textarea": "Input",
|
|
"api-key-input": "Config-API",
|
|
"model-input": "Config-Model",
|
|
"base-url-input": "Config-URL",
|
|
"cypher-textarea": "Query",
|
|
}
|
|
|
|
def __init__(self):
|
|
self._current_index = 0
|
|
|
|
def next_focus(self, app: App) -> None:
|
|
"""切换到下一个焦点"""
|
|
self._current_index = (self._current_index + 1) % len(self.FOCUS_RING)
|
|
widget_id = self.FOCUS_RING[self._current_index]
|
|
self._focus_widget(app, widget_id)
|
|
|
|
def prev_focus(self, app: App) -> None:
|
|
"""切换到上一个焦点"""
|
|
self._current_index = (self._current_index - 1) % len(self.FOCUS_RING)
|
|
widget_id = self.FOCUS_RING[self._current_index]
|
|
self._focus_widget(app, widget_id)
|
|
|
|
def focus_input(self, app: App) -> None:
|
|
"""聚焦到输入框"""
|
|
self._current_index = 0
|
|
self._focus_widget(app, self.FOCUS_RING[0])
|
|
|
|
def focus_query(self, app: App) -> None:
|
|
"""聚焦到查询框"""
|
|
self._current_index = len(self.FOCUS_RING) - 1
|
|
self._focus_widget(app, self.FOCUS_RING[-1])
|
|
|
|
def get_current_focus_name(self) -> str:
|
|
"""获取当前焦点名称"""
|
|
widget_id = self.FOCUS_RING[self._current_index]
|
|
return self.FOCUS_NAMES.get(widget_id, "Unknown")
|
|
|
|
def _focus_widget(self, app: App, widget_id: str) -> None:
|
|
"""聚焦到指定组件"""
|
|
try:
|
|
widget = app.query_one(f"#{widget_id}")
|
|
widget.focus()
|
|
except Exception:
|
|
# 如果找不到组件，回退到输入框
|
|
self.focus_input(app)
|