root/hodgestar/PythonCode/CursesMediaBrowser/browse.py

#!/usr/bin/env python

import os
import sys
import curses
import curses.wrapper
import curses.ascii

SOURCES = [
    'cd.txt',
    'Movies',
    'Series',
]

def titles_from_file(filename):
    """Return a list of (title, info) pairs from the filename.
    
       Each line of file represents a list of titles
       separate by semi-colons.
       
       Returned info consists of the CD number (i.e. line in
       the file) and full list of titles on that CD.
       """
    f = file(filename)
    title_pairs = []
    try:
        for i, line in enumerate(f):
            titles = [x.strip() for x in line.split(';')]
            info = "CD #: %d\n" % i + "\n".join(titles)
            for title in titles:
                title_pairs.append((title, info))
    finally:
        f.close()
    return title_pairs

def titles_from_dir(dirname):
    """Return a list of (title, info) pairs from the dirname.
    
       Each entry in dirname is a title.
       None is returned for info.
       """
    title_pairs = []
    def walker(arg, dirname, fnames):
        """Walk directory, empty fnames so sub folders aren't processed."""
        for entry in fnames:
            title_pairs.append((entry, None))
        del fnames[:]

    os.path.walk(dirname, walker, None)
    return title_pairs

def compile_list(sources):
    """Return a list of titles (title, info, src) triples."""
    titles = []
    for src in sources:
        if os.path.isfile(src):
            titles.extend([(title, info, src)
                            for title, info in titles_from_file(src)])
        elif os.path.isdir(src):
            titles.extend([(title, info, src)
                            for title, info in titles_from_dir(src)])
    return titles

def matches(search, title):
    """Return true if title matches seach.
    
       To match, the characters in search must occur in the same
       order in title.
       """
    pos = 0
    title = title.lower()
    for c in search:
        pos = title.find(c, pos) + 1
        if pos == 0:
            return False
    return True

def setup_windows(window):
    """Setup sub windows."""
    h, w = window.getmaxyx()

    search_h, search_lines = 0, 1
    sep1_h = search_h + search_lines

    info_lines = 10
    sep2_h = h - info_lines -1
    info_h = sep2_h + 1

    list_h, list_lines = sep1_h + 1, sep2_h - sep1_h - 1

    window.hline(sep1_h, 0, "-", w)
    window.hline(sep2_h, 0, "-", w)

    search_window = window.derwin(search_lines, w, search_h, 0)
    list_window = window.derwin(list_lines, w, list_h, 0)
    info_window = window.derwin(info_h, 0)

    return search_window, list_window, info_window, list_lines

def curses_gui(window, titles):
    """GUI for curses."""

    search_window, list_window, info_window, list_len = setup_windows(window)
    list_window.scrollok(1)

    KEY_UP = [ 259 ]
    KEY_DOWN = [ 258 ]
    KEY_BKSPC = [ 260, 263 ]

    titles.sort()
    all_titles = titles[:]

    list_pos = [0]
    cursor_offset = [0]
    def list_populate():
        list_window.erase()
        highlight_pos = min(list_len, len(titles)) - cursor_offset[0]
        for i, (title, info, src) in \
              enumerate(reversed(titles[list_pos[0]+1:list_pos[0]+list_len])):
            if i + 1 == highlight_pos:
                list_window.insstr(title, curses.A_STANDOUT)
            else:
                list_window.insstr(title)
            list_window.insertln()
        if list_pos[0] < len(titles):
            if highlight_pos == min(list_len, len(titles)):
                list_window.insstr(titles[list_pos[0]][0], curses.A_STANDOUT)
            else:
                list_window.insstr(titles[list_pos[0]][0])
        list_window.refresh()
    def list_up():
        if list_pos[0] > 0:
            list_pos[0] -= 1
        elif cursor_offset[0] > 0:
            cursor_offset[0] -= 1
        list_populate()
        info_populate()
    def list_down():
        if list_pos[0] + list_len < len(titles):
            list_pos[0] += 1
        elif cursor_offset[0] + list_pos[0] < len(titles) - 1:
            cursor_offset[0] += 1
        list_populate()
        info_populate()

    def info_populate():
        info_window.erase()
        if titles:
            title, info, src = titles[list_pos[0]+cursor_offset[0]]
            if info:
                for line in reversed(info.split("\n")):
                    info_window.insstr(line)
                    info_window.insertln()
            info_window.insstr(" %s" % src)
            info_window.insstr("Source:", curses.A_BOLD)
            info_window.insertln()
            info_window.insstr(" %s" % title)
            info_window.insstr("Title:", curses.A_BOLD)
        info_window.refresh()

    search_key = [""]
    def update_search(c):
        list_pos[0] = 0
        cursor_offset[0] = 0
        if c is None:
            pass
        elif c in KEY_BKSPC:
            search_key[0] = search_key[0][:-1]
        else:
            search_key[0] += chr(c).lower()
        search_window.erase()
        search_window.insstr(" " + search_key[0])
        search_window.insstr("Search:", curses.A_BOLD)
        search_window.refresh()
        titles[:] = [t for t in all_titles if matches(search_key[0], t[0])]
        list_populate()
        info_populate()

    c = None
    update_search(None)
    while c != ord('q'):
        c = window.getch()
        if c in KEY_UP:
            list_up()
        elif c in KEY_DOWN:
            list_down()
        elif curses.ascii.isalnum(c) or c in KEY_BKSPC:
            update_search(c)
        else:
            pass

def main(args):
    """Allow browsing of titles."""
    titles = compile_list(SOURCES)
    curses.wrapper(curses_gui, titles)

if __name__ == "__main__":
    sys.exit(main(sys.argv))