diff --git a/.gitignore b/.gitignore index 9a143ba..cdced54 100644 --- a/.gitignore +++ b/.gitignore @@ -1,6 +1,6 @@ - users.json schedule.json settings.json json/students.json json/schedule_2sem.json +nohup.out diff --git a/schedules.py b/schedules.py index bccc427..2b5b600 100644 --- a/schedules.py +++ b/schedules.py @@ -7,11 +7,12 @@ from telegram.constants import ParseMode import for_json from admin import admin_command, is_admin, send_admin_message, send_admin_document -# bot = telebot.TeleBot("TOKEN_API") +bot = telebot.TeleBot("TOKEN_API") -@bot.message_handler(commands=['help', 'faq']) + +@bot.message_handler(commands=["help", "faq"]) def help(message): - help_msg = 'Мои команды:\ + help_msg = "Мои команды:\ \n/start - используй, чтобы сменить номер группы.\ \n/schedule - используй, чтобы получить расписание на сегодня.\ \n/random - сделай очередь из людей\ @@ -21,12 +22,12 @@ def help(message): \n/timeout - настрой время, когда бот будет присылать тебе сообщение\ \n/request - используй, чтобы отправить разработчику какой-то запрос\ \n/source - Страница бота на Github\ - \n\nИли же можно всегда написать напрямую: @Kr0sH_512' - + \n\nИли же можно всегда написать напрямую: @Kr0sH_512" + send_message(message.chat.id, help_msg) - + if is_admin(message): - admin_help_msg = 'И команды только для админа:\ + admin_help_msg = "И команды только для админа:\ \n/restart - перезапуск бота.\ \n/update - обновление schedule задач\ \n/stats - получание статистики\ @@ -34,118 +35,136 @@ def help(message): \n/info id_пользователя - узнать настройки пользователя\ \n/spam - сделать рассылку\ \n/pause_all - приостановить бота для всех (каникулы/выходные)\ - \n/stop id_пользователя - приостановить отправку сообщений для пользователя' + \n/stop id_пользователя - приостановить отправку сообщений для пользователя" send_admin_message(admin_help_msg) - + return + ### --//-- -# @bot.message_handler(commands=['test']) -# @admin_command -# def update_schedules(message): -# text = 'Жирный текст\ -# \nКурсивный текст\ -# \nПодчёркнутый\ -# \nПеречёркнутый текст\ -# \nСсылка\ -# \nМоноширинный текст\ -# \n
Форматированный с сохранением     пробелов
\ -# \n
Цитата
' -# send_admin_message(text) -# return - -@bot.message_handler(commands=['restart']) +@bot.message_handler(commands=["test", "t"]) @admin_command -def restart_bot(message): - send_admin_message('bye') - os.execv(sys.executable, ['python'] + sys.argv) - -@bot.message_handler(commands=['update']) -@admin_command -def update_schedules(message): - for_json.create_schedule_tasks(manual=True) - send_admin_message('Произошёл update') +def test(message): + text = 'Жирный текст\ + \nКурсивный текст\ + \nПодчёркнутый\ + \nПеречёркнутый текст\ + \nСсылка\ + \nМоноширинный текст\ + \n
Форматированный с сохранением     пробелов
\ + \n
Цитата
' + send_admin_message(text) return -@bot.message_handler(commands=['stats']) + +@bot.message_handler(commands=["restart", 'r']) +@admin_command +def restart_bot(message): + send_admin_message("bye") + os.execv(sys.executable, ["python"] + sys.argv) + + +@bot.message_handler(commands=["update"]) +@admin_command +def update_schedules(message): + for_json.create_schedule_tasks(manual=True) + send_admin_message("Произошёл update") + + return + + +@bot.message_handler(commands=["stats"]) @admin_command def send_stat(message): text = "Статистика:\ \n(группа: всего/откл. увед./беседы)" - + sum_users = 0 sum_dis_not = 0 sum_chats = 0 - + groups = list(for_json.groups_in_json()) groups.append("other") - + for i in groups: users = for_json.students_in_group(i) sum_users += len(users) - - dis_not = len([i for i, x in enumerate(users) if for_json.return_infos(x)["allow_message"] == "no"]) + + dis_not = len( + [ + i + for i, x in enumerate(users) + if for_json.return_infos(x)["allow_message"] == "no" + ] + ) sum_dis_not += dis_not - + chats = len([i for i, x in enumerate(users) if x[0] == "-"]) sum_chats += chats - + text += "\n{}: {} / {} / {}".format(i, len(users), dis_not, chats) - + text += "\n\nВсего: {} / {} / {}".format(sum_users, sum_dis_not, sum_chats) - + send_admin_message(text) - + return -@bot.message_handler(commands=['json']) + +@bot.message_handler(commands=["json"]) @admin_command def send_json(message): - with open(for_json.path_users, 'rb') as json_file: + with open(for_json.path_users, "rb") as json_file: send_admin_document(json_file) - with open(for_json.path_schedule, 'rb') as json_file: + with open(for_json.path_schedule, "rb") as json_file: send_admin_document(json_file) - with open(for_json.path_students, 'rb') as json_file: + with open(for_json.path_students, "rb") as json_file: send_admin_document(json_file) - + return + # info user_id -@bot.message_handler(commands=['spam']) + +@bot.message_handler(commands=["spam"]) @admin_command def spam(message): - send_admin_message("Пришли мне id, номер группы или all для рассылки сообщения этим пользователям") - + send_admin_message( + "Пришли мне id, номер группы или all для рассылки сообщения этим пользователям" + ) + bot.register_next_step_handler(message, spam_msg) - + return + def spam_msg(message): - if message.text == "stop" or message.text == "стоп" or message.text[0] == '/': - send_admin_message('Отмена отпраки') + if message.text == "stop" or message.text == "стоп" or message.text[0] == "/": + send_admin_message("Отмена отпраки") return - + send_admin_message("Пришли мне текст для рассылки сообщения этим пользователям") - + bot.register_next_step_handler(message, spam_cnf, message.text) - + return + def spam_cnf(message, data): - if message.text == "stop" or message.text == "стоп" or message.text[0] == '/': - send_admin_message('Отмена отпраки') + if message.text == "stop" or message.text == "стоп" or message.text[0] == "/": + send_admin_message("Отмена отпраки") return - + global text_spam text_spam = message.text - + text = "Пожалуйста, проверь итоговое сообщение и подтверди отправку:\ \n\nКому: " - + if data == "all": text += "всем" elif data in for_json.groups_in_json(): @@ -155,31 +174,36 @@ def spam_cnf(message, data): else: send_admin_message("Ошибка при выборе пользователей") return - - text +="\n\nСообщение: {}".format(message.text) - + + text += "\n\nСообщение: {}".format(message.text) + markup = types.InlineKeyboardMarkup() - - markup.add(types.InlineKeyboardButton(text="✅", callback_data="spam#{}".format(data)), - types.InlineKeyboardButton(text="❌", callback_data="spam_no")) - - bot.send_message(message.from_user.id, text, parse_mode=ParseMode.HTML, reply_markup=markup) - + + markup.add( + types.InlineKeyboardButton(text="✅", callback_data="spam#{}".format(data)), + types.InlineKeyboardButton(text="❌", callback_data="spam_no"), + ) + + bot.send_message( + message.from_user.id, text, parse_mode=ParseMode.HTML, reply_markup=markup + ) + return + @bot.callback_query_handler(func=lambda call: "spam" in call.data) def spam_send(call): if call.data == "spam_no": - send_admin_message('Отмена отпраки') + send_admin_message("Отмена отпраки") return - + tmp, data = call.data.split("#") - + global text_spam text = text_spam - + users = [] - + if data == "all": groups = for_json.groups_in_json() for group in groups: @@ -189,372 +213,491 @@ def spam_send(call): users = for_json.students_in_group(data) elif for_json.return_infos(data) != None: users = [data] - + for i in users: send_message(i, text) - - bot.edit_message_text(chat_id=call.message.chat.id, - message_id=call.message.message_id, - text='Успешно!', - parse_mode=ParseMode.HTML) - + + bot.edit_message_text( + chat_id=call.message.chat.id, + message_id=call.message.message_id, + text="Успешно!", + parse_mode=ParseMode.HTML, + ) + return -@bot.message_handler(commands=['pause_all']) + +@bot.message_handler(commands=["pause_all"]) @admin_command def pause_bot(message): for_json.pause_bot() - + return -@bot.message_handler(commands=['stop']) + +@bot.message_handler(commands=["stop"]) @admin_command def stop_msg(message): - if len(str(message.text).split(' ')) != 2: - send_admin_message('Эта команда вида /stop id_пользователя') + if len(str(message.text).split(" ")) != 2: + send_admin_message("Эта команда вида /stop id_пользователя") return - for_json.change_user_param(str(message.text).split(' ')[1], 'allow_message', 'no') - send_admin_message('Успешно') + + for_json.change_user_param(str(message.text).split(" ")[1], "allow_message", "no") + send_admin_message("Успешно") + return -### --//-- -@bot.message_handler(commands=['start']) +### --//-- + + +@bot.message_handler(commands=["start"]) def start(message): - start_txt = 'Привет! Это бот, который будет кидать тебе сообщения перед нужной парой с номером кабинета/фамилией препода\ + start_txt = "Привет! Это бот, который будет кидать тебе сообщения перед нужной парой с номером кабинета/фамилией препода\ \n\nТы можешь настроить отправку сообщений в лс или добавить меня в группу, куда я буду присылать расписание\ -\n\nCreated by: @Kr0sH_512' - if message.chat.type == "supergroup": - if len(message.text.split(' ')) != 2: +\n\nCreated by: @Kr0sH_512" + if message.chat.type == "supergroup": + if len(message.text.split(" ")) != 2: send_message(message.chat.id, start_txt) - send_message(message.chat.id, 'Похоже, что этот бот добавлен в группу. Чтобы он присылал расписание в чат, \ + send_message( + message.chat.id, + "Похоже, что этот бот добавлен в группу. Чтобы он присылал расписание в чат, \ укажите вместе с /start@vmk_schedule_bot номер своей группы.\ - \nПример:\n/start@vmk_schedule_bot 102') + \nПример:\n/start@vmk_schedule_bot 102", + ) else: - temp = message.text.split(' ')[1] - + temp = message.text.split(" ")[1] + if temp.isdigit(): infos = [ - message.chat.id, - message.chat.title, - '', + message.chat.id, + message.chat.title, + "", message.from_user.username, - temp + temp, ] - + if temp in for_json.groups_in_json(): - send_message(message.chat.id, - 'Отлично! Теперь я буду присылать вам расписание {} группы'.format(temp)) + send_message( + message.chat.id, + "Отлично! Теперь я буду присылать вам расписание {} группы".format( + temp + ), + ) else: - send_message(message.chat.id, 'Похоже, что расписания для этой группы ещё не существует. \ - \nРазработчик уже пинается, но можете дополнительно написать ему: @Kr0sH_512') - - send_admin_message('В группе {} был запрос на {} группу.\ - \nid: {}'.format(message.chat.title, temp, message.chat.id)) - temp = 'other' + send_message( + message.chat.id, + "Похоже, что расписания для этой группы ещё не существует. \ + \nРазработчик уже пинается, но можете дополнительно написать ему: @Kr0sH_512", + ) + + send_admin_message( + "В группе {} был запрос на {} группу.\ + \nid: {}".format( + message.chat.title, temp, message.chat.id + ) + ) + temp = "other" for_json.save_user(infos) else: - send_message(message.chat.id, 'Введено не число. Попробуйте снова.\ - \nПример:\n/start@vmk_schedule_bot 102') + send_message( + message.chat.id, + "Введено не число. Попробуйте снова.\ + \nПример:\n/start@vmk_schedule_bot 102", + ) return - + send_message(message.chat.id, start_txt) - + markup = types.InlineKeyboardMarkup() for i in for_json.groups_in_json(): markup.add(types.InlineKeyboardButton(text=i, callback_data=i)) - - markup.add(types.InlineKeyboardButton(text="Моей группы нет в этом списке", callback_data="other")) - bot.send_message(message.chat.id, 'Пожалуйста, выбери свою группу', parse_mode=ParseMode.HTML, reply_markup=markup) - + + markup.add( + types.InlineKeyboardButton( + text="Моей группы нет в этом списке", callback_data="other" + ) + ) + bot.send_message( + message.chat.id, + "Пожалуйста, выбери свою группу", + parse_mode=ParseMode.HTML, + reply_markup=markup, + ) + return -@bot.callback_query_handler(func=lambda call: call.data in for_json.groups_in_json() or call.data == "other") + +@bot.callback_query_handler( + func=lambda call: call.data in for_json.groups_in_json() or call.data == "other" +) def callback_inline(call): infos = [ - call.from_user.id, - call.from_user.first_name, - call.from_user.last_name, - call.from_user.username, - call.data - ] - + call.from_user.id, + call.from_user.first_name, + call.from_user.last_name, + call.from_user.username, + call.data, + ] + for_json.save_user(infos) - - if call.data == 'other': - bot.edit_message_text(chat_id=call.message.chat.id, - message_id=call.message.message_id, - text='Используй команду /request \ + + if call.data == "other": + bot.edit_message_text( + chat_id=call.message.chat.id, + message_id=call.message.message_id, + text="Используй команду /request \ и напиши номер группы, которую хочешь добавить\ - \nПосле создания расписания для твоей группы, я пришлю тебе сообщение', - parse_mode=ParseMode.HTML) + \nПосле создания расписания для твоей группы, я пришлю тебе сообщение", + parse_mode=ParseMode.HTML, + ) else: - bot.edit_message_text(chat_id=call.message.chat.id, - message_id=call.message.message_id, - text='Отлично, теперь тебе будут приходить сообщения!', - parse_mode=ParseMode.HTML) - + bot.edit_message_text( + chat_id=call.message.chat.id, + message_id=call.message.message_id, + text="Отлично, теперь тебе будут приходить сообщения!", + parse_mode=ParseMode.HTML, + ) + return - -@bot.message_handler(commands=['schedule']) +@bot.message_handler(commands=["schedule"]) def send_schedule(message): - text = 'К сожалению, бот не может отправить тебе расписание твоей группы на сегодня :(' + text = ( + "К сожалению, бот не может отправить тебе расписание твоей группы на сегодня :(" + ) try: - day = datetime.today().strftime('%A').lower()[:3] # mon tue ... + day = datetime.today().strftime("%A").lower()[:3] # mon tue ... text = for_json.get_schedule(message.chat.id, day) except Exception as e: - send_admin_message('Schedule error from: {} \n\n {}'.format(message.chat.id, e)) - + send_admin_message("Schedule error from: {} \n\n {}".format(message.chat.id, e)) + markup = types.InlineKeyboardMarkup() - markup.add(types.InlineKeyboardButton(text="◀️", callback_data="left"), - types.InlineKeyboardButton(text="▶️", callback_data="right")) - bot.send_message(message.chat.id, text, ParseMode.HTML, reply_markup=markup, message_thread_id=message.message_thread_id) - + markup.add( + types.InlineKeyboardButton(text="◀️", callback_data="left"), + types.InlineKeyboardButton(text="▶️", callback_data="right"), + ) + bot.send_message( + message.chat.id, + text, + ParseMode.HTML, + reply_markup=markup, + message_thread_id=message.message_thread_id, + ) + return - -@bot.callback_query_handler(func=lambda call: call.data in ['left', 'right']) + + +@bot.callback_query_handler(func=lambda call: call.data in ["left", "right"]) def change_schedule(call): - days = ['mon', 'tue', 'wed', 'thu', 'fri', 'sat'] - russian_days = ['понедельник', 'вторник', 'среду', 'четверг', 'пятницу', 'субботу'] - i = 6 # выходной + days = ["mon", "tue", "wed", "thu", "fri", "sat"] + russian_days = ["понедельник", "вторник", "среду", "четверг", "пятницу", "субботу"] + i = 6 # выходной for day in russian_days: if day in call.message.text: i = russian_days.index(day) break - - delta = (1) if (call.data == 'right') else (-1) + + delta = (1) if (call.data == "right") else (-1) ind = ((i + delta) % 6) if (i + delta > 0) else (i + delta) - + text = for_json.get_schedule(call.message.chat.id, days[ind]) - + markup = types.InlineKeyboardMarkup() - markup.add(types.InlineKeyboardButton(text="◀️", callback_data="left"), - types.InlineKeyboardButton(text="▶️", callback_data="right")) - - bot.edit_message_text(text, call.message.chat.id, call.message.message_id, - parse_mode=ParseMode.HTML, reply_markup=markup) - + markup.add( + types.InlineKeyboardButton(text="◀️", callback_data="left"), + types.InlineKeyboardButton(text="▶️", callback_data="right"), + ) + + bot.edit_message_text( + text, + call.message.chat.id, + call.message.message_id, + parse_mode=ParseMode.HTML, + reply_markup=markup, + ) + return - -@bot.message_handler(commands=['random']) +@bot.message_handler(commands=["random"]) def random_people(message): list_of_num = ["1️⃣", "2️⃣", "3️⃣", "4️⃣", "5️⃣", "6", "7", "8", "9"] - + list_of_var = for_json.students_in_json(message.chat.id) - - text = 'Для вашей группы эта команда не настроена.\n\nНапиши запрос через /request' - + + text = "Для вашей группы эта команда не настроена.\n\nНапиши запрос через /request" + if list_of_var == None: send_message(message.chat.id, text) - send_admin_message('Запрос на random команду от {}'.format( - for_json.return_infos(message.chat.id)["group"])) - + send_admin_message( + "Запрос на random команду от {}".format( + for_json.return_infos(message.chat.id)["group"] + ) + ) + return - - text = '• Выбери как именно мне перемешать вас:\n' - + + text = "• Выбери как именно мне перемешать вас:\n" + l = [] for i, val in enumerate(list_of_var): - text += '\n{} {}'.format(list_of_num[i], val) - l.append(types.InlineKeyboardButton(text=list_of_num[i], callback_data="rand#" + val)) - + text += "\n{} {}".format(list_of_num[i], val) + l.append( + types.InlineKeyboardButton(text=list_of_num[i], callback_data="rand#" + val) + ) + l = [l] - + markup = types.InlineKeyboardMarkup(l) bot.send_message(message.chat.id, text, ParseMode.HTML, reply_markup=markup) - + return + @bot.callback_query_handler(func=lambda call: "rand#" in call.data) def make_random(call): - list_of_stud = for_json.students_in_json(call.message.chat.id, call.data.split("#")[1]) - + list_of_stud = for_json.students_in_json( + call.message.chat.id, call.data.split("#")[1] + ) + random.shuffle(list_of_stud) - - text = '{}:\n\n'.format(call.data.split("#")[1]) - + + text = "{}:\n\n".format(call.data.split("#")[1]) + for i, name in enumerate(list_of_stud): - text += '{}{}. {}\n'.format(' ' if i + 1 < 10 else '', i + 1, name) - - bot.edit_message_text('Хорошо, вот ваше распределение!', call.message.chat.id, - call.message.message_id, parse_mode=ParseMode.HTML) - + text += "{}{}. {}\n".format(" " if i + 1 < 10 else "", i + 1, name) + + bot.edit_message_text( + "Хорошо, вот ваше распределение!", + call.message.chat.id, + call.message.message_id, + parse_mode=ParseMode.HTML, + ) + send_message(call.message.chat.id, text) - + return - -@bot.message_handler(commands=['info']) + +@bot.message_handler(commands=["info"]) def send_info(message): - if len(message.text.split(' ')) == 2 and is_admin(message): - send_admin_message('Окей, держи настройки {}'.format(message.text.split(' ')[1])) - infos = for_json.return_infos(message.text.split(' ')[1]) - text = 'Никнейм: @{}\ + if len(message.text.split(" ")) == 2 and is_admin(message): + send_admin_message( + "Окей, держи настройки {}".format(message.text.split(" ")[1]) + ) + infos = for_json.return_infos(message.text.split(" ")[1]) + text = "Никнейм: @{}\ \nИмя: {} {}\ \nВыбранная группа: {}\ \nВремя напоминания до урока: {}\ - \nРазрешены ли напоминания: {}'.format(infos['username'], infos['first_name'], infos['last_name'], - (infos['group'] if infos['group'] != 'other' else 'не выбрана'), - infos['timeout'], - ('да' if infos['allow_message'] == 'yes' else 'нет')) + \nРазрешены ли напоминания: {}".format( + infos["username"], + infos["first_name"], + infos["last_name"], + (infos["group"] if infos["group"] != "other" else "не выбрана"), + infos["timeout"], + ("да" if infos["allow_message"] == "yes" else "нет"), + ) send_admin_message(text) return - - send_message(message.chat.id, 'Хорошо, вот твои настройки:') + + send_message(message.chat.id, "Хорошо, вот твои настройки:") infos = for_json.return_infos(message.chat.id) - text = 'Выбранная группа: {}\ + text = "Выбранная группа: {}\ \nВремя напоминания до урока: {}\ - \nРазрешены ли напоминания: {}'.format((infos['group'] if infos['group'] != 'other' else 'не выбрана'), - infos['timeout'], - ('да' if infos['allow_message'] == 'yes' else 'нет')) + \nРазрешены ли напоминания: {}".format( + (infos["group"] if infos["group"] != "other" else "не выбрана"), + infos["timeout"], + ("да" if infos["allow_message"] == "yes" else "нет"), + ) send_message(message.chat.id, text) - + return - -@bot.message_handler(commands=['pause']) +@bot.message_handler(commands=["pause"]) def pause_schedule(message): - infos = for_json.return_infos(message.chat.id)['allow_message'] - - if infos == 'yes': - for_json.change_user_param(str(message.chat.id), 'allow_message', 'no') - send_message(message.chat.id, 'Рассылка сообщений прекращена. \ - \nДля возобновления воспользуйтесь командой\n/pause') + infos = for_json.return_infos(message.chat.id)["allow_message"] + + if infos == "yes": + for_json.change_user_param(str(message.chat.id), "allow_message", "no") + send_message( + message.chat.id, + "Рассылка сообщений прекращена. \ + \nДля возобновления воспользуйтесь командой\n/pause", + ) else: - for_json.change_user_param(str(message.chat.id), 'allow_message', 'yes') - send_message(message.chat.id, 'Рассылка сообщений возоблена!') - + for_json.change_user_param(str(message.chat.id), "allow_message", "yes") + send_message(message.chat.id, "Рассылка сообщений возоблена!") + return - - - -@bot.message_handler(commands=['thread']) + + +@bot.message_handler(commands=["thread"]) def change_thread(message): - thread_id = '' - + thread_id = "" + try: thread_id = message.reply_to_message.message_thread_id except AttributeError: thread_id = "General" - - for_json.change_user_param(str(message.chat.id), 'thread', thread_id) - - send_message(message.chat.id, 'Хорошо, теперь я буду отправлять сообщения в этот чат', thread_id) - + + for_json.change_user_param(str(message.chat.id), "thread", thread_id) + + send_message( + message.chat.id, + "Хорошо, теперь я буду отправлять сообщения в этот чат", + thread_id, + ) + return - -@bot.message_handler(commands=['timeout']) +@bot.message_handler(commands=["timeout"]) def set_timeout(message): - send_message(message.chat.id, 'Пришли мне число от 1 до 60. \ - \nЭто будет количество минут, за которое я буду присылать тебе напоминание о паре (По умолчанию: 10)') + send_message( + message.chat.id, + "Пришли мне число от 1 до 60. \ + \nЭто будет количество минут, за которое я буду присылать тебе напоминание о паре (По умолчанию: 10)", + ) bot.register_next_step_handler(message, save_timeout) - + return + def save_timeout(message): - if str(message.text).isdigit() and int(message.text) <= 60 and int(message.text) >= 1: - for_json.change_user_param(message.chat.id, 'timeout', str(message.text)) - send_message(message.chat.id, - 'Хорошо, теперь ты будешь получать напоминания за {} минут до урока'.format(str(message.text))) + if ( + str(message.text).isdigit() + and int(message.text) <= 60 + and int(message.text) >= 1 + ): + for_json.change_user_param(message.chat.id, "timeout", str(message.text)) + send_message( + message.chat.id, + "Хорошо, теперь ты будешь получать напоминания за {} минут до урока".format( + str(message.text) + ), + ) else: - send_message(message.chat.id, - 'Пожалуйста, пришли число от 1 до 60. \ - \nПо умолчанию: 10') - send_message(message.chat.id, 'Изменения не были внесены') - + send_message( + message.chat.id, + "Пожалуйста, пришли число от 1 до 60. \ + \nПо умолчанию: 10", + ) + send_message(message.chat.id, "Изменения не были внесены") + return - -@bot.message_handler(commands=['request']) +@bot.message_handler(commands=["request"]) def request(message): - send_message(message.from_user.id, 'Отправь мне сообщение и я перешлю его разработчику\ - \n(или напиши stop, чтобы отменить отправку)') + send_message( + message.from_user.id, + "Отправь мне сообщение и я перешлю его разработчику\ + \n(или напиши stop, чтобы отменить отправку)", + ) bot.register_next_step_handler(message, send_request) - + return - + + def send_request(message): - if message.text == "stop" or message.text == "стоп" or message.text[0] == '/': - send_message(message.from_user.id, 'Отмена отпраки') + if message.text == "stop" or message.text == "стоп" or message.text[0] == "/": + send_message(message.from_user.id, "Отмена отпраки") return - + infos = [ - message.from_user.id, - message.from_user.first_name, - message.from_user.last_name, + message.from_user.id, + message.from_user.first_name, + message.from_user.last_name, message.from_user.username, - message.text + message.text, ] - + for i in range(len(infos)): if type(infos[i]) == type(None): - infos[i] = '' - - admin_text = 'Сообщение от {} {} (@{}):\n\n{}\nid: {}'.format(infos[1], infos[2], infos[3], infos[4], infos[0]) + infos[i] = "" + + admin_text = "Сообщение от {} {} (@{}):\n\n{}\nid: {}".format( + infos[1], infos[2], infos[3], infos[4], infos[0] + ) send_admin_message(admin_text) - - send_message(message.from_user.id, 'Ваше сообщение успешно доставлено!') - + + send_message(message.from_user.id, "Ваше сообщение успешно доставлено!") + return - -@bot.message_handler(commands=['source']) +@bot.message_handler(commands=["source"]) def send_source(message): - bot.send_message(message.chat.id, 'https://github.com/kr0sh512/Telegram-bot-schedule', parse_mode=ParseMode.HTML) - + bot.send_message( + message.chat.id, + "https://github.com/kr0sh512/Telegram-bot-schedule", + parse_mode=ParseMode.HTML, + ) + return - -@bot.message_handler(content_types=['text']) +@bot.message_handler(content_types=["text"]) def text_message(message): if message.chat.type == "supergroup" or is_admin(message): return - send_message(message.from_user.id, 'К сожалению, я ещё не умею обрабатывать сообщения.\ - \nИспользуй команды из меню или напиши /help.') - + send_message( + message.from_user.id, + "К сожалению, я ещё не умею обрабатывать сообщения.\ + \nИспользуй команды из меню или напиши /help.", + ) + return - -def send_message(id, text, thread_id='General', distribution=False): - if thread_id == 'General': +def send_message(id, text, thread_id="General", distribution=False): + if thread_id == "General": thread_id = None - + if distribution: count_of_weeks = (datetime.now() - datetime(2024, 2, 5)).days // 7 is_odd = (count_of_weeks + 1) % 2 == 1 - + if is_odd and ("(чёт)" in text): return - + if (not is_odd) and ("(нечёт)" in text): return - + try: - bot.send_message(chat_id=id, text=text, parse_mode=ParseMode.HTML, message_thread_id=thread_id,disable_web_page_preview=True) + bot.send_message( + chat_id=id, + text=text, + parse_mode=ParseMode.HTML, + message_thread_id=thread_id, + disable_web_page_preview=True, + ) except Exception as e: time.sleep(5) try: - bot.send_message(chat_id=id, text=text, parse_mode=ParseMode.HTML, message_thread_id=thread_id,disable_web_page_preview=True) + bot.send_message( + chat_id=id, + text=text, + parse_mode=ParseMode.HTML, + message_thread_id=thread_id, + disable_web_page_preview=True, + ) except Exception as e: - text_error = 'Error from user: @{} {}\n{}'.format(for_json.return_infos(id)["username"], id, str(e)) + text_error = "Error from user: @{} {}\n{}".format( + for_json.return_infos(id)["username"], id, str(e) + ) send_admin_message(text_error) print(id, e) - + return -def send_document(id, file, text = ''): + +def send_document(id, file, text=""): try: bot.send_document(id, file, caption=text) except Exception as e: @@ -562,26 +705,27 @@ def send_document(id, file, text = ''): try: bot.send_document(id, file, caption=text) except Exception as e: - text_error = 'Error from user: {}\n{}'.format(id, str(e)) + text_error = "Error from user: {}\n{}".format(id, str(e)) send_admin_message(text_error) print(id, e) - + return - -if __name__ == '__main__': +if __name__ == "__main__": for_json.create_schedule_tasks(True) - send_admin_message('Я перезапустился!') + send_admin_message("Я перезапустился!") print("-------------------------") - - threading.Thread(target=bot.infinity_polling, name='bot_infinity_polling', daemon=True).start() - + + threading.Thread( + target=bot.infinity_polling, name="bot_infinity_polling", daemon=True + ).start() + while True: try: schedule.run_pending() time.sleep(5) except Exception as e: print(e) - time.sleep(10) \ No newline at end of file + time.sleep(10)