예제 #1
0
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))
예제 #2
0
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))