-
Notifications
You must be signed in to change notification settings - Fork 3
/
main.py
88 lines (64 loc) · 3.35 KB
/
main.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
import asyncio
from aiogram import Bot, Dispatcher, executor, types
from aiogram.contrib.fsm_storage.memory import MemoryStorage
from aiogram.types import BotCommand
import middlewares
from config import BOT_TOKEN, ADMINS
import logging
from apscheduler.schedulers.asyncio import AsyncIOScheduler
from handlers.timer.timer_cycle import time_cycle
from handlers.register_cmd import reg_cmd
from data_b.dp_control import dp_all_users_list, dp_all_telegram_id_flc_list, dp_user_create, \
dp_all_telegram_id_time_list
bot = Bot(token=BOT_TOKEN, parse_mode=types.ParseMode.HTML)
dp = Dispatcher(bot, storage=MemoryStorage())
logging.basicConfig(level=logging.INFO)
async def set_commands(bot: Bot):
commands = [
BotCommand(command="/cancel", description="Отмена действия"),
BotCommand(command="/help", description="Просмотр функционала"),
BotCommand(command="/math", description="Задания по математике"),
BotCommand(command="/logic", description="Задания по логике"),
BotCommand(command="/flashcard", description="Карточки для запомнинания"),
BotCommand(command="/statistics", description="Просмотр статистики"),
BotCommand(command="/timer", description="Таймер")
]
await bot.set_my_commands(commands)
async def main():
scheduler = AsyncIOScheduler(timezone="Europe/Moscow")
"""
await bot.delete_webhook(drop_pending_updates=True) - в новых версиях aiogram есть проблема, то что при запуске бота,
он реагирует на сообщения, которые были отправленны ему, пока он был выключен и это не чинилось dp.skip_updates()
подробнее об этой ошибке: https://github.com/aiogram/aiogram/issues/418
"""
# Удаление последнего сообщения
await bot.delete_webhook(drop_pending_updates=True)
# Запуск "антифлуда"
middlewares.setup(dp)
# Это запуск таймера AsyncIOScheduler
scheduler.add_job(time_cycle, "interval", seconds=60, args=(dp,))
scheduler.start()
await set_commands(bot)
# функция регистрации "register_message_handler"
reg_cmd(dp)
# Запуск полинга
await dp.start_polling()
if __name__ == "__main__":
# ------------------------------------------------------
"""
Это блок добавления пользователей в таблицу(users)
Сделанн он, чтобы добавить всех пользователей в таблицу(users), у которых есть карточки flashcards и timer,
т.к там хранится их 'telegram_user_id'
"""
all_users_list = dp_all_users_list()
all_telegram_id_flc = dp_all_telegram_id_flc_list()
for i in all_telegram_id_flc:
if i not in all_users_list:
dp_user_create(i)
all_telegram_id_time = dp_all_telegram_id_time_list()
for i in all_telegram_id_time:
if i not in all_users_list:
dp_user_create(i)
# ------------------------------------------------------
# Ассинхронный запуск бота
asyncio.run(main())