Skip to content

Commit dd446c6

Browse files
committed
Рефакторинг кода проекта.
1 parent de9d08d commit dd446c6

File tree

2 files changed

+10
-8
lines changed

2 files changed

+10
-8
lines changed

src/commentator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ async def write_comments_in_telegram(self, client, page: ft.Page, lv: ft.ListVie
7070
last_message_ids = {name: 0 for name in channels}
7171
for name in channels:
7272
logger.info(name[0])
73-
await SUBSCRIBE().subscribe_to_channel(client, name[0], page, lv) # Подписываемся на канал перед отправкой комментария
73+
await SUBSCRIBE(page).subscribe_to_channel(client, name[0], lv) # Подписываемся на канал перед отправкой комментария
7474
try:
7575
messages = await client.get_messages(await client.get_entity(name[0]), limit=1)
7676
for message in messages:

src/subscribe.py

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ async def action_1(_):
4040
for channel in channel_name:
4141
lv.controls.append(ft.Text(f"Подписка на: {channel[0]}")) # отображаем сообщение в ListView
4242
page.update() # Обновляем страницу
43-
await SUBSCRIBE().subscribe_to_channel(client, channel[0], page, lv)
43+
await SUBSCRIBE(page).subscribe_to_channel(client, channel[0], lv)
4444
lv.controls.append(ft.Text(f"Подписка завершена")) # отображаем сообщение в ListView
4545
page.update() # Обновляем страницу
4646

@@ -56,33 +56,35 @@ async def action_1(_):
5656
class SUBSCRIBE:
5757
"""Класс подписки на группы и каналы Telegram"""
5858

59-
async def subscribe_to_channel(self, client, channel_name, page: ft.Page, lv: ft.ListView) -> None:
59+
def __init__(self, page: ft.Page):
60+
self.page = page
61+
62+
async def subscribe_to_channel(self, client, channel_name, lv: ft.ListView) -> None:
6063
"""
6164
Подписывается на Telegram-канал.
6265
:param channel_name: Имя канала Telegram.
6366
:param client: TelegramClient объект.
64-
:param page: Страница приложения.
6567
:param lv: ListView.
6668
:return: None.
6769
"""
6870
if not channel_name or channel_name.isdigit():
6971
lv.controls.append(ft.Text(f"Неверный username канала: {channel_name}", color=ft.Colors.RED))
70-
page.update()
72+
self.page.update()
7173
return
7274

7375
try:
7476
await client(JoinChannelRequest(channel_name))
7577
lv.controls.append(ft.Text(f"Успешная подписка на {channel_name}", color=ft.Colors.RED))
76-
page.update()
78+
self.page.update()
7779
await asyncio.sleep(int(time_config))
7880
except ChannelPrivateError:
7981
lv.controls.append(ft.Text(f"Канал {channel_name} закрыт",
8082
color=ft.Colors.RED)) # отображаем сообщение в ListView
81-
page.update() # Обновляем страницу
83+
self.page.update() # Обновляем страницу
8284
except FloodWaitError as e:
8385
lv.controls.append(ft.Text(f'Flood! wait for {str(datetime.timedelta(seconds=e.seconds))}',
8486
color=ft.Colors.RED)) # отображаем сообщение в ListView
85-
page.update() # Обновляем страницу
87+
self.page.update() # Обновляем страницу
8688
time.sleep(e.seconds)
8789

8890
except ValueError:

0 commit comments

Comments
 (0)