def command_send_report(m):
# Статистика
if config['STATISTIC_TOKEN'] != '':
track(config['STATISTIC_TOKEN'], m.chat.id, m.text, 'report')
else:
logger.info('report')
cid = m.chat.id
data = m.text.split("/send_report")
if data[1] != '':
report = data[1]
with ScheduleDB(config) as db:
if db.add_report(cid, report):
bot.send_message(cid, "Сообщение принято")
else:
bot.send_message(
cid,
"Случилось что то странное, попробуйте ввести команду заново",
reply_markup=get_date_keyboard())
else:
bot.send_message(
cid,
"Вы отправили пустую строку. Пример: /send_report <сообщение>",
reply_markup=get_date_keyboard())
def command_start(m):
# Статистика
if config['STATISTIC_TOKEN'] != '':
track(config['STATISTIC_TOKEN'], m.chat.id, m.text, 'start')
else:
logger.info('start')
cid = m.chat.id
command_help(m)
try:
with ScheduleDB(config) as db:
user = db.find_user(cid)
if user and user[0] is not None:
bot.send_message(cid,
"Вы уже добавлены в базу данных",
reply_markup=get_date_keyboard())
else:
bot.send_message(
cid,
"Вас ещё нет в базе данных, поэтому пройдите простую процедуру регистрации"
)
command_registration(m)
except BaseException as e:
logger.warning('command start: {0}'.format(str(e)))
bot.send_message(
cid,
"Случилось что то странное, попробуйте ввести команду заново",
reply_markup=get_date_keyboard())
def command_help(m):
# Статистика
if config['STATISTIC_TOKEN'] != '':
track(config['STATISTIC_TOKEN'], m.chat.id, m.text, 'help')
else:
logger.info('help')
cid = m.chat.id
help_text = "Доступны следующие команды: \n"
for key in commands:
help_text += "/" + key + ": "
help_text += commands[key] + "\n"
bot.send_message(cid, help_text, reply_markup=get_date_keyboard())
help_text = (
'Описание кнопок:\nКнопка "Сегодня", как это ни странно выводит расписание на сегодняшний день, '
'причём с учётом типа недели (числитель/знаменатель), но есть один нюанс: если сегодня воскресенье '
'или время больше чем 21:30, то выводится расписание на следующий день\n'
)
bot.send_message(cid, help_text, reply_markup=get_date_keyboard())
guide_url = 'https://github.com/paladin-705/UniversityScheduleBot/wiki/Guide'
help_text = 'Более подробную инструкцию и описание команд (с инструкциями гифками! ^_^) \
вы можете посмотреть по ссылке: {}'.format(guide_url)
bot.send_message(cid, help_text, reply_markup=get_date_keyboard())
def command_auto_posting_off(m):
# Статистика
if config['STATISTIC_TOKEN'] != '':
track(config['STATISTIC_TOKEN'], m.chat.id, m.text, 'auto_posting_off')
else:
logger.info('auto_posting_off')
cid = m.chat.id
try:
db = ScheduleDB(config)
user = db.find_user(cid)
if not user:
bot.send_message(
cid,
"Вас ещё нет в базе данных, поэтому пройдите простую процедуру регистрации"
)
command_registration(m)
return
if db.set_auto_post_time(cid, None, None):
bot.send_message(
cid, "Автоматическая отправка расписания успешно отключена")
else:
bot.send_message(
cid,
"Случилось что то странное, попробуйте ввести команду заново",
reply_markup=get_date_keyboard())
except BaseException as e:
logger.warning('command auto_posting_off: {0}'.format(str(e)))
bot.send_message(
cid, "Случилось что то странное, попробуйте ввести команду заново")
def exams(m):
cid = m.chat.id
# Статистика
track(config['STATISTIC_TOKEN'], cid, m.text, 'exams')
# Если пользователя нет в базе, то ему выведет предложение зарегистрироваться
try:
with ScheduleDB(config) as db:
user = db.find_user(cid)
if not user or user[0] is None:
message = "Вас ещё нет в базе данных, поэтому пройдите простую процедуру регистрации:\n"
message += 'Введите команду(без кавычек):\n\nрегистрация "название вуза" "факультет" "группа"\n\n'
message += 'Если вы допустите ошибку, то просто наберите команду заново.\n'
bot.send_message(cid, message, reply_markup=get_date_keyboard())
except BaseException as e:
bot.send_message(
cid,
'Случилось что то странное, попробуйте ввести команду заново',
reply_markup=get_date_keyboard())
try:
with ScheduleDB(config) as db:
exams_list = db.get_exams(user[0])
message = ''
for exam in exams_list:
message += exam[0].strftime('%d.%m.%Y') + ":\n"
title = ' '.join(str(exam[1]).split())
lecturer = ' '.join(str(exam[2]).split())
classroom = ' '.join(str(exam[3]).split())
message += title + ' | ' + lecturer + ' | ' + classroom + "\n"
message += "------------\n"
if len(message) == 0:
message = 'Похоже расписания экзаменов для вашей группы нет в базе'
except BaseException as e:
message = "Случилось что то странное, попробуйте ввести команду заново"
bot.send_message(cid, message, reply_markup=get_date_keyboard())
def callback_auto_posting(call):
cid = call.message.chat.id
try:
callback_data = re.split(r':', call.data)
# Проверка на соответствие введённых пользователем данных принятому формату
if len(callback_data) != 4:
bot.send_message(
cid,
"Вы отправили пустую строку или строку неправильного формата. Правильный формат ЧЧ:ММ",
reply_markup=get_date_keyboard())
return
hour = ''.join(filter(lambda x: x.isdigit(), callback_data[1]))
minutes = ''.join(filter(lambda x: x.isdigit(), callback_data[2]))
is_today = callback_data[3]
# Проверка на соответствие введённых пользователем данных принятому формату
if not hour.isdigit() or not minutes.isdigit():
bot.send_message(
cid,
"Вы отправили пустую строку или строку неправильного формата. Правильный формат ЧЧ:ММ",
reply_markup=get_date_keyboard())
return
with ScheduleDB(config) as db:
if db.set_auto_post_time(cid,
(hour + ":" + minutes + ":" + "00").rjust(
8, '0'), is_today):
bot.send_message(cid,
"Время установлено",
reply_markup=get_date_keyboard())
else:
bot.send_message(
cid,
"Случилось что то странное, попробуйте ввести команду заново",
reply_markup=get_date_keyboard())
except BaseException as e:
logger.warning('callback_auto_posting: {0}'.format(str(e)))
bot.send_message(
cid, "Случилось что то странное, попробуйте ввести команду заново")
def callback_registration(call):
cid = call.message.chat.id
# Парсинг сообщения указывающего стадию регистрации
# reg : stage : tag
callback_data = re.split(r':', call.data)
# Процедура регистрации проходит в четрые этапа:
# 1 этап: выбор учебного заведения
# 2 этап: выбор факультета
# 3 этап: выбор группы
# 4 этап: добавление данных о принадлежности пользователя к учебному заведению в БД <--
try:
# Статистика
if config['STATISTIC_TOKEN'] != '':
track(config['STATISTIC_TOKEN'], cid, 'stage 4',
'registration-stage-4')
group_id = callback_data[2]
db = ScheduleDB(config)
row = db.get_group(group_id)
user = db.find_user(cid)
if user:
db.update_user(cid, call.message.chat.first_name,
call.message.chat.username, str(row[0][1]))
else:
db.add_user(cid, call.message.chat.first_name,
call.message.chat.username, str(row[0][1]))
bot.send_message(
cid,
"Отлично, вы зарегистрировались, ваша группа: " + row[0][0] +
"\nЕсли вы ошиблись, то просто введиде команду /registration и измените данные",
reply_markup=get_date_keyboard())
bot.send_message(
cid,
"Теперь вы можете настроить автоматическую отправку расписания в заданное вами время,"
" введя команду /auto_posting_on <время>, "
"где <время> должно иметь формат ЧЧ:ММ")
except BaseException as e:
logger.warning('Registration problem: {0}'.format(str(e)))
bot.send_message(
cid,
"Случилось что-то странное, попробуйте начать сначала, введя команду /registration"
)
def command_auto_posting_on(m):
# Статистика
if config['STATISTIC_TOKEN'] != '':
track(config['STATISTIC_TOKEN'], m.chat.id, m.text, 'auto_posting_on')
else:
logger.info('auto_posting_on')
cid = m.chat.id
try:
data = m.text.split("/auto_posting_on")[1].strip()
if re.match(data, r'\d{1,2}:\d\d'):
raise BaseException
except:
bot.send_message(
cid,
"Вы отправили пустую строку или строку неправильного формата. Правильный формат ЧЧ:ММ",
reply_markup=get_date_keyboard())
return None
try:
db = ScheduleDB(config)
user = db.find_user(cid)
if user and user[0] is not None:
keyboard = types.InlineKeyboardMarkup()
callback_button = types.InlineKeyboardButton(
text="На Сегодня", callback_data="ap:{0}:1".format(data))
keyboard.add(callback_button)
callback_button = types.InlineKeyboardButton(
text="На Завтра", callback_data="ap:{0}:0".format(data))
keyboard.add(callback_button)
bot.send_message(
cid,
"Выберите день на который будет приходить расписание:",
reply_markup=keyboard)
else:
bot.send_message(
cid,
"Вас ещё нет в базе данных, поэтому пройдите простую процедуру регистрации"
)
command_registration(m)
except BaseException as e:
logger.warning('command auto_posting_on: {0}'.format(str(e)))
bot.send_message(
cid, "Случилось что то странное, попробуйте ввести команду заново")
def response_msg(m):
cid = m.chat.id
if m.text in ScheduleType:
# Статистика
if config['STATISTIC_TOKEN'] != '':
track(config['STATISTIC_TOKEN'], m.chat.id, m.text, 'schedule')
else:
logger.info('message: {0}'.format(m.text))
# По умолчанию week_type равен -1 и при таком значении будут выводится все занятия,
# т.е и для чётных и для нечётных недель
week_type = -1
if m.text == "Вся неделя":
days = ScheduleType[m.text]
elif m.text == "Сегодня":
today = datetime.now()
# Если запрашивается расписание на сегодняшний день,
# то week_type равен остатку от деления на 2 номера недели в году, т.е он определяет чётная она или нечётная
week_type = get_week_type(today)
# Если сегодня воскресенье, то показывается расписание на понедельник следующей недели
# Также в этом случае, как week_type используется тип следующей недели
if datetime.weekday(today) == 6:
today += timedelta(days=1)
week_type = (week_type + 1) % 2
days = [daysOfWeek[datetime.weekday(today)]]
elif m.text == 'Завтра':
tomorrow = datetime.now()
tomorrow += timedelta(days=1)
# Если запрашивается расписание на сегодняшний день,
# то week_type равен остатку от деления на 2 номера недели в году, т.е он определяет чётная она или нечётная
week_type = get_week_type(tomorrow)
# Если сегодня воскресенье, то показывается расписание на понедельник следующей недели
# Также в этом случае, как week_type используется тип следующей недели
if datetime.weekday(tomorrow) == 6:
tomorrow += timedelta(days=1)
week_type = (week_type + 1) % 2
days = [daysOfWeek[datetime.weekday(tomorrow)]]
else:
days = [ScheduleType[m.text]]
for day in days:
try:
with ScheduleDB(config) as db:
user = db.find_user(cid)
if user and user[0] is not None:
result = create_schedule_text(user[0], day, week_type)
for schedule in result:
bot.send_message(cid,
schedule,
reply_markup=get_date_keyboard())
else:
bot.send_message(
cid,
"Вас ещё нет в базе данных, поэтому пройдите простую процедуру регистрации"
)
command_registration(m)
except BaseException as e:
logger.warning('response_msg: {0}'.format(str(e)))
bot.send_message(
cid,
"Случилось что то странное, попробуйте ввести команду заново"
)
else:
# Статистика
if config['STATISTIC_TOKEN'] != '':
track(config['STATISTIC_TOKEN'], m.chat.id, m.text, 'unknown')
else:
logger.info('unknown message: {0}'.format(m.text))
bot.send_message(cid,
"Неизвестная команда",
reply_markup=get_date_keyboard())