mirror of
https://github.com/avitoras/telegram-tui.git
synced 2025-07-27 11:20:31 +00:00
229 lines
9.1 KiB
Python
229 lines
9.1 KiB
Python
"""Файл с кастомными экранами приложения"""
|
||
|
||
from textual.screen import Screen
|
||
from textual.widgets import Label, Input, Footer, Static, ContentSwitcher
|
||
from textual.containers import Vertical, Horizontal, VerticalScroll
|
||
from textual.events import Key
|
||
from telethon.errors import SessionPasswordNeededError
|
||
from telethon import TelegramClient, events
|
||
from src.widgets import Dialog, Chat, normalize_text
|
||
from textual import log
|
||
|
||
class AuthScreen(Screen):
|
||
"""Класс экрана логина в аккаунт"""
|
||
|
||
def __init__(
|
||
self,
|
||
name = None,
|
||
id = None,
|
||
classes = None,
|
||
telegram_client: TelegramClient | None = None
|
||
):
|
||
super().__init__(name, id, classes)
|
||
self.client = telegram_client
|
||
|
||
def on_mount(self):
|
||
self.ac = self.query_one("#auth_container")
|
||
|
||
def compose(self):
|
||
with Vertical(id="auth_container"):
|
||
yield Label(normalize_text("Добро пожаловать в Telegram TUI"))
|
||
yield Input(placeholder=normalize_text("Номер телефона"), id="phone")
|
||
yield Input(placeholder=normalize_text("Код"), id="code", disabled=True)
|
||
yield Input(
|
||
placeholder=normalize_text("Пароль"),
|
||
id="password",
|
||
password=True,
|
||
disabled=True
|
||
)
|
||
|
||
async def on_input_submitted(self, event: Input.Submitted) -> None:
|
||
if event.input.id == "phone":
|
||
self.phone = normalize_text(event.value)
|
||
self.ac.query_one("#phone").disabled = True
|
||
self.ac.query_one("#code").disabled = False
|
||
await self.client.send_code_request(phone=self.phone)
|
||
elif event.input.id == "code":
|
||
try:
|
||
self.code = normalize_text(event.value)
|
||
self.ac.query_one("#code").disabled = True
|
||
await self.client.sign_in(phone=self.phone, code=self.code)
|
||
self.app.pop_screen()
|
||
self.app.push_screen("chats")
|
||
except SessionPasswordNeededError:
|
||
self.ac.query_one("#code").disabled = True
|
||
self.ac.query_one("#password").disabled = False
|
||
elif event.input.id == "password":
|
||
self.password = normalize_text(event.value)
|
||
await self.client.sign_in(password=self.password)
|
||
await self.client.start()
|
||
self.app.pop_screen()
|
||
self.app.push_screen("chats")
|
||
|
||
class ChatScreen(Screen):
|
||
"""Класс экрана чатов, он же основной экран приложения"""
|
||
|
||
def __init__(
|
||
self,
|
||
name = None,
|
||
id = None,
|
||
classes = None,
|
||
telegram_client: TelegramClient | None = None
|
||
):
|
||
super().__init__(name, id, classes)
|
||
self.telegram_client = telegram_client
|
||
self.search_query = ""
|
||
self.selected_chat_index = 0
|
||
self.chats = []
|
||
self.focused_element = "search" # search, chat_list, dialog
|
||
|
||
async def on_mount(self):
|
||
self.limit = 100
|
||
|
||
self.chat_container = self\
|
||
.query_one("#main_container")\
|
||
.query_one("#chats")\
|
||
.query_one("#chat_container")
|
||
|
||
self.search_input = self.query_one("#search_input")
|
||
self.help_label = self.query_one("#help_label")
|
||
|
||
log("Первоначальная загрузка виджетов чатов...")
|
||
self.mount_chats(
|
||
len(
|
||
await self.telegram_client.get_dialogs(
|
||
limit=self.limit, archived=False
|
||
)
|
||
)
|
||
)
|
||
log("Первоначальная загрузка виджетов чата завершена")
|
||
|
||
self.is_chat_update_blocked = False
|
||
await self.update_chat_list()
|
||
|
||
log("Первоначальная загрузка чатов завершена")
|
||
|
||
for event in (
|
||
events.NewMessage,
|
||
events.MessageDeleted,
|
||
events.MessageEdited
|
||
):
|
||
self.telegram_client.on(event())(self.update_chat_list)
|
||
|
||
def mount_chats(self, limit: int):
|
||
log("Загрузка виджетов чатов...")
|
||
|
||
chats_amount = len(self.chat_container.query(Chat))
|
||
|
||
if limit > chats_amount:
|
||
for i in range(limit - chats_amount):
|
||
chat = Chat(id=f"chat-{i + chats_amount + 1}")
|
||
self.chat_container.mount(chat)
|
||
elif limit < chats_amount:
|
||
for i in range(chats_amount - limit):
|
||
self.chat_container.query(Chat).last().remove()
|
||
|
||
log("Виджеты чатов загружены")
|
||
|
||
async def update_chat_list(self, event = None):
|
||
log("Запрос обновления чатов")
|
||
|
||
if not self.is_chat_update_blocked:
|
||
self.is_chat_update_blocked = True
|
||
|
||
dialogs = await self.telegram_client.get_dialogs(
|
||
limit=self.limit, archived=False
|
||
)
|
||
log("Получены диалоги")
|
||
|
||
# Фильтруем диалоги по поисковому запросу
|
||
if self.search_query:
|
||
dialogs = [
|
||
d for d in dialogs
|
||
if self.search_query.lower() in normalize_text(d.name).lower()
|
||
]
|
||
|
||
limit = len(dialogs)
|
||
self.mount_chats(limit)
|
||
|
||
for i in range(limit):
|
||
chat = self.chat_container.query_one(f"#chat-{i + 1}")
|
||
chat.username = normalize_text(str(dialogs[i].name))
|
||
chat.msg = normalize_text(str(dialogs[i].message.message))
|
||
chat.peer_id = dialogs[i].id
|
||
chat.is_selected = (i == self.selected_chat_index)
|
||
chat.is_focused = (self.focused_element == "chat_list" and i == self.selected_chat_index)
|
||
|
||
self.is_chat_update_blocked = False
|
||
log("Чаты обновлены")
|
||
else:
|
||
log("Обновление чатов невозможно: уже выполняется")
|
||
|
||
def on_input_changed(self, event: Input.Changed) -> None:
|
||
if event.input.id == "search_input":
|
||
self.search_query = normalize_text(event.value)
|
||
self.selected_chat_index = 0
|
||
self.update_chat_list()
|
||
|
||
def on_key(self, event: Key) -> None:
|
||
if event.key == "tab":
|
||
# Переключаем фокус между элементами
|
||
if self.focused_element == "search":
|
||
self.focused_element = "chat_list"
|
||
self.search_input.blur()
|
||
self.update_chat_list()
|
||
elif self.focused_element == "chat_list":
|
||
self.focused_element = "search"
|
||
self.search_input.focus()
|
||
self.update_chat_list()
|
||
return
|
||
|
||
if self.focused_element == "search":
|
||
return
|
||
|
||
chats = self.chat_container.query(Chat)
|
||
if not chats:
|
||
return
|
||
|
||
if event.key == "up":
|
||
self.selected_chat_index = max(0, self.selected_chat_index - 1)
|
||
for i, chat in enumerate(chats):
|
||
chat.is_selected = (i == self.selected_chat_index)
|
||
chat.is_focused = (i == self.selected_chat_index)
|
||
# Прокручиваем к выбранному чату
|
||
selected_chat = chats[self.selected_chat_index]
|
||
self.chat_container.scroll_to(selected_chat, animate=False)
|
||
elif event.key == "down":
|
||
self.selected_chat_index = min(len(chats) - 1, self.selected_chat_index + 1)
|
||
for i, chat in enumerate(chats):
|
||
chat.is_selected = (i == self.selected_chat_index)
|
||
chat.is_focused = (i == self.selected_chat_index)
|
||
# Прокручиваем к выбранному чату
|
||
selected_chat = chats[self.selected_chat_index]
|
||
self.chat_container.scroll_to(selected_chat, animate=False)
|
||
elif event.key == "enter":
|
||
chats[self.selected_chat_index].on_click()
|
||
elif event.key == "escape":
|
||
# Возвращаемся к списку чатов
|
||
self.app.pop_screen()
|
||
self.app.push_screen("chats")
|
||
elif event.key == "/":
|
||
# Фокус на поиск
|
||
self.focused_element = "search"
|
||
self.search_input.focus()
|
||
self.update_chat_list()
|
||
|
||
def compose(self):
|
||
yield Footer()
|
||
with Horizontal(id="main_container"):
|
||
with Vertical(id="chats"):
|
||
yield Label(
|
||
"Навигация: Tab - переключение фокуса, ↑↓ - выбор чата, Enter - открыть, Esc - назад, / - поиск",
|
||
id="help_label",
|
||
classes="help-text"
|
||
)
|
||
yield Input(placeholder=normalize_text("Поиск чатов..."), id="search_input")
|
||
yield VerticalScroll(id="chat_container")
|
||
yield ContentSwitcher(id="dialog_switcher")
|
||
#yield Dialog(telegram_client=self.telegram_client)
|