From c37bd01b06186af56d6bd8ff826a1d319a0723f2 Mon Sep 17 00:00:00 2001 From: kr0sh512 Date: Tue, 10 Oct 2023 20:53:54 +0300 Subject: [PATCH] add thread command --- .gitignore | 4 ++ for_json.py | 42 ++++++++------- schedule_expl.json => json/schedule_expl.json | 0 users_expl.json => json/users_expl.json | 3 +- schedules.py | 53 +++++++++++++++---- 5 files changed, 71 insertions(+), 31 deletions(-) create mode 100644 .gitignore rename schedule_expl.json => json/schedule_expl.json (100%) rename users_expl.json => json/users_expl.json (74%) diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..71e586a --- /dev/null +++ b/.gitignore @@ -0,0 +1,4 @@ + +users.json +schedule.json +settings.json diff --git a/for_json.py b/for_json.py index 3dbd42d..aa0999f 100644 --- a/for_json.py +++ b/for_json.py @@ -3,8 +3,8 @@ import json, schedule import schedules as bot from datetime import datetime -path = "users.json" # Нужный путь до json файлов -path_schedule = "schedule.json" +path_users = "json/users.json" # Нужный путь до json файлов +path_schedule = "json/schedule.json" def save_user(infos): for i in range(len(infos)): @@ -20,7 +20,7 @@ def save_user(infos): } data = {} - with open(path, 'r', encoding='utf-8') as json_file: + with open(path_users, 'r', encoding='utf-8') as json_file: data = json.load(json_file) data[infos["id"]] = { @@ -29,34 +29,36 @@ def save_user(infos): "last_name": infos["last_name"], "group": infos["group"], "timeout": "10", - "allow_message": "yes" + "allow_message": "yes", + "thread": "General" } - with open(path, 'w', encoding='utf-8') as json_file: + with open(path_users, 'w', encoding='utf-8') as json_file: json.dump(data, json_file, ensure_ascii=False, indent=4) create_schedule_tasks() + return def change_user_param(id, key, value): data = {} - with open(path, 'r', encoding='utf-8') as json_file: + with open(path_users, 'r', encoding='utf-8') as json_file: data = json.load(json_file) data[id][key] = value - with open(path, 'w', encoding='utf-8') as json_file: + with open(path_users, 'w', encoding='utf-8') as json_file: json.dump(data, json_file, ensure_ascii=False, indent=4) create_schedule_tasks() + return def create_schedule_tasks(): schedule.clear() users = {} schdl = {} - with open(path, 'r', encoding='utf-8') as json_file: + with open(path_users, 'r', encoding='utf-8') as json_file: users = json.load(json_file) with open(path_schedule, 'r', encoding='utf-8') as json_file: schdl = json.load(json_file) for id, params in users.items(): - # print(id, '=>', params) if params['allow_message'] != 'yes': continue group = params['group'] @@ -65,7 +67,7 @@ def create_schedule_tasks(): timeout = params['timeout'] for day in schdl[group]: for i, lesson in schdl[group][day].items(): - if lesson["name"] == "": #проверка на лоха + if lesson["name"] == "": # Неправильный формат schedule continue start, end = i.split("-") @@ -94,6 +96,7 @@ def create_schedule_tasks(): print(lesson["name"]) continue + thread_id = params["thread"] delta = '00:' + params["timeout"] format = '%H:%M' start_task = datetime.strptime(start, format) - datetime.strptime(delta, format) @@ -102,19 +105,20 @@ def create_schedule_tasks(): 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) + schedule.every().monday.at(start_task).do(bot.send_message, id, text, thread_id) elif day == 'tue': - schedule.every().tuesday.at(start_task).do(bot.send_message, id, text) + schedule.every().tuesday.at(start_task).do(bot.send_message, id, text, thread_id) elif day == 'wed': - schedule.every().wednesday.at(start_task).do(bot.send_message, id, text) + schedule.every().wednesday.at(start_task).do(bot.send_message, id, text, thread_id) elif day == 'thu': - schedule.every().thursday.at(start_task).do(bot.send_message, id, text) + schedule.every().thursday.at(start_task).do(bot.send_message, id, text, thread_id) elif day == 'fri': - schedule.every().friday.at(start_task).do(bot.send_message, id, text) + schedule.every().friday.at(start_task).do(bot.send_message, id, text, thread_id) elif day == 'sat': - schedule.every().saturday.at(start_task).do(bot.send_message, id, text) + schedule.every().saturday.at(start_task).do(bot.send_message, id, text, thread_id) - bot.send_message(bot.krosh, '🛑 Произошёл update') + bot.send_message(bot.admin_id, '🛑 Произошёл update') + return def check_group_in_json(number): @@ -125,10 +129,10 @@ def check_group_in_json(number): def return_infos(id): data = {} - with open(path, 'r', encoding='utf-8') as json_file: + 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'], data['timeout'], ('да' if data['allow_message'] == 'yes' else 'нет')) - return text \ No newline at end of file + return text diff --git a/schedule_expl.json b/json/schedule_expl.json similarity index 100% rename from schedule_expl.json rename to json/schedule_expl.json diff --git a/users_expl.json b/json/users_expl.json similarity index 74% rename from users_expl.json rename to json/users_expl.json index a56a3ae..6957ce4 100644 --- a/users_expl.json +++ b/json/users_expl.json @@ -5,6 +5,7 @@ "last_name": "", "group": "102", "timeout": "12", - "allow_message": "yes" + "allow_message": "yes", + "thread": "General" } } \ No newline at end of file diff --git a/schedules.py b/schedules.py index fe592e7..fbd7271 100644 --- a/schedules.py +++ b/schedules.py @@ -5,20 +5,21 @@ from telebot import types from telegram.constants import ParseMode import for_json -bot = telebot.TeleBot("API_TOKEN") -admin_id = 'ADMIN_ID' +bot = telebot.TeleBot("TOKEN") +admin_id = 'id' start_txt = 'Привет! Это бот, который будет кидать тебе сообщения перед нужной парой с номером кабинета/фамилией препода\ \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() + return - @bot.message_handler(commands=['restart']) def restart_bot(message): if (str(message.from_user.id) != admin_id): @@ -26,16 +27,18 @@ def restart_bot(message): else: 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, 'rb') as json_file: + 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 @bot.message_handler(commands=['start']) @@ -80,6 +83,7 @@ def start(message): markup.add(types.InlineKeyboardButton(text="105", callback_data="105")) markup.add(types.InlineKeyboardButton(text="Моей группы нет в этом списке", callback_data="other")) bot.send_message(message.chat.id, 'Пожалуйста, выбери свою группу', parse_mode='Markdown', reply_markup=markup) + return @bot.callback_query_handler(func=lambda call: True) def callback_inline(call): @@ -103,8 +107,7 @@ def callback_inline(call): message_id=call.message.message_id, text='Отлично, теперь тебе будут приходить сообщения!', parse_mode='Markdown') - - + return @bot.message_handler(commands=['help', 'faq']) def help(message): @@ -112,8 +115,10 @@ def help(message): \n/start - используй, чтобы сменить номер группы.\ \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') if str(message.from_user.id) == admin_id and message.from_user.id == message.chat.id: @@ -123,12 +128,14 @@ def help(message): \n/json - получить файл пользователей и расписания\ \n/info id\_пользователя - узнать настройки пользователя' bot.send_message(message.from_user.id, help_msg, parse_mode='Markdown') + return @bot.message_handler(commands=['request']) def request(message): bot.send_message(message.from_user.id, 'Отправь мне сообщение и я перешлю его разработчику\ \n(или напиши stop, чтобы отменить отправку)', parse_mode='Markdown') bot.register_next_step_handler(message, send_request) + return def send_request(message): if message.text == "stop" or message.text == "стоп" or message.text[0] == '/': @@ -151,20 +158,21 @@ def send_request(message): 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') - - + return @bot.message_handler(commands=['schedule']) ### На доработке def send_schedule(message): # команда для отправки всего расписания bot.send_message(message.chat.id, 'Пожалуйста, притворись, что ты получил сейчас своё расписание на неделю\ \n\n(Функция в разработке)', parse_mode='Markdown') + return @bot.message_handler(commands=['timeout']) def set_timeout(message): bot.send_message(message.chat.id, 'Пришли мне число от 1 до 60. \ \nЭто будет количество минут, за которое я буду присылать тебе напоминание о уроке (По умолчанию: 10)', parse_mode='Markdown') bot.register_next_step_handler(message, save_timeout) + return def save_timeout(message): if message.text[0] == '/': @@ -175,6 +183,7 @@ def save_timeout(message): else: bot.send_message(message.chat.id, 'Пожалуйста, пришли число от 1 до 60. \nПо умолчанию: 10', parse_mode='Markdown') bot.register_next_step_handler(message, save_timeout) + return @bot.message_handler(commands=['info']) def send_info(message): @@ -186,12 +195,31 @@ def send_info(message): bot.send_message(message.chat.id, 'Хорошо, вот твои настройки:', parse_mode='Markdown') text = for_json.return_infos(str(message.chat.id)) bot.send_message(message.chat.id, text, parse_mode=ParseMode.HTML) + 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='Markdown') + 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) + return @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') + return @bot.message_handler(content_types=['text']) def text_message(message): @@ -199,16 +227,19 @@ def text_message(message): return bot.send_message(message.from_user.id, 'К сожалению, я ещё не умею обрабатывать сообщения.\ \nИспользуй команды из меню или напиши /help.', parse_mode='Markdown') + return -def send_message(id, text): - bot.send_message(id, text, parse_mode=ParseMode.HTML) +def send_message(id, text, thread_id='General'): + bot.send_message(id, text, parse_mode=ParseMode.HTML, message_thread_id=thread_id) + return if __name__ == '__main__': for_json.create_schedule_tasks() send_message(admin_id, '🛑 я перезапустился!') 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() + bot.polling(none_stop=True) while True: schedule.run_pending()