"""Задание на расчет интервалов, и вычисление времени, которое ученик
и учитель присутствовали на уроке одновременно."""
from intervals_cleaning import normilize_intervals

tests = [
    {
        "data": {
            "lesson": [1594663200, 1594666800],
            "pupil": [
                1594663340,
                1594663389,
                1594663390,
                1594663395,
                1594663396,
                1594666472,
            ],
            "tutor": [1594663290, 1594663430, 1594663443, 1594666473],
        },
        "answer": 3117,
    },
    {
        "data": {
            "lesson": [1594702800, 1594706400],
            "pupil": [
                1594702789,
                1594704500,
                1594702807,
                1594704542,
                1594704512,
                1594704513,
                1594704564,
                1594705150,
                1594704581,
                1594704582,
                1594704734,
                1594705009,
                1594705095,
                1594705096,
                1594705106,
                1594706480,
                1594705158,
                1594705773,
                1594705849,
                1594706480,
                1594706500,
                1594706875,
                1594706502,
                1594706503,
                1594706524,
                1594706524,
                1594706579,
                1594706641,
            ],
            "tutor": [
                1594700035,
                1594700364,
                1594702749,
                1594705148,
                1594705149,
                1594706463,
            ],
        },
        "answer": 3577,
    },
    {
        "data": {
            "lesson": [1594692000, 1594695600],
            "pupil": [1594692033, 1594696347],
            "tutor": [1594692017, 1594692066, 1594692068, 1594696341],
        },
        "answer": 3565,
    },
]


def get_intersection(interval_A, interval_B) -> list:
    """Возвращает пересечение двух интервалов.

    Пересечение двух интервалов -- это наибольшее из начал A и B, и
    наименьшее из концов A и B."""
    # если один из интервалов пустой, то пересечение будет пустым
    if not interval_A or not interval_B:
        return []

    # если какой-то из интервалов имеет нулевую длину (начало=конец),
    # то пересечение пустое
    if interval_A[0] == interval_A[1] or interval_B[0] == interval_B[1]:
        return []

    # нужно проверить, что интервалы вообще пересекаются
    # А и Б не пересекаются, если
    # начало А > конец Б, или если начало Б > конец А
    if interval_A[0] > interval_B[1] or interval_B[0] > interval_A[1]:
        return []

    # начало интервала это наименьшее из начал интервалов А и Б
    # конец интервала это наименьший из концов интервалов А и Б
    interval_begin = max(interval_A[0], interval_B[0])
    interval_end = min(interval_A[1], interval_B[1])
    return [interval_begin, interval_end]


def appearance(intervals) -> int:
    """Возвращает время одновременного присутствия ученика и учителя на
    уроке в секундах."""
    lesson = intervals["lesson"]
    pupil = normilize_intervals(intervals["pupil"])
    tutor = normilize_intervals(intervals["tutor"])

    # получаем все интервалы студента на уроке
    pupil_on_lesson = [
        get_intersection(lesson, p) for p in pupil if get_intersection(lesson, p)
    ]

    # по очереди сравниваем каждый из интервалов преподавателя с
    # каждым из интервалов пересечения студента на уроке
    tutor_on_pupil = [
        get_intersection(pol, t) for t in tutor for pol in pupil_on_lesson
    ]

    # на данном этапе мы уже имеем интересующие нас интервалы
    # (пересечение ученик-учитель-урок); остается узнать
    # продолжительность каждого из этих интервалов в секундах и
    # сложить их.
    answer = sum([intr[1] - intr[0] for intr in tutor_on_pupil if intr])
    return answer


if __name__ == "__main__":
    for i, test in enumerate(tests):
        test_answer = appearance(test["data"])
        assert (
            test_answer == test["answer"]
        ), f'Error on test case {i}, got {test_answer}, expected {test["answer"]}'