from datetime import datetime, timedelta
from colorama import Fore

easy = Fore.GREEN
medium = Fore.LIGHTYELLOW_EX
hard = Fore.YELLOW
impossible = Fore.RED

subjects = [medium + 'Английский язык', impossible + 'Биология', medium + 'География', impossible + 'Информатика',
            hard + 'История', easy + 'Литература', hard + 'Математика', medium + 'Обществознание', easy + 'ОБЖ',
            hard + 'Русский язык', impossible + 'Физика', easy + 'Физ-ра', medium + 'Химия']

for subj in subjects:
    subjects[subjects.index(subj)] += Fore.RESET


timetable = [[1, 0, 4, 10, 6, 9, 5],
             [7, 0, 6, 8, 9, 9, 2],
             [0, 4, 11, 5, 12, 5],
             [11, 7, 3, 1, 6, 6, 0],
             [2, 7, 6, 12, 10, 6]]

lesson_duration = timedelta(minutes=40)
lessons_time = ['8:30', '9:20', '10:15', '11:10', '12:05', '13:00', '13:55']

for time in lessons_time:
    lessons_time[lessons_time.index(time)] = datetime.strptime(time, '%H:%M')


def show_all_subjects():
    for subject in subjects:
        print(str(subjects.index(subject)) + ') ' + subject)


def show_next_lessons(day=datetime.now().isoweekday()):
    now_time = datetime.now().time()
    now_time = datetime.strptime('8:40', '%H:%M').time()

    day -= 1
    if day > 4:
        print('Сеголня нет уроков')
    else:
        print('Расписание на сегодня:')
        i = 1
        for subject in timetable[day]:

            lesson_start_time = lessons_time[timetable[day].index(subject)].time()

            if now_time < lesson_start_time:
                print(str(i) + ') ' + subjects[subject])
            i += 1

    day += 1
    if day > 4:
        day = 0
    i = 1
    print('\n\nРасписание на след. день:')
    for subject in timetable[day]:
        print(str(i) + ') ' + subjects[subject])
        i += 1


def show_current_lesson(day=datetime.now().isoweekday()):
    now_time = datetime.now().time()
    now_time = datetime.strptime('8:40', '%H:%M').time()

    day -= 1

    if not day > 4:
        for subject in timetable[day]:

            lesson_start_time = lessons_time[timetable[day].index(subject)].time()
            lesson_end_time = (lessons_time[timetable[day].index(subject)] + lesson_duration).time()

            if lesson_start_time < now_time < lesson_end_time:
                print('Сейчас: ' + subjects[subject] + Fore.RESET)
                print('Закончится через ' + Fore.CYAN + str(round(((lessons_time[timetable[day].index(subject)]
                                                                    + lesson_duration)
                                                                   - datetime.strptime('8:40', '%H:%M')).seconds / 60))
                      + ' мин\n' + Fore.RESET)
                break


def show_time():
    now_time = datetime.now()
    now_time = datetime.strptime('8:40', '%H:%M')
    print('Время: ' + Fore.CYAN + datetime.strftime(now_time, '%H:%M') + Fore.RESET)


show_time()
show_current_lesson()
show_next_lessons()