university-setup/scripts/rofi-lectures.py

#!/usr/bin/python3
from courses import Courses
from rofi import rofi
from utils import generate_short_title
from config_loader import MAX_LEN

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

sorted_lectures = sorted(lectures, key=lambda l: -l.number)

options = [
    "{number: >2}. <b>{title: <{fill}}</b> <span size='smaller'>{date}  ({week})</span>".format(
        fill=MAX_LEN,
        number=lecture.number,
        title=generate_short_title(lecture.title),
        date=lecture.date.strftime('%a %d %b'),
        week=lecture.week
    )
    for lecture in sorted_lectures
]

key, index, selected = rofi('Select lecture', options, [
    '-lines', 5,
    '-markup-rows',
    '-kb-row-down', 'Down',
    '-kb-custom-1', 'Alt+n',
    '-kb-custom-2', 'Alt+m',
    '-kb-custom-3', 'Alt+s'
])

if key == 0:
    sorted_lectures[index].edit()
elif key == 1:
    new_lecture = notes.new_lecture()
    new_lecture.edit()
elif key == 2:
    notes.edit_master()
elif key == 3:
    notes.edit_full()