Archived
add thread command
This commit is contained in:
kr0sh512
+42
-11
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user