formating code

This commit is contained in:
kr0sh512
2024-08-31 21:55:33 +03:00
parent f31f670bd2
commit 681c3eee64
2 changed files with 443 additions and 299 deletions
+1 -1
View File
@@ -1,6 +1,6 @@
users.json users.json
schedule.json schedule.json
settings.json settings.json
json/students.json json/students.json
json/schedule_2sem.json json/schedule_2sem.json
nohup.out
+327 -183
View File
@@ -7,11 +7,12 @@ from telegram.constants import ParseMode
import for_json import for_json
from admin import admin_command, is_admin, send_admin_message, send_admin_document from admin import admin_command, is_admin, send_admin_message, send_admin_document
# bot = telebot.TeleBot("TOKEN_API") bot = telebot.TeleBot("TOKEN_API")
@bot.message_handler(commands=['help', 'faq'])
@bot.message_handler(commands=["help", "faq"])
def help(message): def help(message):
help_msg = 'Мои команды:\ help_msg = "Мои команды:\
\n/start - используй, чтобы сменить номер группы.\ \n/start - используй, чтобы сменить номер группы.\
\n/schedule - используй, чтобы получить расписание на сегодня.\ \n/schedule - используй, чтобы получить расписание на сегодня.\
\n/random - сделай очередь из людей\ \n/random - сделай очередь из людей\
@@ -21,12 +22,12 @@ def help(message):
\n/timeout - настрой время, когда бот будет присылать тебе сообщение\ \n/timeout - настрой время, когда бот будет присылать тебе сообщение\
\n/request - используй, чтобы отправить разработчику какой-то запрос\ \n/request - используй, чтобы отправить разработчику какой-то запрос\
\n/source - Страница бота на Github\ \n/source - Страница бота на Github\
\n\nИли же можно всегда написать напрямую: @Kr0sH_512' \n\nИли же можно всегда написать напрямую: @Kr0sH_512"
send_message(message.chat.id, help_msg) send_message(message.chat.id, help_msg)
if is_admin(message): if is_admin(message):
admin_help_msg = 'И команды только для админа:\ admin_help_msg = "И команды только для админа:\
\n/restart - перезапуск бота.\ \n/restart - перезапуск бота.\
\n/update - обновление schedule задач\ \n/update - обновление schedule задач\
\n/stats - получание статистики\ \n/stats - получание статистики\
@@ -34,43 +35,48 @@ def help(message):
\n/info <i>id_пользователя</i> - узнать настройки пользователя\ \n/info <i>id_пользователя</i> - узнать настройки пользователя\
\n/spam - сделать рассылку\ \n/spam - сделать рассылку\
\n/pause_all - приостановить бота для всех (каникулы/выходные)\ \n/pause_all - приостановить бота для всех (каникулы/выходные)\
\n/stop <i>id_пользователя</i> - приостановить отправку сообщений для пользователя' \n/stop <i>id_пользователя</i> - приостановить отправку сообщений для пользователя"
send_admin_message(admin_help_msg) send_admin_message(admin_help_msg)
return return
### --//-- ### --//--
# @bot.message_handler(commands=['test'])
# @admin_command
# def update_schedules(message):
# text = '<b>Жирный текст</b>\
# \n<i>Курсивный текст</i>\
# \n<u>Подчёркнутый</u>\
# \n<s>Перечёркнутый текст</s>\
# \n<a href="google.com">Ссылка</a>\
# \n<code>Моноширинный текст</code>\
# \n<pre>Форматированный с сохранением пробелов</pre>\
# \n<blockquote>Цитата</blockquote>'
# send_admin_message(text)
# return @bot.message_handler(commands=["test", "t"])
@bot.message_handler(commands=['restart'])
@admin_command @admin_command
def restart_bot(message): def test(message):
send_admin_message('bye') text = '<b>Жирный текст</b>\
os.execv(sys.executable, ['python'] + sys.argv) \n<i>Курсивный текст</i>\
\n<u>Подчёркнутый</u>\
@bot.message_handler(commands=['update']) \n<s>Перечёркнутый текст</s>\
@admin_command \n<a href="google.com">Ссылка</a>\
def update_schedules(message): \n<code>Моноширинный текст</code>\
for_json.create_schedule_tasks(manual=True) \n<pre>Форматированный с сохранением пробелов</pre>\
send_admin_message('Произошёл update') \n<blockquote>Цитата</blockquote>'
send_admin_message(text)
return return
@bot.message_handler(commands=['stats'])
@bot.message_handler(commands=["restart", 'r'])
@admin_command
def restart_bot(message):
send_admin_message("bye")
os.execv(sys.executable, ["python"] + sys.argv)
@bot.message_handler(commands=["update"])
@admin_command
def update_schedules(message):
for_json.create_schedule_tasks(manual=True)
send_admin_message("Произошёл update")
return
@bot.message_handler(commands=["stats"])
@admin_command @admin_command
def send_stat(message): def send_stat(message):
text = "<u>Статистика:</u>\ text = "<u>Статистика:</u>\
@@ -87,7 +93,13 @@ def send_stat(message):
users = for_json.students_in_group(i) users = for_json.students_in_group(i)
sum_users += len(users) sum_users += len(users)
dis_not = len([i for i, x in enumerate(users) if for_json.return_infos(x)["allow_message"] == "no"]) dis_not = len(
[
i
for i, x in enumerate(users)
if for_json.return_infos(x)["allow_message"] == "no"
]
)
sum_dis_not += dis_not sum_dis_not += dis_not
chats = len([i for i, x in enumerate(users) if x[0] == "-"]) chats = len([i for i, x in enumerate(users) if x[0] == "-"])
@@ -101,32 +113,38 @@ def send_stat(message):
return return
@bot.message_handler(commands=['json'])
@bot.message_handler(commands=["json"])
@admin_command @admin_command
def send_json(message): def send_json(message):
with open(for_json.path_users, 'rb') as json_file: with open(for_json.path_users, "rb") as json_file:
send_admin_document(json_file) send_admin_document(json_file)
with open(for_json.path_schedule, 'rb') as json_file: with open(for_json.path_schedule, "rb") as json_file:
send_admin_document(json_file) send_admin_document(json_file)
with open(for_json.path_students, 'rb') as json_file: with open(for_json.path_students, "rb") as json_file:
send_admin_document(json_file) send_admin_document(json_file)
return return
# info user_id # info user_id
@bot.message_handler(commands=['spam'])
@bot.message_handler(commands=["spam"])
@admin_command @admin_command
def spam(message): def spam(message):
send_admin_message("Пришли мне id, номер группы или <i>all</i> для рассылки сообщения этим пользователям") send_admin_message(
"Пришли мне id, номер группы или <i>all</i> для рассылки сообщения этим пользователям"
)
bot.register_next_step_handler(message, spam_msg) bot.register_next_step_handler(message, spam_msg)
return return
def spam_msg(message): def spam_msg(message):
if message.text == "stop" or message.text == "стоп" or message.text[0] == '/': if message.text == "stop" or message.text == "стоп" or message.text[0] == "/":
send_admin_message('Отмена отпраки') send_admin_message("Отмена отпраки")
return return
send_admin_message("Пришли мне текст для рассылки сообщения этим пользователям") send_admin_message("Пришли мне текст для рассылки сообщения этим пользователям")
@@ -135,9 +153,10 @@ def spam_msg(message):
return return
def spam_cnf(message, data): def spam_cnf(message, data):
if message.text == "stop" or message.text == "стоп" or message.text[0] == '/': if message.text == "stop" or message.text == "стоп" or message.text[0] == "/":
send_admin_message('Отмена отпраки') send_admin_message("Отмена отпраки")
return return
global text_spam global text_spam
@@ -160,17 +179,22 @@ def spam_cnf(message, data):
markup = types.InlineKeyboardMarkup() markup = types.InlineKeyboardMarkup()
markup.add(types.InlineKeyboardButton(text="", callback_data="spam#{}".format(data)), markup.add(
types.InlineKeyboardButton(text="", callback_data="spam_no")) types.InlineKeyboardButton(text="", callback_data="spam#{}".format(data)),
types.InlineKeyboardButton(text="", callback_data="spam_no"),
)
bot.send_message(message.from_user.id, text, parse_mode=ParseMode.HTML, reply_markup=markup) bot.send_message(
message.from_user.id, text, parse_mode=ParseMode.HTML, reply_markup=markup
)
return return
@bot.callback_query_handler(func=lambda call: "spam" in call.data) @bot.callback_query_handler(func=lambda call: "spam" in call.data)
def spam_send(call): def spam_send(call):
if call.data == "spam_no": if call.data == "spam_no":
send_admin_message('Отмена отпраки') send_admin_message("Отмена отпраки")
return return
tmp, data = call.data.split("#") tmp, data = call.data.split("#")
@@ -193,69 +217,94 @@ def spam_send(call):
for i in users: for i in users:
send_message(i, text) send_message(i, text)
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,
text='Успешно!', text="Успешно!",
parse_mode=ParseMode.HTML) parse_mode=ParseMode.HTML,
)
return return
@bot.message_handler(commands=['pause_all'])
@bot.message_handler(commands=["pause_all"])
@admin_command @admin_command
def pause_bot(message): def pause_bot(message):
for_json.pause_bot() for_json.pause_bot()
return return
@bot.message_handler(commands=['stop'])
@bot.message_handler(commands=["stop"])
@admin_command @admin_command
def stop_msg(message): def stop_msg(message):
if len(str(message.text).split(' ')) != 2: if len(str(message.text).split(" ")) != 2:
send_admin_message('Эта команда вида /stop <i>id_пользователя</i>') send_admin_message("Эта команда вида /stop <i>id_пользователя</i>")
return return
for_json.change_user_param(str(message.text).split(' ')[1], 'allow_message', 'no')
send_admin_message('Успешно') for_json.change_user_param(str(message.text).split(" ")[1], "allow_message", "no")
send_admin_message("Успешно")
return return
### --//-- ### --//--
@bot.message_handler(commands=['start'])
@bot.message_handler(commands=["start"])
def start(message): def start(message):
start_txt = 'Привет! Это бот, который будет кидать тебе сообщения перед нужной парой с номером кабинета/фамилией препода\ start_txt = "Привет! Это бот, который будет кидать тебе сообщения перед нужной парой с номером кабинета/фамилией препода\
\n\nТы можешь настроить отправку сообщений в лс или добавить меня в группу, куда я буду присылать расписание\ \n\nТы можешь настроить отправку сообщений в лс или добавить меня в группу, куда я буду присылать расписание\
\n\nCreated by: @Kr0sH_512' \n\nCreated by: @Kr0sH_512"
if message.chat.type == "supergroup": if message.chat.type == "supergroup":
if len(message.text.split(' ')) != 2: if len(message.text.split(" ")) != 2:
send_message(message.chat.id, start_txt) send_message(message.chat.id, start_txt)
send_message(message.chat.id, 'Похоже, что этот бот добавлен в группу. Чтобы он присылал расписание в чат, \ send_message(
message.chat.id,
"Похоже, что этот бот добавлен в группу. Чтобы он присылал расписание в чат, \
укажите вместе с /start@vmk_schedule_bot номер своей группы.\ укажите вместе с /start@vmk_schedule_bot номер своей группы.\
\nПример:\n/start@vmk_schedule_bot 102') \nПример:\n/start@vmk_schedule_bot 102",
)
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,
message.chat.title, message.chat.title,
'', "",
message.from_user.username, message.from_user.username,
temp temp,
] ]
if temp in for_json.groups_in_json(): if temp in for_json.groups_in_json():
send_message(message.chat.id, send_message(
'Отлично! Теперь я буду присылать вам расписание {} группы'.format(temp)) message.chat.id,
"Отлично! Теперь я буду присылать вам расписание {} группы".format(
temp
),
)
else: else:
send_message(message.chat.id, 'Похоже, что расписания для этой группы ещё не существует. \ send_message(
\nРазработчик уже пинается, но можете дополнительно написать ему: @Kr0sH_512') message.chat.id,
"Похоже, что расписания для этой группы ещё не существует. \
\nРазработчик уже пинается, но можете дополнительно написать ему: @Kr0sH_512",
)
send_admin_message('В группе {} был запрос на {} группу.\ send_admin_message(
\nid: {}'.format(message.chat.title, temp, message.chat.id)) "В группе {} был запрос на {} группу.\
temp = 'other' \nid: {}".format(
message.chat.title, temp, message.chat.id
)
)
temp = "other"
for_json.save_user(infos) for_json.save_user(infos)
else: else:
send_message(message.chat.id, 'Введено не число. Попробуйте снова.\ send_message(
\nПример:\n/start@vmk_schedule_bot 102') message.chat.id,
"Введено не число. Попробуйте снова.\
\nПример:\n/start@vmk_schedule_bot 102",
)
return return
@@ -265,103 +314,140 @@ def start(message):
for i in for_json.groups_in_json(): for i in for_json.groups_in_json():
markup.add(types.InlineKeyboardButton(text=i, callback_data=i)) markup.add(types.InlineKeyboardButton(text=i, callback_data=i))
markup.add(types.InlineKeyboardButton(text="Моей группы нет в этом списке", callback_data="other")) markup.add(
bot.send_message(message.chat.id, 'Пожалуйста, выбери свою группу', parse_mode=ParseMode.HTML, reply_markup=markup) types.InlineKeyboardButton(
text="Моей группы нет в этом списке", callback_data="other"
)
)
bot.send_message(
message.chat.id,
"Пожалуйста, выбери свою группу",
parse_mode=ParseMode.HTML,
reply_markup=markup,
)
return return
@bot.callback_query_handler(func=lambda call: call.data in for_json.groups_in_json() or call.data == "other")
@bot.callback_query_handler(
func=lambda call: call.data in for_json.groups_in_json() or call.data == "other"
)
def callback_inline(call): def callback_inline(call):
infos = [ infos = [
call.from_user.id, call.from_user.id,
call.from_user.first_name, call.from_user.first_name,
call.from_user.last_name, call.from_user.last_name,
call.from_user.username, call.from_user.username,
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,
text='Используй команду /request \ text="Используй команду /request \
и напиши номер группы, которую хочешь добавить\ и напиши номер группы, которую хочешь добавить\
\nПосле создания расписания для твоей группы, я пришлю тебе сообщение', \nПосле создания расписания для твоей группы, я пришлю тебе сообщение",
parse_mode=ParseMode.HTML) parse_mode=ParseMode.HTML,
)
else: else:
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,
text='Отлично, теперь тебе будут приходить сообщения!', text="Отлично, теперь тебе будут приходить сообщения!",
parse_mode=ParseMode.HTML) parse_mode=ParseMode.HTML,
)
return return
@bot.message_handler(commands=["schedule"])
@bot.message_handler(commands=['schedule'])
def send_schedule(message): def send_schedule(message):
text = 'К сожалению, бот не может отправить тебе расписание твоей группы на сегодня :(' text = (
"К сожалению, бот не может отправить тебе расписание твоей группы на сегодня :("
)
try: try:
day = datetime.today().strftime('%A').lower()[:3] # mon tue ... day = datetime.today().strftime("%A").lower()[:3] # mon tue ...
text = for_json.get_schedule(message.chat.id, day) text = for_json.get_schedule(message.chat.id, day)
except Exception as e: except Exception as e:
send_admin_message('Schedule error from: {} \n\n {}'.format(message.chat.id, e)) send_admin_message("Schedule error from: {} \n\n {}".format(message.chat.id, e))
markup = types.InlineKeyboardMarkup() markup = types.InlineKeyboardMarkup()
markup.add(types.InlineKeyboardButton(text="◀️", callback_data="left"), markup.add(
types.InlineKeyboardButton(text="", callback_data="right")) types.InlineKeyboardButton(text="", callback_data="left"),
bot.send_message(message.chat.id, text, ParseMode.HTML, reply_markup=markup, message_thread_id=message.message_thread_id) types.InlineKeyboardButton(text="▶️", callback_data="right"),
)
bot.send_message(
message.chat.id,
text,
ParseMode.HTML,
reply_markup=markup,
message_thread_id=message.message_thread_id,
)
return return
@bot.callback_query_handler(func=lambda call: call.data in ['left', 'right'])
@bot.callback_query_handler(func=lambda call: call.data in ["left", "right"])
def change_schedule(call): def change_schedule(call):
days = ['mon', 'tue', 'wed', 'thu', 'fri', 'sat'] days = ["mon", "tue", "wed", "thu", "fri", "sat"]
russian_days = ['понедельник', 'вторник', 'среду', 'четверг', 'пятницу', 'субботу'] russian_days = ["понедельник", "вторник", "среду", "четверг", "пятницу", "субботу"]
i = 6 # выходной i = 6 # выходной
for day in russian_days: for day in russian_days:
if day in call.message.text: if day in call.message.text:
i = russian_days.index(day) i = russian_days.index(day)
break break
delta = (1) if (call.data == 'right') else (-1) delta = (1) if (call.data == "right") else (-1)
ind = ((i + delta) % 6) if (i + delta > 0) else (i + delta) ind = ((i + delta) % 6) if (i + delta > 0) else (i + delta)
text = for_json.get_schedule(call.message.chat.id, days[ind]) text = for_json.get_schedule(call.message.chat.id, days[ind])
markup = types.InlineKeyboardMarkup() markup = types.InlineKeyboardMarkup()
markup.add(types.InlineKeyboardButton(text="◀️", callback_data="left"), markup.add(
types.InlineKeyboardButton(text="", callback_data="right")) types.InlineKeyboardButton(text="", callback_data="left"),
types.InlineKeyboardButton(text="▶️", callback_data="right"),
)
bot.edit_message_text(text, call.message.chat.id, call.message.message_id, bot.edit_message_text(
parse_mode=ParseMode.HTML, reply_markup=markup) text,
call.message.chat.id,
call.message.message_id,
parse_mode=ParseMode.HTML,
reply_markup=markup,
)
return return
@bot.message_handler(commands=["random"])
@bot.message_handler(commands=['random'])
def random_people(message): def random_people(message):
list_of_num = ["1️⃣", "2️⃣", "3️⃣", "4️⃣", "5️⃣", "6", "7", "8", "9"] list_of_num = ["1️⃣", "2️⃣", "3️⃣", "4️⃣", "5️⃣", "6", "7", "8", "9"]
list_of_var = for_json.students_in_json(message.chat.id) list_of_var = for_json.students_in_json(message.chat.id)
text = 'Для вашей группы эта команда не настроена.\n\nНапиши запрос через /request' text = "Для вашей группы эта команда не настроена.\n\nНапиши запрос через /request"
if list_of_var == None: if list_of_var == None:
send_message(message.chat.id, text) send_message(message.chat.id, text)
send_admin_message('Запрос на random команду от {}'.format( send_admin_message(
for_json.return_infos(message.chat.id)["group"])) "Запрос на random команду от {}".format(
for_json.return_infos(message.chat.id)["group"]
)
)
return return
text = '• Выбери как именно мне перемешать вас:\n' text = "• Выбери как именно мне перемешать вас:\n"
l = [] l = []
for i, val in enumerate(list_of_var): for i, val in enumerate(list_of_var):
text += '\n{} {}'.format(list_of_num[i], val) text += "\n{} {}".format(list_of_num[i], val)
l.append(types.InlineKeyboardButton(text=list_of_num[i], callback_data="rand#" + val)) l.append(
types.InlineKeyboardButton(text=list_of_num[i], callback_data="rand#" + val)
)
l = [l] l = [l]
@@ -370,121 +456,157 @@ def random_people(message):
return return
@bot.callback_query_handler(func=lambda call: "rand#" in call.data) @bot.callback_query_handler(func=lambda call: "rand#" in call.data)
def make_random(call): def make_random(call):
list_of_stud = for_json.students_in_json(call.message.chat.id, call.data.split("#")[1]) list_of_stud = for_json.students_in_json(
call.message.chat.id, call.data.split("#")[1]
)
random.shuffle(list_of_stud) random.shuffle(list_of_stud)
text = '<b><u>{}</u></b>:\n\n'.format(call.data.split("#")[1]) text = "<b><u>{}</u></b>:\n\n".format(call.data.split("#")[1])
for i, name in enumerate(list_of_stud): for i, name in enumerate(list_of_stud):
text += '<code>{}{}</code>. {}\n'.format(' ' if i + 1 < 10 else '', i + 1, name) text += "<code>{}{}</code>. {}\n".format(" " if i + 1 < 10 else "", i + 1, name)
bot.edit_message_text('Хорошо, вот ваше распределение!', call.message.chat.id, bot.edit_message_text(
call.message.message_id, parse_mode=ParseMode.HTML) "Хорошо, вот ваше распределение!",
call.message.chat.id,
call.message.message_id,
parse_mode=ParseMode.HTML,
)
send_message(call.message.chat.id, text) send_message(call.message.chat.id, text)
return return
@bot.message_handler(commands=['info']) @bot.message_handler(commands=["info"])
def send_info(message): def send_info(message):
if len(message.text.split(' ')) == 2 and is_admin(message): if len(message.text.split(" ")) == 2 and is_admin(message):
send_admin_message('Окей, держи настройки <code>{}</code>'.format(message.text.split(' ')[1])) send_admin_message(
infos = for_json.return_infos(message.text.split(' ')[1]) "Окей, держи настройки <code>{}</code>".format(message.text.split(" ")[1])
text = '<i>Никнейм</i>: <b>@{}</b>\ )
infos = for_json.return_infos(message.text.split(" ")[1])
text = "<i>Никнейм</i>: <b>@{}</b>\
\n<i>Имя</i>: <b>{} {}</b>\ \n<i>Имя</i>: <b>{} {}</b>\
\n<i>Выбранная группа</i>: <b>{}</b>\ \n<i>Выбранная группа</i>: <b>{}</b>\
\n<i>Время напоминания до урока</i>: <b>{}</b>\ \n<i>Время напоминания до урока</i>: <b>{}</b>\
\n<i>Разрешены ли напоминания</i>: <b>{}</b>'.format(infos['username'], infos['first_name'], infos['last_name'], \n<i>Разрешены ли напоминания</i>: <b>{}</b>".format(
(infos['group'] if infos['group'] != 'other' else 'не выбрана'), infos["username"],
infos['timeout'], infos["first_name"],
('да' if infos['allow_message'] == 'yes' else 'нет')) infos["last_name"],
(infos["group"] if infos["group"] != "other" else "не выбрана"),
infos["timeout"],
("да" if infos["allow_message"] == "yes" else "нет"),
)
send_admin_message(text) send_admin_message(text)
return return
send_message(message.chat.id, 'Хорошо, вот твои настройки:') send_message(message.chat.id, "Хорошо, вот твои настройки:")
infos = for_json.return_infos(message.chat.id) infos = for_json.return_infos(message.chat.id)
text = '<i>Выбранная группа</i>: <b>{}</b>\ text = "<i>Выбранная группа</i>: <b>{}</b>\
\n<i>Время напоминания до урока</i>: <b>{}</b>\ \n<i>Время напоминания до урока</i>: <b>{}</b>\
\n<i>Разрешены ли напоминания</i>: <b>{}</b>'.format((infos['group'] if infos['group'] != 'other' else 'не выбрана'), \n<i>Разрешены ли напоминания</i>: <b>{}</b>".format(
infos['timeout'], (infos["group"] if infos["group"] != "other" else "не выбрана"),
('да' if infos['allow_message'] == 'yes' else 'нет')) infos["timeout"],
("да" if infos["allow_message"] == "yes" else "нет"),
)
send_message(message.chat.id, text) send_message(message.chat.id, text)
return return
@bot.message_handler(commands=["pause"])
@bot.message_handler(commands=['pause'])
def pause_schedule(message): def pause_schedule(message):
infos = for_json.return_infos(message.chat.id)['allow_message'] infos = for_json.return_infos(message.chat.id)["allow_message"]
if infos == 'yes': if infos == "yes":
for_json.change_user_param(str(message.chat.id), 'allow_message', 'no') for_json.change_user_param(str(message.chat.id), "allow_message", "no")
send_message(message.chat.id, 'Рассылка сообщений прекращена. \ send_message(
\nДля возобновления воспользуйтесь командой\n/pause') message.chat.id,
"Рассылка сообщений прекращена. \
\nДля возобновления воспользуйтесь командой\n/pause",
)
else: else:
for_json.change_user_param(str(message.chat.id), 'allow_message', 'yes') for_json.change_user_param(str(message.chat.id), "allow_message", "yes")
send_message(message.chat.id, 'Рассылка сообщений возоблена!') send_message(message.chat.id, "Рассылка сообщений возоблена!")
return return
@bot.message_handler(commands=["thread"])
@bot.message_handler(commands=['thread'])
def change_thread(message): def change_thread(message):
thread_id = '' thread_id = ""
try: try:
thread_id = message.reply_to_message.message_thread_id thread_id = message.reply_to_message.message_thread_id
except AttributeError: except AttributeError:
thread_id = "General" thread_id = "General"
for_json.change_user_param(str(message.chat.id), 'thread', thread_id) for_json.change_user_param(str(message.chat.id), "thread", thread_id)
send_message(message.chat.id, 'Хорошо, теперь я буду отправлять сообщения в этот чат', thread_id) send_message(
message.chat.id,
"Хорошо, теперь я буду отправлять сообщения в этот чат",
thread_id,
)
return return
@bot.message_handler(commands=["timeout"])
@bot.message_handler(commands=['timeout'])
def set_timeout(message): def set_timeout(message):
send_message(message.chat.id, 'Пришли мне число от 1 до 60. \ send_message(
\nЭто будет количество минут, за которое я буду присылать тебе напоминание о паре (По умолчанию: 10)') message.chat.id,
"Пришли мне число от 1 до 60. \
\nЭто будет количество минут, за которое я буду присылать тебе напоминание о паре (По умолчанию: 10)",
)
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):
if str(message.text).isdigit() and int(message.text) <= 60 and int(message.text) >= 1: if (
for_json.change_user_param(message.chat.id, 'timeout', str(message.text)) str(message.text).isdigit()
send_message(message.chat.id, and int(message.text) <= 60
'Хорошо, теперь ты будешь получать напоминания за {} минут до урока'.format(str(message.text))) and int(message.text) >= 1
):
for_json.change_user_param(message.chat.id, "timeout", str(message.text))
send_message(
message.chat.id,
"Хорошо, теперь ты будешь получать напоминания за {} минут до урока".format(
str(message.text)
),
)
else: else:
send_message(message.chat.id, send_message(
'Пожалуйста, пришли <u>число</u> от 1 до 60. \ message.chat.id,
\nПо умолчанию: 10') "Пожалуйста, пришли <u>число</u> от 1 до 60. \
send_message(message.chat.id, 'Изменения не были внесены') \nПо умолчанию: 10",
)
send_message(message.chat.id, "Изменения не были внесены")
return return
@bot.message_handler(commands=["request"])
@bot.message_handler(commands=['request'])
def request(message): def request(message):
send_message(message.from_user.id, 'Отправь мне сообщение и я перешлю его разработчику\ send_message(
\n(или напиши stop, чтобы отменить отправку)') message.from_user.id,
"Отправь мне сообщение и я перешлю его разработчику\
\n(или напиши stop, чтобы отменить отправку)",
)
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] == "/":
send_message(message.from_user.id, 'Отмена отпраки') send_message(message.from_user.id, "Отмена отпраки")
return return
infos = [ infos = [
@@ -492,43 +614,49 @@ def send_request(message):
message.from_user.first_name, message.from_user.first_name,
message.from_user.last_name, message.from_user.last_name,
message.from_user.username, message.from_user.username,
message.text message.text,
] ]
for i in range(len(infos)): for i in range(len(infos)):
if type(infos[i]) == type(None): if type(infos[i]) == type(None):
infos[i] = '' infos[i] = ""
admin_text = 'Сообщение от {} {} (@{}):\n\n{}\nid: <code>{}</code>'.format(infos[1], infos[2], infos[3], infos[4], infos[0]) admin_text = "Сообщение от {} {} (@{}):\n\n{}\nid: <code>{}</code>".format(
infos[1], infos[2], infos[3], infos[4], infos[0]
)
send_admin_message(admin_text) send_admin_message(admin_text)
send_message(message.from_user.id, 'Ваше сообщение успешно доставлено!') send_message(message.from_user.id, "Ваше сообщение успешно доставлено!")
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=ParseMode.HTML) bot.send_message(
message.chat.id,
"https://github.com/kr0sh512/Telegram-bot-schedule",
parse_mode=ParseMode.HTML,
)
return return
@bot.message_handler(content_types=["text"])
@bot.message_handler(content_types=['text'])
def text_message(message): def text_message(message):
if message.chat.type == "supergroup" or is_admin(message): if message.chat.type == "supergroup" or is_admin(message):
return return
send_message(message.from_user.id, 'К сожалению, я ещё не умею обрабатывать сообщения.\ send_message(
\nИспользуй команды из меню или напиши /help.') message.from_user.id,
"К сожалению, я ещё не умею обрабатывать сообщения.\
\nИспользуй команды из меню или напиши /help.",
)
return return
def send_message(id, text, thread_id="General", distribution=False):
def send_message(id, text, thread_id='General', distribution=False): if thread_id == "General":
if thread_id == 'General':
thread_id = None thread_id = None
if distribution: if distribution:
@@ -542,19 +670,34 @@ def send_message(id, text, thread_id='General', distribution=False):
return return
try: try:
bot.send_message(chat_id=id, text=text, parse_mode=ParseMode.HTML, message_thread_id=thread_id,disable_web_page_preview=True) bot.send_message(
chat_id=id,
text=text,
parse_mode=ParseMode.HTML,
message_thread_id=thread_id,
disable_web_page_preview=True,
)
except Exception as e: except Exception as e:
time.sleep(5) time.sleep(5)
try: try:
bot.send_message(chat_id=id, text=text, parse_mode=ParseMode.HTML, message_thread_id=thread_id,disable_web_page_preview=True) bot.send_message(
chat_id=id,
text=text,
parse_mode=ParseMode.HTML,
message_thread_id=thread_id,
disable_web_page_preview=True,
)
except Exception as e: except Exception as e:
text_error = 'Error from user: @{} <code>{}</code>\n{}'.format(for_json.return_infos(id)["username"], id, str(e)) text_error = "Error from user: @{} <code>{}</code>\n{}".format(
for_json.return_infos(id)["username"], id, str(e)
)
send_admin_message(text_error) send_admin_message(text_error)
print(id, e) print(id, e)
return return
def send_document(id, file, text = ''):
def send_document(id, file, text=""):
try: try:
bot.send_document(id, file, caption=text) bot.send_document(id, file, caption=text)
except Exception as e: except Exception as e:
@@ -562,21 +705,22 @@ def send_document(id, file, text = ''):
try: try:
bot.send_document(id, file, caption=text) bot.send_document(id, file, caption=text)
except Exception as e: except Exception as e:
text_error = 'Error from user: <code>{}</code>\n{}'.format(id, str(e)) text_error = "Error from user: <code>{}</code>\n{}".format(id, str(e))
send_admin_message(text_error) send_admin_message(text_error)
print(id, e) print(id, e)
return return
if __name__ == "__main__":
if __name__ == '__main__':
for_json.create_schedule_tasks(True) for_json.create_schedule_tasks(True)
send_admin_message('Я перезапустился!') send_admin_message("Я перезапустился!")
print("-------------------------") 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()
while True: while True:
try: try: