mirror of
https://github.com/EDeev/y.calendarkin.git
synced 2026-10-08 00:39:27 +03:00
Egor Deev
18e79e24a5
В репозитории оставался код на aiogram 2 (executor, message_handler), а requirements.txt уже требовал aiogram 3 — по README бот не запускался (issue #1). Переписанная версия с разделением на bot.py / handlers.py / init.py давно работала на сервере, теперь она и здесь. Переводы строк приведены к LF.
104 lines
4.7 KiB
Python
104 lines
4.7 KiB
Python
import asyncio, logging, time, wget, os
|
||
|
||
from init import *
|
||
from script import *
|
||
from handlers import router
|
||
|
||
|
||
# ПРОВЕРКА НА СОБЫТИЕ
|
||
async def alarm(wait_for):
|
||
while True:
|
||
await asyncio.sleep(wait_for)
|
||
|
||
users_id = du.all_users()
|
||
for user_id in users_id:
|
||
user_id = user_id[0]
|
||
|
||
if dc.clock_exists(user_id):
|
||
if du.get_status(user_id):
|
||
events = sorted(text_ical(user_id, du.get_tz(user_id)))
|
||
tg_id = str(du.get_first_user_id(user_id))
|
||
|
||
start = dc.get_start(user_id)
|
||
daily = dc.get_daily(user_id)
|
||
alarm = dc.get_alarm(user_id)
|
||
|
||
alarm2_status = dc.get_status2(user_id)
|
||
|
||
today = dt_now(du.get_tz(user_id)).date()
|
||
time_check = dt_now(du.get_tz(user_id)).time()
|
||
|
||
# ДЛЯ DAILY
|
||
counter = 0
|
||
txt = "<b>События сегодня</b>\n\n"
|
||
|
||
delta_daily1 = time(hour=8, minute=0)
|
||
delta_daily2 = time(hour=8, minute=1)
|
||
# --------------------------------
|
||
|
||
for event in events:
|
||
if event[0] == today:
|
||
d_event = datetime.combine(today, event[1])
|
||
|
||
if daily and delta_daily1 <= time_check < delta_daily2:
|
||
counter += 1
|
||
txt += message_form(counter, event[3])
|
||
|
||
delta_start = delta_time(d_event, 1, 0)
|
||
if start and delta_start[0] < time_check <= delta_start[1]:
|
||
await bot.send_message(chat_id=tg_id, disable_web_page_preview=True,
|
||
text=f"<b>Событие начинается!</b>\n\n" + message_form(0, event[3]))
|
||
|
||
delta_alarm1 = delta_time(d_event, alarm[0], alarm[0] - 1)
|
||
if delta_alarm1[0] < time_check <= delta_alarm1[1]:
|
||
await bot.send_message(chat_id=tg_id, disable_web_page_preview=True,
|
||
text=f"<b>Напоминаю!</b>\n<i>Через {alarm[0]} минут "
|
||
f"будет событие:</i>\n\n{message_form(0, event[3])}")
|
||
|
||
if alarm2_status:
|
||
delta_alarm2 = delta_time(d_event, alarm[1], alarm[1] - 1)
|
||
if delta_alarm2[0] < time_check <= delta_alarm2[1]:
|
||
await bot.send_message(chat_id=tg_id, disable_web_page_preview=True,
|
||
text=f"<b>Напоминаю!</b>\n<i>Через {alarm[1]} минут "
|
||
f"будет событие:</i>\n\n{message_form(0, event[3])}")
|
||
|
||
if daily and delta_daily1 <= time_check < delta_daily2:
|
||
if counter:
|
||
await bot.send_message(chat_id=tg_id, disable_web_page_preview=True, text=txt)
|
||
else:
|
||
await bot.send_message(chat_id=tg_id, disable_web_page_preview=True, text=f"Сегодня событий <b>нет</b>")
|
||
|
||
|
||
async def update(wait_for):
|
||
while True:
|
||
await asyncio.sleep(wait_for)
|
||
|
||
users_id = du.all_users()
|
||
for user_id in users_id:
|
||
user_id = user_id[0]
|
||
|
||
if du.url_exists(user_id):
|
||
if du.get_status(user_id):
|
||
wget.download(du.get_url(user_id), f'../data/icals/{str(user_id)}_new.ics')
|
||
|
||
os.remove(f'../data/icals/{str(user_id)}.ics')
|
||
os.rename(f'../data/icals/{str(user_id)}_new.ics', f'../data/icals/{str(user_id)}.ics')
|
||
|
||
|
||
async def main() -> None:
|
||
dp.include_router(router)
|
||
await bot.delete_webhook(drop_pending_updates=True)
|
||
await dp.start_polling(bot, allowed_updates=dp.resolve_used_update_types(), skipUpdates=True)
|
||
|
||
|
||
async def process() -> None:
|
||
await asyncio.gather(main(), # ЗАПУСК ОСНОВНОЙ ПРОГРАММЫ
|
||
alarm(60), # ПРОВЕРКА КАЖДУЮ 1 МИНУТУ
|
||
update(780)) # ПРОВЕРКА КАЖДУЮ 13 МИНУТУ
|
||
|
||
|
||
if __name__ == "__main__":
|
||
logging.basicConfig(level=logging.INFO)
|
||
|
||
try: asyncio.run(process())
|
||
except KeyboardInterrupt: pass
|