add thread command

.gitignore

@@ -0,0 +1,4 @@
+
+users.json
+schedule.json
+settings.json

for_json.py

@@ -3,8 +3,8 @@ import json, schedule
 import schedules as bot
 from datetime import datetime
 
-path = "users.json" # Нужный путь до json файлов
+path_users = "json/users.json" # Нужный путь до json файлов
-path_schedule = "schedule.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 = '<i>Выбранная группа</i>: <b>{}</b>\
         \n<i>Время напоминания до урока</i>: <b>{}</b>\
         \n<i>Разрешены ли напоминания</i>: <b>{}</b>'.format(data['group'], data['timeout'], ('да' if data['allow_message'] == 'yes' else 'нет'))
-    return text
+    return text

json/users_expl.json

@@ -5,6 +5,7 @@
         "last_name": "",
         "group": "102",
         "timeout": "12",
-        "allow_message": "yes"
+        "allow_message": "yes",
+        "thread": "General"
     }
 }

schedules.py

@@ -5,20 +5,21 @@ from telebot import types
 from telegram.constants import ParseMode
 import for_json
 
-bot = telebot.TeleBot("API_TOKEN")
+bot = telebot.TeleBot("TOKEN")
-admin_id = 'ADMIN_ID'
+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):
+def send_message(id, text, thread_id='General'):
-    bot.send_message(id, text, parse_mode=ParseMode.HTML)
+    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()