kotolex
9103669cac
- пакет app переименован в src - файл БД теперь должен лежать в data - удалена папка коммандс, все команды вынесены в модуль коммандс или админ_коммандс - нет импортов в ините - убраны все импорты * - теперь в .env можно указать use_reminder="FALSE", тогда не будет стартовать ремайндер - можно указать урл для Правил - в конфиге теперь только конфиг -то, что из енва получаем - константы теперь лежат отдельно - добавлены явные return там, где не было - бот и вебхук разнесены отдельно - при старте проверяем, что Правила доступны - добавлено еще логирование
71 lines
2.3 KiB
Python
71 lines
2.3 KiB
Python
""" Schedule holidays. """
|
|
import csv
|
|
import logging
|
|
import time
|
|
from datetime import datetime as dt
|
|
from threading import Thread
|
|
|
|
from schedule import repeat, every, get_jobs, run_pending
|
|
from telebot import TeleBot
|
|
|
|
from .config import TOKEN, USE_REMINDER
|
|
from .constants import PYTHONCHATRU
|
|
from .report import create_report_text, reset_report_stats
|
|
|
|
if USE_REMINDER:
|
|
logging.debug("Starts reminder!!!")
|
|
# We need synchronous instance of bot here as we use synchronous library 'schedule' in a separate thread.
|
|
bot = TeleBot(TOKEN, "HTML", disable_web_page_preview=True, allow_sending_without_reply=True)
|
|
else:
|
|
bot = None
|
|
|
|
|
|
def scheduler():
|
|
"""Main loop of schedule library."""
|
|
while True:
|
|
run_pending()
|
|
time.sleep(60)
|
|
|
|
|
|
@repeat(every().day.at("07:00"), PYTHONCHATRU, None)
|
|
def remind(chat_to_repeat, today):
|
|
"""Remind holiday."""
|
|
if not today:
|
|
today = dt.today()
|
|
with open("holidays.csv", newline="", encoding="utf-8") as holidays_file:
|
|
holidays = tuple(csv.reader(holidays_file))[1:]
|
|
for entry in holidays:
|
|
date, holiday, description = entry
|
|
if today.year % 4 == 0 and holiday == "День программиста":
|
|
date = "09-12-1000" # instead of 09-13-1000 in non-leap-year
|
|
date = dt.strptime(date, "%m-%d-%Y")
|
|
if today.month == date.month and today.day == date.day:
|
|
notification = f"🎉 Сегодня <b>{holiday.upper()}</b>!\n\n{description}"
|
|
if date.year != 1000:
|
|
age = today.year - date.year
|
|
notification += f"\n\n🥳 <i>{age}-ая годовщина</i>"
|
|
bot.send_message(chat_to_repeat, notification)
|
|
|
|
|
|
@repeat(every().day.at("06:00"), PYTHONCHATRU)
|
|
def stat_report(chat_to_repeat):
|
|
"""Everyday group statistic info."""
|
|
rep = create_report_text(PYTHONCHATRU)
|
|
if rep:
|
|
bot.send_message(chat_to_repeat, rep)
|
|
reset_report_stats(PYTHONCHATRU)
|
|
|
|
|
|
def print_get_jobs():
|
|
"""Get jobs from schedule and prepare a formatted info text."""
|
|
all_jobs = get_jobs()
|
|
text = f"<b>{len(all_jobs)} jobs</b>\
|
|
\n..:: {dt.now().strftime('%d-%m-%Y %H:%M:%S')} ::..\n"
|
|
for i in all_jobs:
|
|
text += f"\n · {i!r}"
|
|
return text
|
|
|
|
|
|
if USE_REMINDER:
|
|
Thread(target=scheduler).start()
|