This commit is contained in:
kr0sh512
2023-10-24 11:50:12 +03:00
parent 4e563bc106
commit fbeb91f491
2 changed files with 39 additions and 18 deletions
+1 -1
View File
@@ -79,7 +79,7 @@ def create_schedule_tasks():
\n<b>{}</b>\ \n<b>{}</b>\
\n<i>{}</i>'.format(start, end, room, lesson["name"], teacher) \n<i>{}</i>'.format(start, end, room, lesson["name"], teacher)
elif len(lesson["infos"].split('|')) == 4: elif len(lesson["infos"].split('|')) == 4:
infos = lesson["infos"].split('|'); infos = lesson["infos"].split('|')
text = '{}-{}\ text = '{}-{}\
\n<b>{}</b>\ \n<b>{}</b>\
\n({}) <i>{}</i>\ \n({}) <i>{}</i>\
+38 -17
View File
@@ -5,10 +5,8 @@ from telebot import types
from telegram.constants import ParseMode from telegram.constants import ParseMode
import for_json import for_json
import asyncio bot = telebot.TeleBot("TOKEN_API")
admin_id = '856850518'
bot = telebot.TeleBot("TOKEN")
admin_id = 'id'
start_txt = 'Привет! Это бот, который будет кидать тебе сообщения перед нужной парой с номером кабинета/фамилией препода\ start_txt = 'Привет! Это бот, который будет кидать тебе сообщения перед нужной парой с номером кабинета/фамилией препода\
\n\nCreated by: @Kr0sH\_512' \n\nCreated by: @Kr0sH\_512'
@@ -20,6 +18,7 @@ def update_schedules(message):
bot.send_message(message.from_user.id, 'Команда доступна только администратору', parse_mode='Markdown') bot.send_message(message.from_user.id, 'Команда доступна только администратору', parse_mode='Markdown')
else: else:
for_json.create_schedule_tasks() for_json.create_schedule_tasks()
return return
@bot.message_handler(commands=['restart']) @bot.message_handler(commands=['restart'])
@@ -29,6 +28,7 @@ def restart_bot(message):
else: else:
bot.send_message(admin_id, '🛑 bye', parse_mode=ParseMode.HTML) bot.send_message(admin_id, '🛑 bye', parse_mode=ParseMode.HTML)
os.execv(sys.executable, ['python'] + sys.argv) os.execv(sys.executable, ['python'] + sys.argv)
return return
@bot.message_handler(commands=['json']) @bot.message_handler(commands=['json'])
@@ -40,6 +40,7 @@ def send_json(message):
bot.send_document(admin_id, json_file) bot.send_document(admin_id, json_file)
with open(for_json.path_schedule, 'rb') as json_file: with open(for_json.path_schedule, 'rb') as json_file:
bot.send_document(admin_id, json_file) bot.send_document(admin_id, json_file)
return return
@@ -49,6 +50,7 @@ def start(message):
print(message.text) print(message.text)
print(message.chat.id) print(message.chat.id)
print(message.text.split(' ')) print(message.text.split(' '))
if len(message.text.split(' ')) != 2: if len(message.text.split(' ')) != 2:
bot.send_message(message.chat.id, start_txt, parse_mode='Markdown') bot.send_message(message.chat.id, start_txt, parse_mode='Markdown')
bot.send_message(message.chat.id, 'Похоже, что этот бот добавлен в группу. Чтобы он присылал расписание в чат, \ bot.send_message(message.chat.id, 'Похоже, что этот бот добавлен в группу. Чтобы он присылал расписание в чат, \
@@ -56,6 +58,7 @@ def start(message):
\nПример:\n/start@vmk_schedule_bot 102', parse_mode='Markdown') \nПример:\n/start@vmk_schedule_bot 102', parse_mode='Markdown')
else: else:
temp = message.text.split(' ')[1] temp = message.text.split(' ')[1]
if temp.isdigit(): if temp.isdigit():
infos = [ infos = [
message.chat.id, message.chat.id,
@@ -64,6 +67,7 @@ def start(message):
message.from_user.username, message.from_user.username,
temp temp
] ]
if for_json.check_group_in_json(temp): if for_json.check_group_in_json(temp):
bot.send_message(message.chat.id, \ bot.send_message(message.chat.id, \
'Отлично! Теперь я буду присылать вам расписание {} группы'.format(temp), parse_mode='Markdown') 'Отлично! Теперь я буду присылать вам расписание {} группы'.format(temp), parse_mode='Markdown')
@@ -79,6 +83,7 @@ def start(message):
\nПример:\n/start@vmk_schedule_bot 102', parse_mode='Markdown') \nПример:\n/start@vmk_schedule_bot 102', parse_mode='Markdown')
return return
bot.send_message(message.chat.id, start_txt, parse_mode='Markdown') bot.send_message(message.chat.id, start_txt, parse_mode='Markdown')
markup = types.InlineKeyboardMarkup() markup = types.InlineKeyboardMarkup()
markup.add(types.InlineKeyboardButton(text="102", callback_data="102")) 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="105", callback_data="105"))
markup.add(types.InlineKeyboardButton(text="Моей группы нет в этом списке", callback_data="other")) markup.add(types.InlineKeyboardButton(text="Моей группы нет в этом списке", callback_data="other"))
bot.send_message(message.chat.id, 'Пожалуйста, выбери свою группу', parse_mode='Markdown', reply_markup=markup) bot.send_message(message.chat.id, 'Пожалуйста, выбери свою группу', parse_mode='Markdown', reply_markup=markup)
return return
@bot.callback_query_handler(func=lambda call: True) @bot.callback_query_handler(func=lambda call: True)
@@ -98,6 +104,7 @@ def callback_inline(call):
call.data call.data
] ]
for_json.save_user(infos) for_json.save_user(infos)
if call.data == 'other': if call.data == 'other':
bot.edit_message_text(chat_id=call.message.chat.id, bot.edit_message_text(chat_id=call.message.chat.id,
message_id=call.message.message_id, message_id=call.message.message_id,
@@ -110,6 +117,7 @@ def callback_inline(call):
message_id=call.message.message_id, message_id=call.message.message_id,
text='Отлично, теперь тебе будут приходить сообщения!', text='Отлично, теперь тебе будут приходить сообщения!',
parse_mode='Markdown') parse_mode='Markdown')
return return
@bot.message_handler(commands=['help', 'faq']) @bot.message_handler(commands=['help', 'faq'])
@@ -131,6 +139,7 @@ def help(message):
\n/json - получить файл пользователей и расписания\ \n/json - получить файл пользователей и расписания\
\n/info id\_пользователя - узнать настройки пользователя' \n/info id\_пользователя - узнать настройки пользователя'
bot.send_message(message.from_user.id, help_msg, parse_mode='Markdown') bot.send_message(message.from_user.id, help_msg, parse_mode='Markdown')
return return
@bot.message_handler(commands=['request']) @bot.message_handler(commands=['request'])
@@ -138,12 +147,14 @@ def request(message):
bot.send_message(message.from_user.id, 'Отправь мне сообщение и я перешлю его разработчику\ bot.send_message(message.from_user.id, 'Отправь мне сообщение и я перешлю его разработчику\
\n(или напиши stop, чтобы отменить отправку)', parse_mode='Markdown') \n(или напиши stop, чтобы отменить отправку)', parse_mode='Markdown')
bot.register_next_step_handler(message, send_request) bot.register_next_step_handler(message, send_request)
return return
def send_request(message): def send_request(message):
if message.text == "stop" or message.text == "стоп" or message.text[0] == '/': if message.text == "stop" or message.text == "стоп" or message.text[0] == '/':
bot.send_message(message.from_user.id, 'Отмена отпраки', parse_mode='Markdown') bot.send_message(message.from_user.id, 'Отмена отпраки', parse_mode='Markdown')
return return
infos = [ infos = [
message.from_user.id, message.from_user.id,
message.from_user.first_name, 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]) 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(admin_id, text, parse_mode='Markdown')
bot.send_message(message.from_user.id, 'Ваше сообщение успешно доставлено!', parse_mode='Markdown') bot.send_message(message.from_user.id, 'Ваше сообщение успешно доставлено!', parse_mode='Markdown')
return return
@bot.message_handler(commands=['schedule']) ### На доработке @bot.message_handler(commands=['schedule']) ### На доработке
@@ -168,6 +180,7 @@ def send_schedule(message):
# команда для отправки всего расписания # команда для отправки всего расписания
bot.send_message(message.chat.id, 'Пожалуйста, притворись, что ты получил сейчас своё расписание на неделю\ bot.send_message(message.chat.id, 'Пожалуйста, притворись, что ты получил сейчас своё расписание на неделю\
\n\n(Функция в разработке)', parse_mode='Markdown') \n\n(Функция в разработке)', parse_mode='Markdown')
return return
@bot.message_handler(commands=['timeout']) @bot.message_handler(commands=['timeout'])
@@ -175,6 +188,7 @@ def set_timeout(message):
bot.send_message(message.chat.id, 'Пришли мне число от 1 до 60. \ bot.send_message(message.chat.id, 'Пришли мне число от 1 до 60. \
\nЭто будет количество минут, за которое я буду присылать тебе напоминание о уроке (По умолчанию: 10)', parse_mode='Markdown') \nЭто будет количество минут, за которое я буду присылать тебе напоминание о уроке (По умолчанию: 10)', parse_mode='Markdown')
bot.register_next_step_handler(message, save_timeout) bot.register_next_step_handler(message, save_timeout)
return return
def save_timeout(message): def save_timeout(message):
@@ -186,6 +200,7 @@ def save_timeout(message):
else: else:
bot.send_message(message.chat.id, 'Пожалуйста, пришли число от 1 до 60. \nПо умолчанию: 10', parse_mode='Markdown') bot.send_message(message.chat.id, 'Пожалуйста, пришли число от 1 до 60. \nПо умолчанию: 10', parse_mode='Markdown')
bot.register_next_step_handler(message, save_timeout) bot.register_next_step_handler(message, save_timeout)
return return
@bot.message_handler(commands=['info']) @bot.message_handler(commands=['info'])
@@ -198,11 +213,13 @@ def send_info(message):
bot.send_message(message.chat.id, 'Хорошо, вот твои настройки:', parse_mode='Markdown') bot.send_message(message.chat.id, 'Хорошо, вот твои настройки:', parse_mode='Markdown')
text = for_json.return_infos(str(message.chat.id)) text = for_json.return_infos(str(message.chat.id))
bot.send_message(message.chat.id, text, parse_mode=ParseMode.HTML) bot.send_message(message.chat.id, text, parse_mode=ParseMode.HTML)
return return
@bot.message_handler(commands=['source']) @bot.message_handler(commands=['source'])
def send_source(message): def send_source(message):
bot.send_message(message.chat.id, 'https://github.com/kr0sh512/Telegram-bot-schedule', parse_mode='Markdown') bot.send_message(message.chat.id, 'https://github.com/kr0sh512/Telegram-bot-schedule', parse_mode='Markdown')
return return
@bot.message_handler(commands=['thread']) @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) for_json.change_user_param(str(message.chat.id), 'thread', thread_id)
bot.send_message(message.chat.id, 'Хорошо, теперь я буду отправлять сообщения в этот чат', bot.send_message(message.chat.id, 'Хорошо, теперь я буду отправлять сообщения в этот чат',
parse_mode='Markdown', message_thread_id=thread_id) parse_mode='Markdown', message_thread_id=thread_id)
return return
@bot.message_handler(commands=['pause']) @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') for_json.change_user_param(str(message.chat.id), 'allow_message', 'no')
bot.send_message(message.chat.id, 'Рассылка сообщений преращена. \ bot.send_message(message.chat.id, 'Рассылка сообщений преращена. \
\nДля возобновления воспользуйтесь командой\n/start', parse_mode='Markdown') \nДля возобновления воспользуйтесь командой\n/start', parse_mode='Markdown')
return return
@bot.message_handler(content_types=['text']) @bot.message_handler(content_types=['text'])
@@ -230,28 +249,30 @@ def text_message(message):
return return
bot.send_message(message.from_user.id, 'К сожалению, я ещё не умею обрабатывать сообщения.\ bot.send_message(message.from_user.id, 'К сожалению, я ещё не умею обрабатывать сообщения.\
\nИспользуй команды из меню или напиши /help.', parse_mode='Markdown') \nИспользуй команды из меню или напиши /help.', parse_mode='Markdown')
return return
async def send_message(id, text, thread_id='General'): def send_message(id, text, thread_id='General'):
if thread_id == 'General': if thread_id == 'General':
thread_id = None thread_id = None
bot.send_message(id, text, parse_mode=ParseMode.HTML, message_thread_id=thread_id) bot.send_message(id, text, parse_mode=ParseMode.HTML, message_thread_id=thread_id)
return
return
def check_schedule():
while True:
schedule.run_pending()
time.sleep(5)
if __name__ == '__main__': if __name__ == '__main__':
for_json.create_schedule_tasks() for_json.create_schedule_tasks()
send_message(admin_id, '🛑 я перезапустился!') send_message(admin_id, '🛑 я перезапустился!')
print("-------------------------") 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() while True:
try:
# threading.Thread(target=bot.infinity_polling, name='bot_infinity_polling', daemon=True).start() # bot.polling(none_stop=True)
#bot.polling(none_stop=True) schedule.run_pending()
time.sleep(5)
except Exception as e:
print(e)
time.sleep(10)