73 lines
2.2 KiB
Python
73 lines
2.2 KiB
Python
"""输入框组件"""
|
|
|
|
from textual.containers import Container, Horizontal
|
|
from textual.widgets import TextArea, Button
|
|
from textual.message import Message
|
|
|
|
|
|
class InputBox(Container):
|
|
"""输入框组件"""
|
|
|
|
class SendMessage(Message):
|
|
"""发送消息事件"""
|
|
def __init__(self, content: str) -> None:
|
|
self.content = content
|
|
super().__init__()
|
|
|
|
class ClearHistory(Message):
|
|
"""清空聊天记录事件"""
|
|
def __init__(self) -> None:
|
|
super().__init__()
|
|
|
|
def __init__(self, **kwargs):
|
|
super().__init__(**kwargs)
|
|
self._history: list[str] = []
|
|
self._history_index: int = -1
|
|
|
|
def compose(self):
|
|
"""构建输入框"""
|
|
yield TextArea(
|
|
placeholder="输入消息... (Enter换行)",
|
|
id="input-textarea"
|
|
)
|
|
with Horizontal(classes="input-buttons"):
|
|
yield Button("清空", id="clear-button", variant="default")
|
|
yield Button("发送", id="send-button", variant="primary")
|
|
|
|
def on_mount(self) -> None:
|
|
"""组件挂载时"""
|
|
# 设置焦点
|
|
textarea = self.query_one(TextArea)
|
|
textarea.focus()
|
|
|
|
def on_button_pressed(self, event: Button.Pressed) -> None:
|
|
"""处理按钮点击"""
|
|
if event.button.id == "send-button":
|
|
self._send_message()
|
|
elif event.button.id == "clear-button":
|
|
self.post_message(self.ClearHistory())
|
|
|
|
def on_key(self, event) -> None:
|
|
"""处理按键事件"""
|
|
if event.key == "enter" and event.ctrl:
|
|
self._send_message()
|
|
event.stop()
|
|
|
|
def _send_message(self) -> None:
|
|
"""发送消息"""
|
|
textarea = self.query_one(TextArea)
|
|
content = textarea.text.strip()
|
|
if content:
|
|
# 保存到历史
|
|
self._history.append(content)
|
|
self._history_index = len(self._history)
|
|
# 发送消息
|
|
self.post_message(self.SendMessage(content))
|
|
# 清空输入框
|
|
textarea.clear()
|
|
|
|
def focus(self) -> None:
|
|
"""聚焦输入框"""
|
|
textarea = self.query_one(TextArea)
|
|
textarea.focus()
|