From 9654615e4470b6d7228974477494ee415900e6ca Mon Sep 17 00:00:00 2001 From: kr0sh512 Date: Fri, 27 Oct 2023 17:57:21 +0300 Subject: [PATCH 1/8] first commit --- schedules.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/schedules.py b/schedules.py index 659f4f6..833dc24 100644 --- a/schedules.py +++ b/schedules.py @@ -255,9 +255,15 @@ def text_message(message): def send_message(id, text, thread_id='General'): if thread_id == 'General': thread_id = None - bot.send_message(id, text, parse_mode=ParseMode.HTML, message_thread_id=thread_id) + + try: + bot.send_message(chat_id=id, text=text, parse_mode=ParseMode.HTML, message_thread_id=thread_id) + except Exception as e: + print(id, e) + text = '🛑 user:' + id + 'error.\n' + str(e) + bot.send_message(chat_id=admin_id, text=text, parse_mode=ParseMode.HTML, message_thread_id=thread_id) - return + return if __name__ == '__main__': From c8fd747dd1c9759f1ee0ab70585904687a625367 Mon Sep 17 00:00:00 2001 From: kr0sh512 Date: Sat, 28 Oct 2023 16:57:13 +0300 Subject: [PATCH 2/8] intermediate code changes --- for_json.py | 51 ++++++++++++++++++++- schedules.py | 124 +++++++++++++++++++++++++++++++-------------------- 2 files changed, 126 insertions(+), 49 deletions(-) diff --git a/for_json.py b/for_json.py index 962cf50..aeaac4e 100644 --- a/for_json.py +++ b/for_json.py @@ -125,6 +125,7 @@ def check_group_in_json(number): schdl = {} with open(path_schedule, 'r', encoding='utf-8') as json_file: schdl = json.load(json_file) + return str(number) in schdl.keys() def return_infos(id): @@ -134,5 +135,53 @@ def return_infos(id): data = data[id] text = 'Выбранная группа: {}\ \nВремя напоминания до урока: {}\ - \nРазрешены ли напоминания: {}'.format(data['group'], data['timeout'], ('да' if data['allow_message'] == 'yes' else 'нет')) + \nРазрешены ли напоминания: {}'.format((data['group'] if data['group'] != 'other' else 'не выбрана'), + data['timeout'], + ('да' if data['allow_message'] == 'yes' else 'нет')) + + return text + +def get_schedule(id): + text = 'Расписание на сегодня:\n\n' + + schdl_today = {} + with open(path_schedule, 'r', encoding='utf-8') as schedule_file: + day = datetime.today().strftime('%A').lower()[:3] + + group = '' + with open(path_users, 'r', encoding='utf-8') as user_file: + group = json.load(user_file)[id]['group'] + + if (group == 'other'): + return 'У тебя не выбрана группа для рассылки сообщений' + + try: + schdl_today = json.load(schedule_file)[group][day] + except: + return 'Расписания твоей группы на сегодня нет' + + for i, lesson in schdl_today.items(): + if len(lesson["infos"].split('|')) == 2: + teacher = lesson["infos"].split('|')[0] + room = lesson["infos"].split('|')[1] + text += '•{} | {}\ + \n{}\ + \n{}'.format(i, room, lesson["name"], teacher) + + elif len(lesson["infos"].split('|')) == 4: + infos = lesson["infos"].split('|') + text += '•{}\ + \n{}\ + \n({}) {}\ + \n({}) {}'.format(i, lesson["name"], + infos[1], + infos[0], + infos[3], + infos[2]) + else: + text += '•{}\ + \n{}'.format(i, lesson["name"]) + + text += '\n\n' + return text diff --git a/schedules.py b/schedules.py index 833dc24..d327a50 100644 --- a/schedules.py +++ b/schedules.py @@ -9,37 +9,41 @@ bot = telebot.TeleBot("TOKEN_API") admin_id = '856850518' start_txt = 'Привет! Это бот, который будет кидать тебе сообщения перед нужной парой с номером кабинета/фамилией препода\ -\n\nCreated by: @Kr0sH\_512' + \n\nТы можешь настроить отправку сообщений в лс или добавить меня в группу, куда я буду присылать расписание\ +\n\nCreated by: @Kr0sH_512' @bot.message_handler(commands=['update']) def update_schedules(message): if (str(message.from_user.id) != admin_id): - bot.send_message(message.from_user.id, 'Команда доступна только администратору', parse_mode='Markdown') - else: - for_json.create_schedule_tasks() + bot.send_message(message.from_user.id, 'Команда доступна только администратору', parse_mode=ParseMode.HTML) + return + + for_json.create_schedule_tasks() return @bot.message_handler(commands=['restart']) def restart_bot(message): if (str(message.from_user.id) != admin_id): - bot.send_message(message.from_user.id, 'Команда доступна только администратору', parse_mode='Markdown') - else: - bot.send_message(admin_id, '🛑 bye', parse_mode=ParseMode.HTML) - os.execv(sys.executable, ['python'] + sys.argv) - + bot.send_message(message.from_user.id, 'Команда доступна только администратору', parse_mode=ParseMode.HTML) + return + + bot.send_message(admin_id, '🛑 bye', parse_mode=ParseMode.HTML) + os.execv(sys.executable, ['python'] + sys.argv) + return @bot.message_handler(commands=['json']) def send_json(message): if (str(message.from_user.id) != admin_id): - bot.send_message(message.from_user.id, 'Команда доступна только администратору', parse_mode='Markdown') - else: - with open(for_json.path_users, 'rb') as json_file: - bot.send_document(admin_id, json_file) - with open(for_json.path_schedule, 'rb') as json_file: - bot.send_document(admin_id, json_file) + bot.send_message(message.from_user.id, 'Команда доступна только администратору', parse_mode=ParseMode.HTML) + return + + with open(for_json.path_users, 'rb') as json_file: + bot.send_document(admin_id, json_file) + with open(for_json.path_schedule, 'rb') as json_file: + bot.send_document(admin_id, json_file) return @@ -54,8 +58,8 @@ def start(message): if len(message.text.split(' ')) != 2: bot.send_message(message.chat.id, start_txt, parse_mode='Markdown') bot.send_message(message.chat.id, 'Похоже, что этот бот добавлен в группу. Чтобы он присылал расписание в чат, \ - запустите его заново, указав вместе с /start@vmk_schedule_bot номер\_своей\_группы.\ - \nПример:\n/start@vmk_schedule_bot 102', parse_mode='Markdown') + укажите вместе с /start@vmk_schedule_bot номер своей группы.\ + \nПример:\n/start@vmk_schedule_bot 102', parse_mode=ParseMode.HTML) else: temp = message.text.split(' ')[1] @@ -70,27 +74,28 @@ def start(message): if for_json.check_group_in_json(temp): bot.send_message(message.chat.id, \ - 'Отлично! Теперь я буду присылать вам расписание {} группы'.format(temp), parse_mode='Markdown') + 'Отлично! Теперь я буду присылать вам расписание {} группы'.format(temp), parse_mode=ParseMode.HTML) else: bot.send_message(message.chat.id, 'Похоже, что расписания для этой группы ещё не существует. \ - \nРазработчик уже пинается, но можете дополнительно написать ему: @Kr0sH\_512', parse_mode='Markdown') + \nРазработчик уже пинается, но можете дополнительно написать ему: @Kr0sH_512', parse_mode=ParseMode.HTML) send_message(admin_id, '🛑 В группе {} был запрос на {} группу.\ \nid: {}'.format(message.chat.title, temp, message.chat.id)) temp = 'other' for_json.save_user(infos) else: bot.send_message(message.chat.id, 'Введено не число. Попробуйте снова.\ - \nПример:\n/start@vmk_schedule_bot 102', parse_mode='Markdown') + \nПример:\n/start@vmk_schedule_bot 102', parse_mode=ParseMode.HTML) return - bot.send_message(message.chat.id, start_txt, parse_mode='Markdown') + bot.send_message(message.chat.id, start_txt, parse_mode=ParseMode.HTML) markup = types.InlineKeyboardMarkup() markup.add(types.InlineKeyboardButton(text="102", callback_data="102")) markup.add(types.InlineKeyboardButton(text="103", callback_data="103")) markup.add(types.InlineKeyboardButton(text="105", callback_data="105")) + markup.add(types.InlineKeyboardButton(text="110", callback_data="110")) markup.add(types.InlineKeyboardButton(text="Моей группы нет в этом списке", callback_data="other")) - bot.send_message(message.chat.id, 'Пожалуйста, выбери свою группу', parse_mode='Markdown', reply_markup=markup) + bot.send_message(message.chat.id, 'Пожалуйста, выбери свою группу', parse_mode=ParseMode.HTML, reply_markup=markup) return @@ -103,6 +108,7 @@ def callback_inline(call): call.from_user.username, call.data ] + for_json.save_user(infos) if call.data == 'other': @@ -111,12 +117,12 @@ def callback_inline(call): text='Используй команду /request \ и напиши номер группы, которую хочешь добавить\ \nПосле создания расписания для твоей группы, я пришлю тебе сообщение', - parse_mode='Markdown') + parse_mode=ParseMode.HTML) else: bot.edit_message_text(chat_id=call.message.chat.id, message_id=call.message.message_id, text='Отлично, теперь тебе будут приходить сообщения!', - parse_mode='Markdown') + parse_mode=ParseMode.HTML) return @@ -124,35 +130,38 @@ def callback_inline(call): def help(message): help_msg = 'Мои команды:\ \n/start - используй, чтобы сменить номер группы.\ + \n/schedule - используй, чтобы получить расписание на сегодня.\ \n/info - используй, чтобы узнать твои настройки бота\ \n/pause - используй, чтобы прекратить получать сообщения от бота\ \n/thread - используй в нужном чате канала, чтобы бот отправлял сообщения именно туда\ \n/timeout - настрой время, когда бот будет присылать тебе сообщение\ \n/request - используй, чтобы отправить разработчику какой-то запрос\ \n/source - Страница бота на Github\ - \n\nИли же можно всегда написать напрямую: @Kr0sH\_512' - bot.send_message(message.chat.id, help_msg, parse_mode='Markdown') + \n\nИли же можно всегда написать напрямую: @Kr0sH_512' + + bot.send_message(message.chat.id, help_msg, parse_mode=ParseMode.HTML) + if str(message.from_user.id) == admin_id and message.from_user.id == message.chat.id: help_msg = '🛑 И команды только для админа:\ \n/restart - перезапуск бота.\ \n/update - обновление schedule задач\ \n/json - получить файл пользователей и расписания\ - \n/info id\_пользователя - узнать настройки пользователя' - bot.send_message(message.from_user.id, help_msg, parse_mode='Markdown') + \n/info id_пользователя - узнать настройки пользователя' + bot.send_message(message.from_user.id, help_msg, parse_mode=ParseMode.HTML) return @bot.message_handler(commands=['request']) def request(message): bot.send_message(message.from_user.id, 'Отправь мне сообщение и я перешлю его разработчику\ - \n(или напиши stop, чтобы отменить отправку)', parse_mode='Markdown') + \n(или напиши stop, чтобы отменить отправку)', parse_mode=ParseMode.HTML) bot.register_next_step_handler(message, send_request) return def send_request(message): if message.text == "stop" or message.text == "стоп" or message.text[0] == '/': - bot.send_message(message.from_user.id, 'Отмена отпраки', parse_mode='Markdown') + bot.send_message(message.from_user.id, 'Отмена отпраки', parse_mode=ParseMode.HTML) return infos = [ @@ -166,51 +175,63 @@ def send_request(message): for i in range(len(infos)): if type(infos[i]) == type(None): infos[i] = '' - if '_' in str(infos[i]): - infos[i] = str(infos[i]).replace('_', '\_') text = '🛑 Сообщение от {} {} (@{}):\n\n{}\nid: {}'.format(infos[1], infos[2], infos[3], infos[4], infos[0]) - bot.send_message(admin_id, text, parse_mode='Markdown') - bot.send_message(message.from_user.id, 'Ваше сообщение успешно доставлено!', parse_mode='Markdown') + bot.send_message(admin_id, text, parse_mode=ParseMode.HTML) + + bot.send_message(message.from_user.id, 'Ваше сообщение успешно доставлено!', parse_mode=ParseMode.HTML) return -@bot.message_handler(commands=['schedule']) ### На доработке +@bot.message_handler(commands=['schedule']) def send_schedule(message): - # команда для отправки всего расписания - bot.send_message(message.chat.id, 'Пожалуйста, притворись, что ты получил сейчас своё расписание на неделю\ - \n\n(Функция в разработке)', parse_mode='Markdown') + text = 'К сожалению, бот не может отправить тебе расписание твоей группы на сегодня :(' + try: + text = for_json.get_schedule(str(message.chat.id)) + except Exception as e: + bot.send_message(admin_id, + '🛑 schedule error from: {} \n\n {}'.format(message.chat.id, e), + parse_mode=ParseMode.HTML) + bot.send_message(message.chat.id, text, parse_mode=ParseMode.HTML) return @bot.message_handler(commands=['timeout']) def set_timeout(message): bot.send_message(message.chat.id, 'Пришли мне число от 1 до 60. \ - \nЭто будет количество минут, за которое я буду присылать тебе напоминание о уроке (По умолчанию: 10)', parse_mode='Markdown') + \nЭто будет количество минут, за которое я буду присылать тебе напоминание о уроке (По умолчанию: 10)', parse_mode=ParseMode.HTML) bot.register_next_step_handler(message, save_timeout) return def save_timeout(message): if message.text[0] == '/': + bot.send_message(message.chat.id, 'Изменения не были внесены', parse_mode=ParseMode.HTML) return if str(message.text).isdigit() and int(message.text) <= 60 and int(message.text) >= 1: for_json.change_user_param(str(message.chat.id), 'timeout', str(message.text)) - bot.send_message(message.chat.id, 'Хорошо, теперь ты будешь получать напоминания за {} минут до урока'.format(str(message.text)), parse_mode='Markdown') + bot.send_message(message.chat.id, + 'Хорошо, теперь ты будешь получать напоминания за {} минут до урока'.format(str(message.text)), + parse_mode=ParseMode.HTML) else: - bot.send_message(message.chat.id, 'Пожалуйста, пришли число от 1 до 60. \nПо умолчанию: 10', parse_mode='Markdown') - bot.register_next_step_handler(message, save_timeout) + bot.send_message(message.chat.id, + 'Пожалуйста, пришли число от 1 до 60. \ + \nПо умолчанию: 10', + parse_mode=ParseMode.HTML) + bot.send_message(message.chat.id, 'Изменения не были внесены', parse_mode=ParseMode.HTML) return @bot.message_handler(commands=['info']) def send_info(message): if len(message.text.split(' ')) == 2 and str(message.from_user.id) == admin_id: - bot.send_message(message.chat.id, '🛑 Окей, держи', parse_mode='Markdown') + bot.send_message(message.chat.id, '🛑 Окей, держи настройки {}'.format(message.text.split(' ')[1]), + parse_mode=ParseMode.HTML) text = for_json.return_infos(message.text.split(' ')[1]) bot.send_message(message.chat.id, text, parse_mode=ParseMode.HTML) return - bot.send_message(message.chat.id, 'Хорошо, вот твои настройки:', parse_mode='Markdown') + + bot.send_message(message.chat.id, 'Хорошо, вот твои настройки:', parse_mode=ParseMode.HTML) text = for_json.return_infos(str(message.chat.id)) bot.send_message(message.chat.id, text, parse_mode=ParseMode.HTML) @@ -218,18 +239,21 @@ def send_info(message): @bot.message_handler(commands=['source']) def send_source(message): - bot.send_message(message.chat.id, 'https://github.com/kr0sh512/Telegram-bot-schedule', parse_mode='Markdown') + bot.send_message(message.chat.id, 'https://github.com/kr0sh512/Telegram-bot-schedule', parse_mode=ParseMode.HTML) return @bot.message_handler(commands=['thread']) def change_thread(message): 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) + bot.send_message(message.chat.id, 'Хорошо, теперь я буду отправлять сообщения в этот чат', parse_mode='Markdown', message_thread_id=thread_id) @@ -238,6 +262,7 @@ def change_thread(message): @bot.message_handler(commands=['pause']) def pause_schedule(message): for_json.change_user_param(str(message.chat.id), 'allow_message', 'no') + bot.send_message(message.chat.id, 'Рассылка сообщений преращена. \ \nДля возобновления воспользуйтесь командой\n/start', parse_mode='Markdown') @@ -259,9 +284,13 @@ def send_message(id, text, thread_id='General'): try: bot.send_message(chat_id=id, text=text, parse_mode=ParseMode.HTML, message_thread_id=thread_id) except Exception as e: - print(id, e) text = '🛑 user:' + id + 'error.\n' + str(e) - bot.send_message(chat_id=admin_id, text=text, parse_mode=ParseMode.HTML, message_thread_id=thread_id) + bot.send_message(admin_id, text, parse_mode=ParseMode.HTML, message_thread_id=thread_id) + time.sleep(5) + try: + bot.send_message(chat_id=id, text=text, parse_mode=ParseMode.HTML, message_thread_id=thread_id) + except Exception as e: + print(id, e) return @@ -275,7 +304,6 @@ if __name__ == '__main__': while True: try: - # bot.polling(none_stop=True) schedule.run_pending() time.sleep(5) except Exception as e: From 81febd3e1a56ae0931287505da54a8984e1d6e50 Mon Sep 17 00:00:00 2001 From: kr0sh512 Date: Tue, 31 Oct 2023 10:51:39 +0300 Subject: [PATCH 3/8] . --- schedules.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/schedules.py b/schedules.py index d327a50..5a5055c 100644 --- a/schedules.py +++ b/schedules.py @@ -284,8 +284,9 @@ def send_message(id, text, thread_id='General'): try: bot.send_message(chat_id=id, text=text, parse_mode=ParseMode.HTML, message_thread_id=thread_id) except Exception as e: - text = '🛑 user:' + id + 'error.\n' + str(e) - bot.send_message(admin_id, text, parse_mode=ParseMode.HTML, message_thread_id=thread_id) + text_error = '🛑 user:' + id + 'error.\n' + str(e) + bot.send_message(admin_id, text_error, parse_mode=ParseMode.HTML, message_thread_id=thread_id) + print(id, e) time.sleep(5) try: bot.send_message(chat_id=id, text=text, parse_mode=ParseMode.HTML, message_thread_id=thread_id) From 38adcea5658fcc18cf9d001d312153fdfe96942c Mon Sep 17 00:00:00 2001 From: kr0sh512 Date: Thu, 2 Nov 2023 16:18:50 +0300 Subject: [PATCH 4/8] code refactoring. add "admin" file --- admin.py | 28 +++++++ for_json.py | 19 +++-- schedules.py | 212 ++++++++++++++++++++++++++++++--------------------- 3 files changed, 164 insertions(+), 95 deletions(-) create mode 100644 admin.py diff --git a/admin.py b/admin.py new file mode 100644 index 0000000..bffd9e4 --- /dev/null +++ b/admin.py @@ -0,0 +1,28 @@ +import schedules as bot + +admin_id = '856850518' + +def admin_command(func): + def wrapper(*args, **kwargs): + message = args[0] + if str(message.from_user.id) != admin_id: + bot.send_message(message.from_user.id, 'Команда доступна только администратору') + return + + func(*args, **kwargs) + return + + return wrapper + +def is_admin(message): + if str(message.from_user.id) == admin_id and message.from_user.id == message.chat.id: + return True + + return False + +def send_admin_message(text): + bot.send_message(admin_id, '🛑 ' + text) + +def send_admin_document(file): + bot.send_document(admin_id, file, '🛑 Admin file') + \ No newline at end of file diff --git a/for_json.py b/for_json.py index aeaac4e..a917c4e 100644 --- a/for_json.py +++ b/for_json.py @@ -11,6 +11,7 @@ def save_user(infos): if type(infos[i]) == type(None): infos[i] = '' infos[i] = str(infos[i]) + infos = { "id" : infos[0], "first_name" : infos[1], @@ -39,6 +40,7 @@ def save_user(infos): return def change_user_param(id, key, value): + id = str(id) data = {} with open(path_users, 'r', encoding='utf-8') as json_file: data = json.load(json_file) @@ -117,7 +119,6 @@ def create_schedule_tasks(): elif day == 'sat': schedule.every().saturday.at(start_task).do(bot.send_message, id, text, thread_id) - bot.send_message(bot.admin_id, '🛑 Произошёл update') return @@ -129,19 +130,21 @@ def check_group_in_json(number): return str(number) in schdl.keys() def return_infos(id): + id = str(id) data = {} with open(path_users, 'r', encoding='utf-8') as json_file: data = json.load(json_file) data = data[id] - text = 'Выбранная группа: {}\ - \nВремя напоминания до урока: {}\ - \nРазрешены ли напоминания: {}'.format((data['group'] if data['group'] != 'other' else 'не выбрана'), - data['timeout'], - ('да' if data['allow_message'] == 'yes' else 'нет')) - - return text + + return data + +def pause_bot(): + schedule.clear() + + return def get_schedule(id): + id = str(id) text = 'Расписание на сегодня:\n\n' schdl_today = {} diff --git a/schedules.py b/schedules.py index 5a5055c..03e0047 100644 --- a/schedules.py +++ b/schedules.py @@ -4,62 +4,79 @@ import os, sys from telebot import types 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") -admin_id = '856850518' start_txt = 'Привет! Это бот, который будет кидать тебе сообщения перед нужной парой с номером кабинета/фамилией препода\ \n\nТы можешь настроить отправку сообщений в лс или добавить меня в группу, куда я буду присылать расписание\ \n\nCreated by: @Kr0sH_512' +@bot.message_handler(commands=['test']) +@admin_command +def update_schedules(message): + text = 'Жирный текст\ + \nКурсивный текст\ + \nПеречёркнутый текст\ + \nСсылка\ + \nМоноширинный текст\ + \n
Форматированный с сохранением     пробелов
\ + \n
Цитата
' + send_admin_message(text) + + return @bot.message_handler(commands=['update']) +@admin_command def update_schedules(message): - if (str(message.from_user.id) != admin_id): - bot.send_message(message.from_user.id, 'Команда доступна только администратору', parse_mode=ParseMode.HTML) - return - for_json.create_schedule_tasks() - + send_admin_message('Произошёл update') + return @bot.message_handler(commands=['restart']) +@admin_command def restart_bot(message): - if (str(message.from_user.id) != admin_id): - bot.send_message(message.from_user.id, 'Команда доступна только администратору', parse_mode=ParseMode.HTML) - return - - bot.send_message(admin_id, '🛑 bye', parse_mode=ParseMode.HTML) + send_admin_message('bye') os.execv(sys.executable, ['python'] + sys.argv) + +@bot.message_handler(commands=['json']) +@admin_command +def send_json(message): + 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: + send_admin_document(json_file) + + return + +@bot.message_handler(commands=['pause_all']) +@admin_command +def pause_bot(message): + for_json.pause_bot() + send_admin_message('Бот больше не отправляет расписание') return - -@bot.message_handler(commands=['json']) -def send_json(message): - if (str(message.from_user.id) != admin_id): - bot.send_message(message.from_user.id, 'Команда доступна только администратору', parse_mode=ParseMode.HTML) - return - with open(for_json.path_users, 'rb') as json_file: - bot.send_document(admin_id, json_file) - with open(for_json.path_schedule, 'rb') as json_file: - bot.send_document(admin_id, json_file) - +@bot.message_handler(commands=['stop']) +@admin_command +def pause_bot(message): + 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('Успешно') return @bot.message_handler(commands=['start']) def start(message): - if message.chat.type == "supergroup": - print(message.text) - print(message.chat.id) - print(message.text.split(' ')) - + if message.chat.type == "supergroup": if len(message.text.split(' ')) != 2: - bot.send_message(message.chat.id, start_txt, parse_mode='Markdown') - bot.send_message(message.chat.id, 'Похоже, что этот бот добавлен в группу. Чтобы он присылал расписание в чат, \ + send_message(message.chat.id, start_txt) + send_message(message.chat.id, 'Похоже, что этот бот добавлен в группу. Чтобы он присылал расписание в чат, \ укажите вместе с /start@vmk_schedule_bot номер своей группы.\ - \nПример:\n/start@vmk_schedule_bot 102', parse_mode=ParseMode.HTML) + \nПример:\n/start@vmk_schedule_bot 102') else: temp = message.text.split(' ')[1] @@ -73,22 +90,24 @@ def start(message): ] if for_json.check_group_in_json(temp): - bot.send_message(message.chat.id, \ - 'Отлично! Теперь я буду присылать вам расписание {} группы'.format(temp), parse_mode=ParseMode.HTML) + send_message(message.chat.id, + 'Отлично! Теперь я буду присылать вам расписание {} группы'.format(temp)) else: - bot.send_message(message.chat.id, 'Похоже, что расписания для этой группы ещё не существует. \ - \nРазработчик уже пинается, но можете дополнительно написать ему: @Kr0sH_512', parse_mode=ParseMode.HTML) - send_message(admin_id, '🛑 В группе {} был запрос на {} группу.\ + 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: - bot.send_message(message.chat.id, 'Введено не число. Попробуйте снова.\ - \nПример:\n/start@vmk_schedule_bot 102', parse_mode=ParseMode.HTML) + send_message(message.chat.id, 'Введено не число. Попробуйте снова.\ + \nПример:\n/start@vmk_schedule_bot 102') return - bot.send_message(message.chat.id, start_txt, parse_mode=ParseMode.HTML) + send_message(message.chat.id, start_txt) + markup = types.InlineKeyboardMarkup() markup.add(types.InlineKeyboardButton(text="102", callback_data="102")) markup.add(types.InlineKeyboardButton(text="103", callback_data="103")) @@ -139,29 +158,31 @@ def help(message): \n/source - Страница бота на Github\ \n\nИли же можно всегда написать напрямую: @Kr0sH_512' - bot.send_message(message.chat.id, help_msg, parse_mode=ParseMode.HTML) + send_message(message.chat.id, help_msg) - if str(message.from_user.id) == admin_id and message.from_user.id == message.chat.id: - help_msg = '🛑 И команды только для админа:\ + if is_admin(message): + admin_help_msg = 'И команды только для админа:\ \n/restart - перезапуск бота.\ \n/update - обновление schedule задач\ \n/json - получить файл пользователей и расписания\ - \n/info id_пользователя - узнать настройки пользователя' - bot.send_message(message.from_user.id, help_msg, parse_mode=ParseMode.HTML) + \n/info id_пользователя - узнать настройки пользователя\ + \n/pause_all - приостановить бота для всех (каникулы/выходные)\ + \n/stop id_пользователя - приостановить отправку сообщений для пользователя' + send_admin_message(admin_help_msg) return @bot.message_handler(commands=['request']) def request(message): - bot.send_message(message.from_user.id, 'Отправь мне сообщение и я перешлю его разработчику\ - \n(или напиши stop, чтобы отменить отправку)', parse_mode=ParseMode.HTML) + 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] == '/': - bot.send_message(message.from_user.id, 'Отмена отпраки', parse_mode=ParseMode.HTML) + send_message(message.from_user.id, 'Отмена отпраки') return infos = [ @@ -176,10 +197,10 @@ def send_request(message): if type(infos[i]) == type(None): infos[i] = '' - text = '🛑 Сообщение от {} {} (@{}):\n\n{}\nid: {}'.format(infos[1], infos[2], infos[3], infos[4], infos[0]) - bot.send_message(admin_id, text, parse_mode=ParseMode.HTML) + admin_text = 'Сообщение от {} {} (@{}):\n\n{}\nid: {}'.format(infos[1], infos[2], infos[3], infos[4], infos[0]) + send_admin_message(admin_text) - bot.send_message(message.from_user.id, 'Ваше сообщение успешно доставлено!', parse_mode=ParseMode.HTML) + send_message(message.from_user.id, 'Ваше сообщение успешно доставлено!') return @@ -189,51 +210,56 @@ def send_schedule(message): try: text = for_json.get_schedule(str(message.chat.id)) except Exception as e: - bot.send_message(admin_id, - '🛑 schedule error from: {} \n\n {}'.format(message.chat.id, e), - parse_mode=ParseMode.HTML) - bot.send_message(message.chat.id, text, parse_mode=ParseMode.HTML) + send_admin_message('Schedule error from: {} \n\n {}'.format(message.chat.id, e)) + send_message(message.chat.id, text) return @bot.message_handler(commands=['timeout']) def set_timeout(message): - bot.send_message(message.chat.id, 'Пришли мне число от 1 до 60. \ - \nЭто будет количество минут, за которое я буду присылать тебе напоминание о уроке (По умолчанию: 10)', parse_mode=ParseMode.HTML) + send_message(message.chat.id, 'Пришли мне число от 1 до 60. \ + \nЭто будет количество минут, за которое я буду присылать тебе напоминание о уроке (По умолчанию: 10)') bot.register_next_step_handler(message, save_timeout) return def save_timeout(message): - if message.text[0] == '/': - bot.send_message(message.chat.id, 'Изменения не были внесены', parse_mode=ParseMode.HTML) - return if str(message.text).isdigit() and int(message.text) <= 60 and int(message.text) >= 1: - for_json.change_user_param(str(message.chat.id), 'timeout', str(message.text)) - bot.send_message(message.chat.id, - 'Хорошо, теперь ты будешь получать напоминания за {} минут до урока'.format(str(message.text)), - parse_mode=ParseMode.HTML) + for_json.change_user_param(message.chat.id, 'timeout', str(message.text)) + send_message(message.chat.id, + 'Хорошо, теперь ты будешь получать напоминания за {} минут до урока'.format(str(message.text))) else: - bot.send_message(message.chat.id, + send_message(message.chat.id, 'Пожалуйста, пришли число от 1 до 60. \ - \nПо умолчанию: 10', - parse_mode=ParseMode.HTML) - bot.send_message(message.chat.id, 'Изменения не были внесены', parse_mode=ParseMode.HTML) + \nПо умолчанию: 10') + send_message(message.chat.id, 'Изменения не были внесены') return @bot.message_handler(commands=['info']) def send_info(message): - if len(message.text.split(' ')) == 2 and str(message.from_user.id) == admin_id: - bot.send_message(message.chat.id, '🛑 Окей, держи настройки {}'.format(message.text.split(' ')[1]), - parse_mode=ParseMode.HTML) - text = for_json.return_infos(message.text.split(' ')[1]) - bot.send_message(message.chat.id, text, parse_mode=ParseMode.HTML) + 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 'нет')) + send_admin_message(text) return - bot.send_message(message.chat.id, 'Хорошо, вот твои настройки:', parse_mode=ParseMode.HTML) - text = for_json.return_infos(str(message.chat.id)) - bot.send_message(message.chat.id, text, parse_mode=ParseMode.HTML) + send_message(message.chat.id, 'Хорошо, вот твои настройки:') + infos = for_json.return_infos(message.chat.id) + text = 'Выбранная группа: {}\ + \nВремя напоминания до урока: {}\ + \nРазрешены ли напоминания: {}'.format((infos['group'] if infos['group'] != 'other' else 'не выбрана'), + infos['timeout'], + ('да' if infos['allow_message'] == 'yes' else 'нет')) + send_message(message.chat.id, text) return @@ -254,8 +280,7 @@ def change_thread(message): for_json.change_user_param(str(message.chat.id), 'thread', thread_id) - bot.send_message(message.chat.id, 'Хорошо, теперь я буду отправлять сообщения в этот чат', - parse_mode='Markdown', message_thread_id=thread_id) + send_message(message.chat.id, 'Хорошо, теперь я буду отправлять сообщения в этот чат', thread_id) return @@ -263,8 +288,8 @@ def change_thread(message): def pause_schedule(message): for_json.change_user_param(str(message.chat.id), 'allow_message', 'no') - bot.send_message(message.chat.id, 'Рассылка сообщений преращена. \ - \nДля возобновления воспользуйтесь командой\n/start', parse_mode='Markdown') + send_message(message.chat.id, 'Рассылка сообщений преращена. \ + \nДля возобновления воспользуйтесь командой\n/start') return @@ -272,8 +297,8 @@ def pause_schedule(message): def text_message(message): if message.chat.type == "supergroup": return - bot.send_message(message.from_user.id, 'К сожалению, я ещё не умею обрабатывать сообщения.\ - \nИспользуй команды из меню или напиши /help.', parse_mode='Markdown') + send_message(message.from_user.id, 'К сожалению, я ещё не умею обрабатывать сообщения.\ + \nИспользуй команды из меню или напиши /help.') return @@ -282,23 +307,36 @@ def send_message(id, text, thread_id='General'): thread_id = None try: - bot.send_message(chat_id=id, text=text, parse_mode=ParseMode.HTML, message_thread_id=thread_id) + 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 = '🛑 user:' + id + 'error.\n' + str(e) - bot.send_message(admin_id, text_error, parse_mode=ParseMode.HTML, message_thread_id=thread_id) - print(id, e) time.sleep(5) try: - bot.send_message(chat_id=id, text=text, parse_mode=ParseMode.HTML, message_thread_id=thread_id) + 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(id, str(e)) + send_admin_message(text_error) + print(id, e) + + return + +def send_document(id, file, text = ''): + try: + bot.send_document(id, file, caption=text) + except Exception as e: + time.sleep(5) + try: + bot.send_document(id, file, caption=text) + except Exception as e: + text_error = 'Error from user: {}\n{}'.format(id, str(e)) + send_admin_message(text_error) print(id, e) return - if __name__ == '__main__': for_json.create_schedule_tasks() - send_message(admin_id, '🛑 я перезапустился!') + + send_admin_message('Я перезапустился!') print("-------------------------") threading.Thread(target=bot.infinity_polling, name='bot_infinity_polling', daemon=True).start() From f92d4b6320f8c8b82a3a0877f5caa548336407d3 Mon Sep 17 00:00:00 2001 From: kr0sh512 Date: Sat, 11 Nov 2023 01:44:46 +0300 Subject: [PATCH 5/8] add buttons for changing schedule day in /schedule --- for_json.py | 114 ++++++++++++++++++++++++++------------------------- schedules.py | 51 ++++++++++++++++++----- 2 files changed, 98 insertions(+), 67 deletions(-) diff --git a/for_json.py b/for_json.py index a917c4e..dadfa7c 100644 --- a/for_json.py +++ b/for_json.py @@ -6,6 +6,8 @@ from datetime import datetime path_users = "json/users.json" # Нужный путь до json файлов path_schedule = "json/schedule.json" +allow_update = True + def save_user(infos): for i in range(len(infos)): if type(infos[i]) == type(None): @@ -52,7 +54,44 @@ def change_user_param(id, key, value): create_schedule_tasks() return -def create_schedule_tasks(): +def parse_lesson(time, lesson): + text = '' + start, end = time.split("-") + + if len(lesson["infos"].split('|')) == 2: + teacher = lesson["infos"].split('|')[0] + room = lesson["infos"].split('|')[1] + text = '{}-{} | {}\ + \n{}\ + \n{}'.format(start, end, room, lesson["name"], teacher) + elif len(lesson["infos"].split('|')) == 4: + infos = lesson["infos"].split('|') + text = '{}-{}\ + \n{}\ + \n({}) {}\ + \n({}) {}'.format(start, end, lesson["name"], + infos[1], + infos[0], + infos[3], + infos[2]) + elif len(lesson["infos"].split('|')) == 1: + text = '{}-{}\ + \n{}'.format(start, end, lesson["name"]) + else: + print("что-то не так") + print(lesson["name"]) + + return text + +def create_schedule_tasks(manual=False): + global allow_update + if not manual: + bot.send_admin_message('Произошёл auto-update') + if not allow_update: + return + + allow_update = True + schedule.clear() users = {} schdl = {} @@ -72,33 +111,14 @@ def create_schedule_tasks(): if lesson["name"] == "": # Неправильный формат schedule continue - start, end = i.split("-") - text = '' - if len(lesson["infos"].split('|')) == 2: - teacher = lesson["infos"].split('|')[0] - room = lesson["infos"].split('|')[1] - text = '{}-{} | {}\ - \n{}\ - \n{}'.format(start, end, room, lesson["name"], teacher) - elif len(lesson["infos"].split('|')) == 4: - infos = lesson["infos"].split('|') - text = '{}-{}\ - \n{}\ - \n({}) {}\ - \n({}) {}'.format(start, end, lesson["name"], - infos[1], - infos[0], - infos[3], - infos[2]) - elif len(lesson["infos"].split('|')) == 1: - text = '{}-{}\ - \n{}'.format(start, end, lesson["name"]) - else: #неправильный формат infos в json файле - print("что-то не так") - print(lesson["name"]) + start = i.split("-")[0] + text = parse_lesson(i, lesson) + + if text == '': continue thread_id = params["thread"] + delta = '00:' + params["timeout"] format = '%H:%M' start_task = datetime.strptime(start, format) - datetime.strptime(delta, format) @@ -106,6 +126,7 @@ def create_schedule_tasks(): for i in str(start_task).split(':'): tmp += (('0' + i ) if len(i) < 2 else i) + ":" start_task = tmp[:-1] + if day == 'mon': schedule.every().monday.at(start_task).do(bot.send_message, id, text, thread_id) elif day == 'tue': @@ -121,13 +142,12 @@ def create_schedule_tasks(): return - -def check_group_in_json(number): +def groups_in_json(): schdl = {} with open(path_schedule, 'r', encoding='utf-8') as json_file: - schdl = json.load(json_file) + schdl = json.load(json_file) - return str(number) in schdl.keys() + return schdl.keys() def return_infos(id): id = str(id) @@ -140,17 +160,20 @@ def return_infos(id): def pause_bot(): schedule.clear() + global allow_update + allow_update = False return -def get_schedule(id): +def get_schedule(id, day): id = str(id) - text = 'Расписание на сегодня:\n\n' + days = ['mon', 'tue', 'wed', 'thu', 'fri', 'sat'] + russian_days = ['понедельник', 'вторник', 'среду', 'четверг', 'пятницу', 'субботу'] + + text = 'Расписание на {}:\n\n'.format(russian_days[days.index(day)]) schdl_today = {} with open(path_schedule, 'r', encoding='utf-8') as schedule_file: - day = datetime.today().strftime('%A').lower()[:3] - group = '' with open(path_users, 'r', encoding='utf-8') as user_file: group = json.load(user_file)[id]['group'] @@ -161,30 +184,9 @@ def get_schedule(id): try: schdl_today = json.load(schedule_file)[group][day] except: - return 'Расписания твоей группы на сегодня нет' + return 'Расписания твоей группы на {} нет'.format(russian_days[days.index(day)]) for i, lesson in schdl_today.items(): - if len(lesson["infos"].split('|')) == 2: - teacher = lesson["infos"].split('|')[0] - room = lesson["infos"].split('|')[1] - text += '•{} | {}\ - \n{}\ - \n{}'.format(i, room, lesson["name"], teacher) - - elif len(lesson["infos"].split('|')) == 4: - infos = lesson["infos"].split('|') - text += '•{}\ - \n{}\ - \n({}) {}\ - \n({}) {}'.format(i, lesson["name"], - infos[1], - infos[0], - infos[3], - infos[2]) - else: - text += '•{}\ - \n{}'.format(i, lesson["name"]) - - text += '\n\n' + text += '•' + parse_lesson(i, lesson) + '\n\n' return text diff --git a/schedules.py b/schedules.py index 03e0047..91a5217 100644 --- a/schedules.py +++ b/schedules.py @@ -1,12 +1,13 @@ #!/usr/bin/python3.3 -import threading, telebot, schedule, time +import threading, telebot, schedule, time, re +from datetime import datetime import os, sys from telebot import types 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("6355753103:AAHWhA8nES_toURP0TWj8P0UNxtMrK8RaVI") start_txt = 'Привет! Это бот, который будет кидать тебе сообщения перед нужной парой с номером кабинета/фамилией препода\ \n\nТы можешь настроить отправку сообщений в лс или добавить меня в группу, куда я буду присылать расписание\ @@ -29,7 +30,7 @@ def update_schedules(message): @bot.message_handler(commands=['update']) @admin_command def update_schedules(message): - for_json.create_schedule_tasks() + for_json.create_schedule_tasks(manual=True) send_admin_message('Произошёл update') return @@ -89,7 +90,7 @@ def start(message): temp ] - if for_json.check_group_in_json(temp): + if temp in for_json.groups_in_json(): send_message(message.chat.id, 'Отлично! Теперь я буду присылать вам расписание {} группы'.format(temp)) else: @@ -109,16 +110,15 @@ def start(message): send_message(message.chat.id, start_txt) markup = types.InlineKeyboardMarkup() - markup.add(types.InlineKeyboardButton(text="102", callback_data="102")) - markup.add(types.InlineKeyboardButton(text="103", callback_data="103")) - markup.add(types.InlineKeyboardButton(text="105", callback_data="105")) - markup.add(types.InlineKeyboardButton(text="110", callback_data="110")) + 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) return -@bot.callback_query_handler(func=lambda call: True) +# @bot.callback_query_handler(func=lambda call: True) +@bot.callback_query_handler(func=lambda call: not(call.data in ['left', 'right'])) def callback_inline(call): infos = [ call.from_user.id, @@ -208,10 +208,39 @@ def send_request(message): def send_schedule(message): text = 'К сожалению, бот не может отправить тебе расписание твоей группы на сегодня :(' try: - text = for_json.get_schedule(str(message.chat.id)) + 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_message(message.chat.id, text) + + 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) + + return + +@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 # выходной + 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) + 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) return From 2f0804316381f2777419dee60a1c76bded24bbf8 Mon Sep 17 00:00:00 2001 From: kr0sh512 Date: Sat, 11 Nov 2023 01:49:36 +0300 Subject: [PATCH 6/8] add buttons for changing schedule day in /schedule --- for_json.py | 114 ++++++++++++++++++++++++++------------------------- schedules.py | 51 ++++++++++++++++++----- 2 files changed, 98 insertions(+), 67 deletions(-) diff --git a/for_json.py b/for_json.py index a917c4e..dadfa7c 100644 --- a/for_json.py +++ b/for_json.py @@ -6,6 +6,8 @@ from datetime import datetime path_users = "json/users.json" # Нужный путь до json файлов path_schedule = "json/schedule.json" +allow_update = True + def save_user(infos): for i in range(len(infos)): if type(infos[i]) == type(None): @@ -52,7 +54,44 @@ def change_user_param(id, key, value): create_schedule_tasks() return -def create_schedule_tasks(): +def parse_lesson(time, lesson): + text = '' + start, end = time.split("-") + + if len(lesson["infos"].split('|')) == 2: + teacher = lesson["infos"].split('|')[0] + room = lesson["infos"].split('|')[1] + text = '{}-{} | {}\ + \n{}\ + \n{}'.format(start, end, room, lesson["name"], teacher) + elif len(lesson["infos"].split('|')) == 4: + infos = lesson["infos"].split('|') + text = '{}-{}\ + \n{}\ + \n({}) {}\ + \n({}) {}'.format(start, end, lesson["name"], + infos[1], + infos[0], + infos[3], + infos[2]) + elif len(lesson["infos"].split('|')) == 1: + text = '{}-{}\ + \n{}'.format(start, end, lesson["name"]) + else: + print("что-то не так") + print(lesson["name"]) + + return text + +def create_schedule_tasks(manual=False): + global allow_update + if not manual: + bot.send_admin_message('Произошёл auto-update') + if not allow_update: + return + + allow_update = True + schedule.clear() users = {} schdl = {} @@ -72,33 +111,14 @@ def create_schedule_tasks(): if lesson["name"] == "": # Неправильный формат schedule continue - start, end = i.split("-") - text = '' - if len(lesson["infos"].split('|')) == 2: - teacher = lesson["infos"].split('|')[0] - room = lesson["infos"].split('|')[1] - text = '{}-{} | {}\ - \n{}\ - \n{}'.format(start, end, room, lesson["name"], teacher) - elif len(lesson["infos"].split('|')) == 4: - infos = lesson["infos"].split('|') - text = '{}-{}\ - \n{}\ - \n({}) {}\ - \n({}) {}'.format(start, end, lesson["name"], - infos[1], - infos[0], - infos[3], - infos[2]) - elif len(lesson["infos"].split('|')) == 1: - text = '{}-{}\ - \n{}'.format(start, end, lesson["name"]) - else: #неправильный формат infos в json файле - print("что-то не так") - print(lesson["name"]) + start = i.split("-")[0] + text = parse_lesson(i, lesson) + + if text == '': continue thread_id = params["thread"] + delta = '00:' + params["timeout"] format = '%H:%M' start_task = datetime.strptime(start, format) - datetime.strptime(delta, format) @@ -106,6 +126,7 @@ def create_schedule_tasks(): for i in str(start_task).split(':'): tmp += (('0' + i ) if len(i) < 2 else i) + ":" start_task = tmp[:-1] + if day == 'mon': schedule.every().monday.at(start_task).do(bot.send_message, id, text, thread_id) elif day == 'tue': @@ -121,13 +142,12 @@ def create_schedule_tasks(): return - -def check_group_in_json(number): +def groups_in_json(): schdl = {} with open(path_schedule, 'r', encoding='utf-8') as json_file: - schdl = json.load(json_file) + schdl = json.load(json_file) - return str(number) in schdl.keys() + return schdl.keys() def return_infos(id): id = str(id) @@ -140,17 +160,20 @@ def return_infos(id): def pause_bot(): schedule.clear() + global allow_update + allow_update = False return -def get_schedule(id): +def get_schedule(id, day): id = str(id) - text = 'Расписание на сегодня:\n\n' + days = ['mon', 'tue', 'wed', 'thu', 'fri', 'sat'] + russian_days = ['понедельник', 'вторник', 'среду', 'четверг', 'пятницу', 'субботу'] + + text = 'Расписание на {}:\n\n'.format(russian_days[days.index(day)]) schdl_today = {} with open(path_schedule, 'r', encoding='utf-8') as schedule_file: - day = datetime.today().strftime('%A').lower()[:3] - group = '' with open(path_users, 'r', encoding='utf-8') as user_file: group = json.load(user_file)[id]['group'] @@ -161,30 +184,9 @@ def get_schedule(id): try: schdl_today = json.load(schedule_file)[group][day] except: - return 'Расписания твоей группы на сегодня нет' + return 'Расписания твоей группы на {} нет'.format(russian_days[days.index(day)]) for i, lesson in schdl_today.items(): - if len(lesson["infos"].split('|')) == 2: - teacher = lesson["infos"].split('|')[0] - room = lesson["infos"].split('|')[1] - text += '•{} | {}\ - \n{}\ - \n{}'.format(i, room, lesson["name"], teacher) - - elif len(lesson["infos"].split('|')) == 4: - infos = lesson["infos"].split('|') - text += '•{}\ - \n{}\ - \n({}) {}\ - \n({}) {}'.format(i, lesson["name"], - infos[1], - infos[0], - infos[3], - infos[2]) - else: - text += '•{}\ - \n{}'.format(i, lesson["name"]) - - text += '\n\n' + text += '•' + parse_lesson(i, lesson) + '\n\n' return text diff --git a/schedules.py b/schedules.py index 03e0047..d3c034f 100644 --- a/schedules.py +++ b/schedules.py @@ -1,12 +1,13 @@ #!/usr/bin/python3.3 -import threading, telebot, schedule, time +import threading, telebot, schedule, time, re +from datetime import datetime import os, sys from telebot import types 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("TELEBOT_API") start_txt = 'Привет! Это бот, который будет кидать тебе сообщения перед нужной парой с номером кабинета/фамилией препода\ \n\nТы можешь настроить отправку сообщений в лс или добавить меня в группу, куда я буду присылать расписание\ @@ -29,7 +30,7 @@ def update_schedules(message): @bot.message_handler(commands=['update']) @admin_command def update_schedules(message): - for_json.create_schedule_tasks() + for_json.create_schedule_tasks(manual=True) send_admin_message('Произошёл update') return @@ -89,7 +90,7 @@ def start(message): temp ] - if for_json.check_group_in_json(temp): + if temp in for_json.groups_in_json(): send_message(message.chat.id, 'Отлично! Теперь я буду присылать вам расписание {} группы'.format(temp)) else: @@ -109,16 +110,15 @@ def start(message): send_message(message.chat.id, start_txt) markup = types.InlineKeyboardMarkup() - markup.add(types.InlineKeyboardButton(text="102", callback_data="102")) - markup.add(types.InlineKeyboardButton(text="103", callback_data="103")) - markup.add(types.InlineKeyboardButton(text="105", callback_data="105")) - markup.add(types.InlineKeyboardButton(text="110", callback_data="110")) + 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) return -@bot.callback_query_handler(func=lambda call: True) +# @bot.callback_query_handler(func=lambda call: True) +@bot.callback_query_handler(func=lambda call: not(call.data in ['left', 'right'])) def callback_inline(call): infos = [ call.from_user.id, @@ -208,10 +208,39 @@ def send_request(message): def send_schedule(message): text = 'К сожалению, бот не может отправить тебе расписание твоей группы на сегодня :(' try: - text = for_json.get_schedule(str(message.chat.id)) + 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_message(message.chat.id, text) + + 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) + + return + +@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 # выходной + 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) + 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) return From c4751fd4f1b09ea9426fb5ae2f723901965a4d51 Mon Sep 17 00:00:00 2001 From: kr0sh512 Date: Sat, 11 Nov 2023 01:56:27 +0300 Subject: [PATCH 7/8] / --- schedules.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/schedules.py b/schedules.py index 24d7ce6..49af9e4 100644 --- a/schedules.py +++ b/schedules.py @@ -1,5 +1,5 @@ #!/usr/bin/python3.3 -import threading, telebot, schedule, time, re +import threading, telebot, schedule, time from datetime import datetime import os, sys from telebot import types From 6f53e4edc84b8f3b16b05b4fe81e5b43fd6fefdc Mon Sep 17 00:00:00 2001 From: kr0sh512 Date: Thu, 16 Nov 2023 05:21:55 +0300 Subject: [PATCH 8/8] small refactor --- for_json.py | 2 + schedules.py | 230 +++++++++++++++++++++++++++++---------------------- 2 files changed, 132 insertions(+), 100 deletions(-) diff --git a/for_json.py b/for_json.py index dadfa7c..03c3e5f 100644 --- a/for_json.py +++ b/for_json.py @@ -167,6 +167,8 @@ def pause_bot(): def get_schedule(id, day): id = str(id) + if day == 'sun': + day = 'mon' days = ['mon', 'tue', 'wed', 'thu', 'fri', 'sat'] russian_days = ['понедельник', 'вторник', 'среду', 'четверг', 'пятницу', 'субботу'] diff --git a/schedules.py b/schedules.py index 49af9e4..c77a709 100644 --- a/schedules.py +++ b/schedules.py @@ -9,9 +9,36 @@ from admin import admin_command, is_admin, send_admin_message, send_admin_docume bot = telebot.TeleBot("TOKEN_API") -start_txt = 'Привет! Это бот, который будет кидать тебе сообщения перед нужной парой с номером кабинета/фамилией препода\ - \n\nТы можешь настроить отправку сообщений в лс или добавить меня в группу, куда я буду присылать расписание\ -\n\nCreated by: @Kr0sH_512' + + +@bot.message_handler(commands=['help', 'faq']) +def help(message): + help_msg = 'Мои команды:\ + \n/start - используй, чтобы сменить номер группы.\ + \n/schedule - используй, чтобы получить расписание на сегодня.\ + \n/info - используй, чтобы узнать твои настройки бота\ + \n/pause - используй, чтобы прекратить получать сообщения от бота\ + \n/thread - используй в нужном чате канала, чтобы бот отправлял сообщения именно туда\ + \n/timeout - настрой время, когда бот будет присылать тебе сообщение\ + \n/request - используй, чтобы отправить разработчику какой-то запрос\ + \n/source - Страница бота на Github\ + \n\nИли же можно всегда написать напрямую: @Kr0sH_512' + + send_message(message.chat.id, help_msg) + + if is_admin(message): + admin_help_msg = 'И команды только для админа:\ + \n/restart - перезапуск бота.\ + \n/update - обновление schedule задач\ + \n/json - получить файл пользователей и расписания\ + \n/info id_пользователя - узнать настройки пользователя\ + \n/pause_all - приостановить бота для всех (каникулы/выходные)\ + \n/stop id_пользователя - приостановить отправку сообщений для пользователя' + send_admin_message(admin_help_msg) + + return + +### --//-- @bot.message_handler(commands=['test']) @admin_command @@ -27,6 +54,12 @@ def update_schedules(message): return +@bot.message_handler(commands=['restart']) +@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): @@ -34,12 +67,6 @@ def update_schedules(message): send_admin_message('Произошёл update') return - -@bot.message_handler(commands=['restart']) -@admin_command -def restart_bot(message): - send_admin_message('bye') - os.execv(sys.executable, ['python'] + sys.argv) @bot.message_handler(commands=['json']) @admin_command @@ -51,6 +78,8 @@ def send_json(message): return +# info user_id + @bot.message_handler(commands=['pause_all']) @admin_command def pause_bot(message): @@ -68,10 +97,14 @@ def pause_bot(message): for_json.change_user_param(str(message.text).split(' ')[1], 'allow_message', 'no') send_admin_message('Успешно') return - + +### --//-- @bot.message_handler(commands=['start']) def start(message): + start_txt = 'Привет! Это бот, который будет кидать тебе сообщения перед нужной парой с номером кабинета/фамилией препода\ + \n\nТы можешь настроить отправку сообщений в лс или добавить меня в группу, куда я буду присылать расписание\ +\n\nCreated by: @Kr0sH_512' if message.chat.type == "supergroup": if len(message.text.split(' ')) != 2: send_message(message.chat.id, start_txt) @@ -117,7 +150,6 @@ def start(message): return -# @bot.callback_query_handler(func=lambda call: True) @bot.callback_query_handler(func=lambda call: not(call.data in ['left', 'right'])) def callback_inline(call): infos = [ @@ -145,65 +177,8 @@ def callback_inline(call): return -@bot.message_handler(commands=['help', 'faq']) -def help(message): - help_msg = 'Мои команды:\ - \n/start - используй, чтобы сменить номер группы.\ - \n/schedule - используй, чтобы получить расписание на сегодня.\ - \n/info - используй, чтобы узнать твои настройки бота\ - \n/pause - используй, чтобы прекратить получать сообщения от бота\ - \n/thread - используй в нужном чате канала, чтобы бот отправлял сообщения именно туда\ - \n/timeout - настрой время, когда бот будет присылать тебе сообщение\ - \n/request - используй, чтобы отправить разработчику какой-то запрос\ - \n/source - Страница бота на Github\ - \n\nИли же можно всегда написать напрямую: @Kr0sH_512' - - send_message(message.chat.id, help_msg) - - if is_admin(message): - admin_help_msg = 'И команды только для админа:\ - \n/restart - перезапуск бота.\ - \n/update - обновление schedule задач\ - \n/json - получить файл пользователей и расписания\ - \n/info id_пользователя - узнать настройки пользователя\ - \n/pause_all - приостановить бота для всех (каникулы/выходные)\ - \n/stop id_пользователя - приостановить отправку сообщений для пользователя' - send_admin_message(admin_help_msg) - - return - -@bot.message_handler(commands=['request']) -def request(message): - 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, 'Отмена отпраки') - return - - infos = [ - message.from_user.id, - message.from_user.first_name, - message.from_user.last_name, - message.from_user.username, - 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]) - send_admin_message(admin_text) - - send_message(message.from_user.id, 'Ваше сообщение успешно доставлено!') - - return - + + @bot.message_handler(commands=['schedule']) def send_schedule(message): text = 'К сожалению, бот не может отправить тебе расписание твоей группы на сегодня :(' @@ -243,28 +218,9 @@ def change_schedule(call): parse_mode=ParseMode.HTML, reply_markup=markup) return - -@bot.message_handler(commands=['timeout']) -def set_timeout(message): - 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))) - else: - send_message(message.chat.id, - 'Пожалуйста, пришли число от 1 до 60. \ - \nПо умолчанию: 10') - send_message(message.chat.id, 'Изменения не были внесены') - - return - + + @bot.message_handler(commands=['info']) def send_info(message): if len(message.text.split(' ')) == 2 and is_admin(message): @@ -291,13 +247,25 @@ def send_info(message): send_message(message.chat.id, text) return + + + +@bot.message_handler(commands=['pause']) +def pause_schedule(message): + infos = for_json.return_infos(message.chat.id)['allow_message'] -@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) + 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, 'Рассылка сообщений возоблена!') return + + @bot.message_handler(commands=['thread']) def change_thread(message): thread_id = '' @@ -312,16 +280,74 @@ def change_thread(message): send_message(message.chat.id, 'Хорошо, теперь я буду отправлять сообщения в этот чат', thread_id) return + + + +@bot.message_handler(commands=['timeout']) +def set_timeout(message): + send_message(message.chat.id, 'Пришли мне число от 1 до 60. \ + \nЭто будет количество минут, за которое я буду присылать тебе напоминание о уроке (По умолчанию: 10)') + bot.register_next_step_handler(message, save_timeout) -@bot.message_handler(commands=['pause']) -def pause_schedule(message): - for_json.change_user_param(str(message.chat.id), 'allow_message', 'no') - - send_message(message.chat.id, 'Рассылка сообщений преращена. \ - \nДля возобновления воспользуйтесь командой\n/start') + 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))) + else: + send_message(message.chat.id, + 'Пожалуйста, пришли число от 1 до 60. \ + \nПо умолчанию: 10') + send_message(message.chat.id, 'Изменения не были внесены') + + return + + + +@bot.message_handler(commands=['request']) +def request(message): + 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, 'Отмена отпраки') + return + + infos = [ + message.from_user.id, + message.from_user.first_name, + message.from_user.last_name, + message.from_user.username, + 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]) + send_admin_message(admin_text) + + send_message(message.from_user.id, 'Ваше сообщение успешно доставлено!') + + return + + + +@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) + + return + + + @bot.message_handler(content_types=['text']) def text_message(message): if message.chat.type == "supergroup": @@ -331,6 +357,8 @@ def text_message(message): return + + def send_message(id, text, thread_id='General'): if thread_id == 'General': thread_id = None @@ -362,6 +390,8 @@ def send_document(id, file, text = ''): return + + if __name__ == '__main__': for_json.create_schedule_tasks()