#!/usr/bin/python3
from courses import Courses
from rofi import rofi

script = Courses().current.notes
lectures = script.lectures

commands = ['last', 'prev-last', 'all', 'prev']
options = ['Current lecture', 'Last two lectures', 'All lectures', 'Previous lectures']

key, index, selected = rofi('Select view', options, [
    '-lines', 4,
    '-auto-select'
])

if index >= 0:
    command = commands[index]
else:
    command = selected

lecture_range = lectures.parse_range_string(command)
script.update_lectures_in_master(lecture_range)
script.compile_master()