def handle_text(message): today, tomorrow = CurrentDates(database_url).get_dates() group = UserPosition(database_url).verification(str(message.chat.id)) weekday_index = datetime.datetime.strptime(today, '%d.%m.%Y').date().weekday() if message.text == 'На сегодня': lessons = Timetable(database_url).get_lessons(group, today) lessons = [lessons[key] for key in sorted(lessons.keys())] if weekday_index == 5: bot.send_message(message.chat.id, 'Выходной день') elif weekday_index == 6: bot.send_message(message.chat.id, 'Выходной день') else: bot.send_message(message.chat.id, 'Расписание на сегодня ({0}):'.format(today[:5])) bot.send_message(message.chat.id, display_schedule(lessons)) if message.text == 'На завтра': lessons = Timetable(database_url).get_lessons(group, tomorrow) lessons = [lessons[key] for key in sorted(lessons.keys())] if weekday_index == 4: bot.send_message(message.chat.id, 'Выходной день') elif weekday_index == 5: bot.send_message(message.chat.id, 'Выходной день') else: bot.send_message( message.chat.id, 'Расписание на завтра ({0}):'.format(tomorrow[:5])) bot.send_message(message.chat.id, display_schedule(lessons)) if message.text == 'Подписаться на эту группу': Subscribers(database_url).create_subscribers() Subscribers(database_url).add_subscriber(str(message.chat.id), group) bot.send_message(message.chat.id, 'Вы подписались на группу {0}.'.format(group))
def handle_text(message): CurrentDates(database_url).add_dates('26.03.2017', '27.03.2017') group = Subscribers(database_url).get_subscriber_group(str( message.chat.id)) today, tomorrow = CurrentDates(database_url).get_dates() weekday_index = datetime.datetime.strptime(today, '%d.%m.%Y').date().weekday() lessons = Timetable(database_url).get_lessons(group, today) lessons_today = [lessons[key] for key in sorted(lessons.keys())] lessons = Timetable(database_url).get_lessons(group, tomorrow) lessons_tomorrow = [lessons[key] for key in sorted(lessons.keys())] if weekday_index == 4: bot.send_message(message.chat.id, 'Расписание на сегодня ({0}):'.format(today[:5])) bot.send_message(message.chat.id, display_schedule(lessons_today)) bot.send_message(message.chat.id, 'Расписание на завтра ({0}):'.format(tomorrow[:5])) bot.send_message(message.chat.id, 'Выходной день.') elif weekday_index == 5: bot.send_message(message.chat.id, 'Выходные дни.') elif weekday_index == 6: bot.send_message(message.chat.id, 'Расписание на сегодня ({0}):'.format(today[:5])) bot.send_message(message.chat.id, 'Выходной день.') bot.send_message(message.chat.id, 'Расписание на завтра ({0}):'.format(tomorrow[:5])) bot.send_message(message.chat.id, display_schedule(lessons_tomorrow)) else: bot.send_message(message.chat.id, 'Расписание на сегодня ({0}):'.format(today[:5])) bot.send_message(message.chat.id, display_schedule(lessons_today)) bot.send_message(message.chat.id, 'Расписание на завтра ({0}):'.format(tomorrow[:5])) bot.send_message(message.chat.id, display_schedule(lessons_tomorrow))