mirror of
https://github.com/Balshgit/mosgortrans.git
synced 2025-09-11 13:00:40 +03:00
129 lines
4.4 KiB
Python
129 lines
4.4 KiB
Python
import asyncio
|
|
from concurrent.futures.thread import ThreadPoolExecutor
|
|
from dataclasses import dataclass
|
|
|
|
from aiogram import Bot, types
|
|
from aiogram.contrib.middlewares.logging import LoggingMiddleware
|
|
from aiogram.dispatcher import Dispatcher
|
|
from aiogram.utils.callback_data import CallbackData
|
|
|
|
from app.core.parse_web import WebParser
|
|
from app.settings import TELEGRAM_API_TOKEN
|
|
|
|
executor = ThreadPoolExecutor(10)
|
|
|
|
|
|
@dataclass
|
|
class TransportBot:
|
|
bot: Bot = Bot(TELEGRAM_API_TOKEN)
|
|
dispatcher: Dispatcher = Dispatcher(bot)
|
|
dispatcher.middleware.setup(LoggingMiddleware())
|
|
stations_cb: CallbackData = CallbackData('station', 'direction')
|
|
|
|
@staticmethod
|
|
@dispatcher.message_handler(commands=['chatid'])
|
|
async def chat_id(message: types.Message) -> types.Message:
|
|
return await TransportBot.bot.send_message(message.chat.id, message.chat.id)
|
|
|
|
@staticmethod
|
|
def get_keyboard() -> types.InlineKeyboardMarkup:
|
|
"""
|
|
Generate keyboard with list of posts
|
|
"""
|
|
markup = types.InlineKeyboardMarkup()
|
|
|
|
markup.row(
|
|
types.InlineKeyboardButton(
|
|
'Дом -> Офис',
|
|
callback_data=TransportBot.stations_cb.new(direction='home->office'),
|
|
),
|
|
types.InlineKeyboardButton(
|
|
'Офис -> Дом',
|
|
callback_data=TransportBot.stations_cb.new(direction='office->home'),
|
|
),
|
|
)
|
|
return markup
|
|
|
|
@staticmethod
|
|
@dispatcher.callback_query_handler(stations_cb.filter(direction='home->office'))
|
|
async def home_office(
|
|
query: types.CallbackQuery, callback_data: dict[str, str]
|
|
) -> types.Message:
|
|
url = 'https://yandex.ru/maps/213/moscow/stops/stop__9640740/?ll=37.527924%2C55.823470&tab=overview&z=21'
|
|
message = 'Остановка Б. Академическая ул, д. 15'
|
|
buses = [
|
|
'300',
|
|
'т19',
|
|
]
|
|
|
|
text = await TransportBot.get_buses_data(url=url, message=message, buses=buses)
|
|
|
|
return await TransportBot.bot.send_message(
|
|
query.message.chat.id, text, reply_markup=TransportBot.get_keyboard()
|
|
)
|
|
|
|
@staticmethod
|
|
@dispatcher.callback_query_handler(stations_cb.filter(direction='office->home'))
|
|
async def office_home(
|
|
query: types.CallbackQuery, callback_data: dict[str, str]
|
|
) -> types.Message:
|
|
url = 'https://yandex.ru/maps/213/moscow/stops/stop__9640288/?ll=37.505402%2C55.800214&tab=overview&z=21'
|
|
message = 'Остановка Улица Алабяна'
|
|
buses = [
|
|
'300',
|
|
'т19',
|
|
]
|
|
|
|
text = await TransportBot.get_buses_data(url=url, message=message, buses=buses)
|
|
|
|
return await TransportBot.bot.send_message(
|
|
query.message.chat.id, text, reply_markup=TransportBot.get_keyboard()
|
|
)
|
|
|
|
@staticmethod
|
|
@dispatcher.message_handler()
|
|
async def echo(message: types.Message) -> types.Message:
|
|
return await TransportBot.bot.send_message(
|
|
message.chat.id,
|
|
'Выбери остановку',
|
|
reply_markup=TransportBot.get_keyboard(),
|
|
)
|
|
|
|
@staticmethod
|
|
async def morning_bus_mailing(
|
|
chat_ids: list[int] | None, show_keyboard: bool = False
|
|
) -> None:
|
|
if not chat_ids:
|
|
return None
|
|
|
|
url = 'https://yandex.ru/maps/213/moscow/stops/stop__9640740/?ll=37.527924%2C55.823470&tab=overview&z=21'
|
|
message = 'Остановка Б. Академическая ул, д. 15'
|
|
buses = [
|
|
'300',
|
|
'т19',
|
|
]
|
|
|
|
text = await TransportBot.get_buses_data(url=url, message=message, buses=buses)
|
|
|
|
kwargs = {'reply_markup': TransportBot.get_keyboard()} if show_keyboard else {}
|
|
|
|
await asyncio.gather(
|
|
*[
|
|
TransportBot.bot.send_message(
|
|
chat_id=chat_id,
|
|
text=text,
|
|
parse_mode=types.ParseMode.HTML,
|
|
**kwargs
|
|
)
|
|
for chat_id in chat_ids
|
|
]
|
|
)
|
|
|
|
@staticmethod
|
|
async def get_buses_data(url: str, message: str, buses: list[str]) -> str:
|
|
driver = WebParser.get_driver()
|
|
loop = asyncio.get_event_loop()
|
|
return await loop.run_in_executor(
|
|
executor, WebParser.parse_yandex_maps, url, message, buses, driver
|
|
)
|