From fbeb91f4910c8fafc631af5f523bb55d97ea6166 Mon Sep 17 00:00:00 2001 From: kr0sh512 Date: Tue, 24 Oct 2023 11:50:12 +0300 Subject: [PATCH] fix bugs --- for_json.py | 2 +- schedules.py | 55 ++++++++++++++++++++++++++++++++++++---------------- 2 files changed, 39 insertions(+), 18 deletions(-) diff --git a/for_json.py b/for_json.py index aa0999f..962cf50 100644 --- a/for_json.py +++ b/for_json.py @@ -79,7 +79,7 @@ def create_schedule_tasks(): \n{}\ \n{}'.format(start, end, room, lesson["name"], teacher) elif len(lesson["infos"].split('|')) == 4: - infos = lesson["infos"].split('|'); + infos = lesson["infos"].split('|') text = '{}-{}\ \n{}\ \n({}) {}\ diff --git a/schedules.py b/schedules.py index 2bc346b..659f4f6 100644 --- a/schedules.py +++ b/schedules.py @@ -5,10 +5,8 @@ from telebot import types from telegram.constants import ParseMode import for_json -import asyncio - -bot = telebot.TeleBot("TOKEN") -admin_id = 'id' +bot = telebot.TeleBot("TOKEN_API") +admin_id = '856850518' start_txt = 'Привет! Это бот, который будет кидать тебе сообщения перед нужной парой с номером кабинета/фамилией препода\ \n\nCreated by: @Kr0sH\_512' @@ -20,6 +18,7 @@ def update_schedules(message): bot.send_message(message.from_user.id, 'Команда доступна только администратору', parse_mode='Markdown') else: for_json.create_schedule_tasks() + return @bot.message_handler(commands=['restart']) @@ -29,6 +28,7 @@ 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']) @@ -40,6 +40,7 @@ def send_json(message): 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 @@ -49,6 +50,7 @@ def start(message): print(message.text) print(message.chat.id) print(message.text.split(' ')) + if len(message.text.split(' ')) != 2: bot.send_message(message.chat.id, start_txt, parse_mode='Markdown') bot.send_message(message.chat.id, 'Похоже, что этот бот добавлен в группу. Чтобы он присылал расписание в чат, \ @@ -56,6 +58,7 @@ def start(message): \nПример:\n/start@vmk_schedule_bot 102', parse_mode='Markdown') else: temp = message.text.split(' ')[1] + if temp.isdigit(): infos = [ message.chat.id, @@ -64,6 +67,7 @@ def start(message): message.from_user.username, temp ] + if for_json.check_group_in_json(temp): bot.send_message(message.chat.id, \ 'Отлично! Теперь я буду присылать вам расписание {} группы'.format(temp), parse_mode='Markdown') @@ -79,6 +83,7 @@ def start(message): \nПример:\n/start@vmk_schedule_bot 102', parse_mode='Markdown') return + bot.send_message(message.chat.id, start_txt, parse_mode='Markdown') markup = types.InlineKeyboardMarkup() markup.add(types.InlineKeyboardButton(text="102", callback_data="102")) @@ -86,6 +91,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) @@ -98,6 +104,7 @@ def callback_inline(call): 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, @@ -110,6 +117,7 @@ def callback_inline(call): message_id=call.message.message_id, text='Отлично, теперь тебе будут приходить сообщения!', parse_mode='Markdown') + return @bot.message_handler(commands=['help', 'faq']) @@ -131,6 +139,7 @@ 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']) @@ -138,12 +147,14 @@ 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] == '/': bot.send_message(message.from_user.id, 'Отмена отпраки', parse_mode='Markdown') return + infos = [ message.from_user.id, message.from_user.first_name, @@ -161,6 +172,7 @@ 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']) ### На доработке @@ -168,6 +180,7 @@ def send_schedule(message): # команда для отправки всего расписания bot.send_message(message.chat.id, 'Пожалуйста, притворись, что ты получил сейчас своё расписание на неделю\ \n\n(Функция в разработке)', parse_mode='Markdown') + return @bot.message_handler(commands=['timeout']) @@ -175,6 +188,7 @@ 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): @@ -186,6 +200,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']) @@ -198,11 +213,13 @@ 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']) @@ -215,6 +232,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) + return @bot.message_handler(commands=['pause']) @@ -222,6 +240,7 @@ 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']) @@ -230,28 +249,30 @@ def text_message(message): return bot.send_message(message.from_user.id, 'К сожалению, я ещё не умею обрабатывать сообщения.\ \nИспользуй команды из меню или напиши /help.', parse_mode='Markdown') + return -async def send_message(id, text, thread_id='General'): +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) - return - -def check_schedule(): - while True: - schedule.run_pending() - time.sleep(5) + + return + if __name__ == '__main__': for_json.create_schedule_tasks() send_message(admin_id, '🛑 я перезапустился!') print("-------------------------") - bot.infinity_polling(skip_pending=True) + threading.Thread(target=bot.infinity_polling, name='bot_infinity_polling', daemon=True).start() - threading.Thread(target=check_schedule, daemon=True).start() - - # threading.Thread(target=bot.infinity_polling, name='bot_infinity_polling', daemon=True).start() - #bot.polling(none_stop=True) - \ No newline at end of file + while True: + try: + # bot.polling(none_stop=True) + schedule.run_pending() + time.sleep(5) + except Exception as e: + print(e) + time.sleep(10) + \ No newline at end of file