simplesync.py

#!/usr/bin/env python
#
#       filename - desc
#       
#       Copyright 2009 John Tyree <johntyree@gmail.com>
#       
#       This program is free software; you can redistribute it and/or modify
#       it under the terms of the GNU General Public License as published by
#       the Free Software Foundation; either version 3 of the License, or
#       (at your option) any later version.
#       
#       This program is distributed in the hope that it will be useful,
#       but WITHOUT ANY WARRANTY; without even the implied warranty of
#       MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#       GNU General Public License for more details.
#       
#       You should have received a copy of the GNU General Public License
#       along with this program; if not, write to the Free Software
#       Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
#       MA 02110-1301, USA.

import gtk, simplesync_db

class dbView:
    '''Main window for viewing simplesync musicDB'''

    def __init__(self, db):
        # Initialize model
        self.db = db
        self.testBool = True

        ## Generate tooltips
        self.tooltips = gtk.Tooltips()

        ## Keyboard Accelerators
        self.AccelGroup = gtk.AccelGroup()
        self.AccelGroup.connect_group(ord('Q'), gtk.gdk.CONTROL_MASK, gtk.ACCEL_LOCKED, lambda w, x, y, z: gtk.main_quit())

        # Track view
        self.tree = gtk.TreeView()
        self.titleCol = gtk.TreeViewColumn("Title")
        self.artistCol = gtk.TreeViewColumn("Artist")
        self.albumCol = gtk.TreeViewColumn("Album")
        self.yearCol = gtk.TreeViewColumn("Year")
        self.genreCol = gtk.TreeViewColumn("Genre")
        self.syncCol = gtk.TreeViewColumn("Sync")
        col_cell_text = gtk.CellRendererText()
        for i, col in enumerate((self.titleCol, self.artistCol, self.albumCol, self.genreCol, self.yearCol)):
            col.pack_start(col_cell_text, True)
            col.add_attribute(col_cell_text, "text", i)
            col.set_resizable(True)
            col.set_clickable(True)
            col.set_reorderable(True)
            col.connect('clicked', lambda w: self.sortByColumn(w))
        for i, col in enumerate((self.syncCol, self.titleCol, self.artistCol, self.albumCol, self.genreCol, self.yearCol)):
            self.tree.append_column(col)
        self.titleCol.set_expand(True)
        col_cell_toggle = gtk.CellRendererToggle()
        self.syncCol.add_attribute( col_cell_toggle, "active", 1)
        col_cell_toggle.set_property('activatable', True)
        col_cell_toggle.connect('toggled', self.toggle_callback, db)
        self.syncCol.pack_start(col_cell_toggle)

        # Track list
        self.listStore = gtk.ListStore(str, str, str, str, str, int, bool)

        # Track window
        self.scroll = gtk.ScrolledWindow()
        self.scroll.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
        self.scroll.add(self.tree)

        # Search bar
        self.searchBar = gtk.Entry()
        self.searchBar.connect('activate', self.search_callback)
        self.tooltips.set_tip(self.searchBar, "Enter query")
        self.searchBar.set_text("Enter query")
        self.searchBar.select_region(0, len(self.searchBar.get_text()))

        #Window layout
        self.vbox1 = gtk.VBox(False, 0)
        self.vbox1.pack_start(self.scroll, True, True, 1)
        self.vbox1.pack_start(self.searchBar, False, False, 1)

        #Initialize
        self.dbwindow = gtk.Window(gtk.WINDOW_TOPLEVEL)
        self.dbwindow.set_title("simplesync")
        self.dbwindow.set_size_request(900, 150)
        self.dbwindow.connect("destroy", lambda w: gtk.main_quit())
        self.dbwindow.add_accel_group(self.AccelGroup)
        self.dbwindow.add(self.vbox1)
        self.searchBar.grab_focus()
        self.dbwindow.show_all()

    def search_callback(self, entry):
        print "Search for '%s'." % entry.get_text()
        return 0

    def toggle_callback(self, cell, toggle, db):
        #db.cursor.execute("UPDATE file SET sync = ?", (self.listStore[toggle][-1],))
        self.testBool = not self.testBool
        print "Toggled %s to %s" % (self.listStore[toggle][0], self.testBool)
        return

    def column_callback(self, column):
        print "Sort by %s." % column.get_title()
        return

def main():
    db = simplesync_db.musicDB(':memory:')
    db.addDir("/media/disk/Music/0-9")
    window = dbView(db)
    for track in db.allList():
        window.listStore.append((track['relpath'], track['title'], track['artist'], track['album'], track['genre'], track['year'], track['sync']))
    window.tree.set_model(window.listStore)
    gtk.main()
    return 0

if __name__ == '__main__': main()