Dmitry
67 lines
2.0 KiB
Python
67 lines
2.0 KiB
Python
import asyncio
|
|
import logging
|
|
from contextlib import suppress
|
|
|
|
from aiogram import Bot, Dispatcher, F
|
|
from aiogram.client.default import DefaultBotProperties
|
|
from aiogram.client.session.aiohttp import AiohttpSession
|
|
from aiogram.enums import ParseMode
|
|
from aiogram.types import Message
|
|
|
|
from config import load_config
|
|
from modules.auth import AllowedUserFilter
|
|
from modules.journal import Journal
|
|
from modules.reminders import send_reminders
|
|
|
|
|
|
async def run_bot() -> None:
|
|
logging.basicConfig(level=logging.INFO)
|
|
config = load_config()
|
|
|
|
session = AiohttpSession(proxy=config.telegram_proxy) if config.telegram_proxy else None
|
|
bot = Bot(
|
|
token=config.bot_token,
|
|
session=session,
|
|
default=DefaultBotProperties(parse_mode=ParseMode.HTML),
|
|
)
|
|
dispatcher = Dispatcher()
|
|
journal = Journal(
|
|
vault_path=config.vault_path,
|
|
daily_notes_dir=config.daily_notes_dir,
|
|
note_path_format=config.note_path_format,
|
|
message_time_format=config.message_time_format,
|
|
)
|
|
|
|
@dispatcher.message(AllowedUserFilter(config.allowed_user_ids), F.text)
|
|
async def save_message(message: Message) -> None:
|
|
await journal.append(message.text or "")
|
|
await message.answer("Записал")
|
|
|
|
@dispatcher.message()
|
|
async def ignore_message(message: Message) -> None:
|
|
if message.from_user and message.from_user.id not in config.allowed_user_ids:
|
|
logging.warning("Rejected message from user_id=%s", message.from_user.id)
|
|
|
|
reminder_task = asyncio.create_task(
|
|
send_reminders(
|
|
bot=bot,
|
|
user_ids=config.allowed_user_ids,
|
|
min_seconds=config.reminder_min_seconds,
|
|
max_seconds=config.reminder_max_seconds,
|
|
)
|
|
)
|
|
try:
|
|
await dispatcher.start_polling(bot)
|
|
finally:
|
|
reminder_task.cancel()
|
|
with suppress(asyncio.CancelledError):
|
|
await reminder_task
|
|
|
|
|
|
def main() -> None:
|
|
asyncio.run(run_bot())
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|