if message.text.lower() in ('cancel', '❌ cancel ❌'):

if message.from_user.id in data:

del data[message.from_user.id]

bot.send_message(message.chat.id, 'Homework {} was cancelled.'.format('adding' if adding else 'deleting'),

reply_markup=MARKUP)

return True

try:

text = message.text.split()[0].split('.')

day, month = map(int, text)

current_date = datetime.datetime.now(TIMEZONE).date()

current_month = current_date.month

if current_month < month + 6:

year = current_date.year

else:

year = current_date.year + 1

date = datetime.date(year=year, month=month, day=day).toordinal()

if date < current_date.toordinal():

bot.send_message(message.chat.id, 'Error: Past date.\n'

'Please enter a future or present date.')

if function:

bot.register_next_step_handler(message, function)

else:

return date

except Exception as e:

if function:

bot.send_message(message.chat.id, 'Error: Incorrect date ({}).\n'

'Make sure to type it in <b>DD.⁠MM</b> format.'.format(str(e)),

parse_mode='HTML')

bot.register_next_step_handler(message, function)

bot.send_message(message.chat.id, 'This is <b>Homework Planning Bot</b>.\n'

'It can monitor your homework.\n'

'To set your schedule, attach .xlsx file with the specific structure.\n'

'To get the blank Excel form with this structure and the information '

'on how to fill it, type /form.\n'

'To view your schedule and homework, press <i>Today</i>, <i>Tomorrow</i>, '

'<i>Week</i> or <i>All</i> or type any future date in <b>DD.⁠MM</b> format.\n'

'To add new homework, press <i>Add</i> and follow instructions.\n'

'To delete existing homework, press <i>Delete</i>.\n'

'You can cancel adding or deleting homework by typing <i>Cancel</i> or pressing '

bot.send_message(message.chat.id, 'Read and follow the instructions to set your schedule.\n'

'In this Excel file there are three main parts.\n'

'The first one (columns <b>A-D</b>) is for the list of all your subjects and '

'the other two (columns <b>F-L</b> and <b>N-T</b>) are for their order.\n\n'

'At first, you need to type your subjects into the column named <i>Subject</i>.\n'

"You can also add your teacher's name or specify the room where each subject "

"takes place, but this is optional.\n\n"

'At second, you need to specify their order.\n'

'It is done by typing indexes of your subjects (displayed in the leftmost column)'

' into the cells of the tables labeled <i>Top (odd) week</i> and '

'<i>Bottom (even) week</i>.\n'

'If you have the same schedule for all 2 types of weeks, you have to fill '

'both of these tables with the exact same data.\n',

reply_markup=MARKUP, parse_mode='HTML')

with open('files/form.xlsx', mode='rb') as form_file:

if not message.document.file_name.endswith('.xlsx'):

bot.send_message(message.chat.id, 'Error: Unsupported file type.')

return

file_info = bot.get_file(message.document.file_id)

file = bot.download_file(file_info.file_path)

path = 'files/' + str(message.from_user.id) + '.xlsx'

with open(path, 'wb') as new_file:

new_file.write(file)

markup = ReplyKeyboardMarkup(resize_keyboard=True).add('Yes', 'No')

bot.send_message(message.chat.id, 'Are you sure you want to change your schedule?\n'

'This will delete all your recorded homework.', reply_markup=markup)

if message.text is None or message.text.lower() not in ('yes', 'no'):

bot.send_message(message.chat.id, 'Error: Incorrect response.\nPlease choose one of the options.')

bot.register_next_step_handler(message, handle_change_schedule_answer)

return

path = 'files/' + str(message.from_user.id) + '.xlsx'

if message.text.lower() == 'yes':

try:

with open(path, 'rb') as file:

set_schedule(message.from_user.id, file)

except Exception as e:

bot.send_message(message.chat.id, 'Error: {}.'.format(str(e)), reply_markup=MARKUP)

else:

bot.send_message(message.chat.id, 'New schedule successfully set.', reply_markup=MARKUP)

else:

bot.send_message(message.chat.id, "Schedule wasn't changed.", reply_markup=MARKUP)

try:

text = message.text.lower()

if text == 'today':

date = datetime.datetime.now(TIMEZONE).date().toordinal()

bot.send_message(message.chat.id, get_schedule(message.from_user.id, date), parse_mode='HTML')

elif text == 'tomorrow':

date = datetime.datetime.now(TIMEZONE).date().toordinal() + 1

bot.send_message(message.chat.id, get_schedule(message.from_user.id, date), parse_mode='HTML')

elif text == 'week':

date = datetime.datetime.now(TIMEZONE).date().toordinal()

for i in range(7):

bot.send_message(message.chat.id, get_schedule(message.from_user.id, date), parse_mode='HTML')

date += 1

elif text == 'all':

dates = get_dates(message.from_user.id, ordinal=True)

if not dates:

bot.send_message(message.chat.id, 'You have no recorded homework.')

return

for date in dates:

bot.send_message(message.chat.id, get_schedule(message.from_user.id, date), parse_mode='HTML')

elif text == 'add':

subjects = get_subjects(message.from_user.id)

markup = ReplyKeyboardMarkup(resize_keyboard=True).add(*subjects, row_width=1).add('❌ Cancel ❌')

bot.send_message(message.chat.id, 'Choose the subject.', reply_markup=markup)

bot.register_next_step_handler(message, handle_subject)

elif text == 'delete':

dates = get_dates(message.from_user.id)

if not dates:

bot.send_message(message.chat.id, 'You have no recorded homework to delete.')

return

markup = ReplyKeyboardMarkup(resize_keyboard=True).add(*dates, row_width=3).add('❌ Cancel ❌')

bot.send_message(message.chat.id, 'Choose the date of the homework you wish to delete.\n'

'(Or type it as <b>DD.⁠MM</b>).', reply_markup=markup, parse_mode='HTML')

bot.register_next_step_handler(message, handle_existing_date)

else:

date = process_date(message)

if date:

bot.send_message(message.chat.id, get_schedule(message.from_user.id, date), parse_mode='HTML')

elif date is False:

bot.send_message(message.chat.id, "Bot didn't understand you.", reply_markup=MARKUP)

except Exception as e:

try:

if message.text is None:

bot.send_message(message.chat.id, 'Error: Empty response.\nPlease choose one of the options.')

bot.register_next_step_handler(message, handle_subject)

return

elif check_cancel(message):

return

subjects = get_subjects(message.from_user.id)

if message.text not in subjects:

bot.send_message(message.chat.id, 'Error: Incorrect response.\nPlease choose one of the options.')

bot.register_next_step_handler(message, handle_subject)

return

index = subjects.index(message.text)

data[message.from_user.id] = [index]

ordinal_date = datetime.datetime.now(TIMEZONE).date().toordinal() + 1

markup = ReplyKeyboardMarkup(resize_keyboard=True).add('Next lesson').add('Today', 'Tomorrow')

dates = []

for i in range(12):

ordinal_date += 1

date = datetime.date.fromordinal(ordinal_date)

dates.append('{}.{} ({})'.format(str(date.day).zfill(2), str(date.month).zfill(2),

SHORT_WEEK_DAYS[date.weekday()]))

markup.add(*dates, row_width=3).add('❌ Cancel ❌')

bot.send_message(message.chat.id, 'Choose the deadline: press one of the buttons '

'or type your own date in <b>DD.⁠MM</b> format.',

reply_markup=markup, parse_mode='HTML')

bot.register_next_step_handler(message, handle_new_date)

except Exception as e:

try:

if message.text is None:

bot.send_message(message.chat.id, 'Error: Empty date message.')

bot.register_next_step_handler(message, handle_new_date)

return

elif check_cancel(message):

return

text = message.text.lower()

if text == 'next lesson':

data[message.from_user.id].append(None)

elif text == 'today':

date = datetime.datetime.now(TIMEZONE).date().toordinal()

data[message.from_user.id].append(date)

elif text == 'tomorrow':

date = datetime.datetime.now(TIMEZONE).date().toordinal() + 1

data[message.from_user.id].append(date)

else:

date = process_date(message, handle_new_date)

if not date:

return

data[message.from_user.id].append(date)

subject, date = data[message.from_user.id]

if date and in_schedule(message.from_user.id, date, subject):

bot.send_message(message.chat.id, 'Is this deadline set for the lesson or the end of the day?',

reply_markup=ReplyKeyboardMarkup(resize_keyboard=True)

.add('Lesson', 'Day').add('❌ Cancel ❌'))

bot.register_next_step_handler(message, handle_type)

return

elif date:

data[message.from_user.id].append(False)

else:

data[message.from_user.id].append(True)

bot.send_message(message.chat.id, 'Write homework description. Type <i>Cancel</i> to cancel adding the '

'homework.', reply_markup=ReplyKeyboardRemove(), parse_mode='HTML')

bot.register_next_step_handler(message, handle_description)

except Exception as e:

try:

if message.text is None or message.text.lower() not in ('lesson', 'day', 'cancel', '❌ cancel ❌'):

bot.send_message(message.chat.id, 'Error: Incorrect response.\nPlease choose one of the options.')

bot.register_next_step_handler(message, handle_type)

return

elif check_cancel(message):

return

text = message.text.lower()

if text == 'lesson':

data[message.from_user.id].append(True)

elif text == 'day':

data[message.from_user.id].append(False)

bot.send_message(message.chat.id, 'Write homework description. Type <i>Cancel</i> to cancel adding the '

'homework.', reply_markup=ReplyKeyboardRemove(), parse_mode='HTML')

bot.register_next_step_handler(message, handle_description)

except Exception as e:

try:

if message.text is None:

bot.send_message(message.chat.id, 'Error: Empty description.')

bot.register_next_step_handler(message, handle_description)

return

elif check_cancel(message):

return

subject, date, for_lesson = data.pop(message.from_user.id)

date = add_homework(message.from_user.id, subject, date, for_lesson, message.text)

bot.send_message(message.chat.id, 'Homework successfully added:')

bot.send_message(message.chat.id, get_schedule(message.from_user.id, date),

reply_markup=MARKUP, parse_mode='HTML')

except Exception as e:

if message.text is None:

bot.send_message(message.chat.id, 'Error: Empty date message.')

bot.register_next_step_handler(message, handle_existing_date)

return

elif check_cancel(message, adding=False):

return

text = message.text.lower()

if text == 'today':

date = datetime.datetime.now(TIMEZONE).date().toordinal()

elif text == 'tomorrow':

date = datetime.datetime.now(TIMEZONE).date().toordinal() + 1

else:

date = process_date(message, handle_existing_date)

if not date:

return

homework = get_homework(message.from_user.id, date)

if not homework:

bot.send_message(message.chat.id, 'You have no recorded homework for that date.\nChoose the date again.')

bot.register_next_step_handler(message, handle_existing_date)

return

data[message.from_user.id] = date

markup = ReplyKeyboardMarkup(resize_keyboard=True).add(*homework, row_width=1).add('❌ Cancel ❌')

bot.send_message(message.chat.id, 'Choose the homework to delete.', reply_markup=markup)

if message.text is None:

bot.send_message(message.chat.id, 'Error: Empty homework message.')

bot.register_next_step_handler(message, handle_homework)

return

elif check_cancel(message, adding=False):

return

elif message.from_user.id not in data:

bot.send_message(message.chat.id, 'Error: Date was lost.\n'

'Please try starting over.', reply_markup=MARKUP)

return

try:

date = delete_homework(message.from_user.id, data[message.from_user.id], message.text)

bot.send_message(message.chat.id, 'Homework deleted successfully:', reply_markup=MARKUP)

bot.send_message(message.chat.id, get_schedule(message.from_user.id, date),

reply_markup=MARKUP, parse_mode='HTML')

del data[message.from_user.id]

except Exception as e:

bot.send_message(message.chat.id, 'Error: {}.\n'

'Choose the homework you want to delete again.'.format(e))

notifications = get_notifications()

if not notifications:

return

for n in notifications:

schedule.every().day.do(delete_past_homework)

schedule.every().day.at('18:00').do(send_notifications)

while True:

schedule.run_pending()