From 38adcea5658fcc18cf9d001d312153fdfe96942c Mon Sep 17 00:00:00 2001 From: kr0sh512 Date: Thu, 2 Nov 2023 16:18:50 +0300 Subject: [PATCH] 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()