root/trunk/msync.py

#!/usr/bin/env python

import os
import sys
import msync
import optparse

def main(argv=None):
    if argv is None or len(argv) == 1:
        msync.help.print_help()
        return
    argv.pop(0) # pop the called filename
    if argv[0] == 'help':
        argv.pop(0)
        msync.help.print_help(argv)
    elif argv[0] in ('library', 'lib'):
        argv.pop(0)
        library(argv)
    elif argv[0] in ('playlist', 'pl'):
        argv.pop(0)
        playlist(argv)
    elif argv[0] in ('synclist', 'sl'):
        argv.pop(0)
        synclist(argv)
    else:
        msync.help.print_help(argv)

def library_file():
    return os.path.normpath(os.path.expanduser('~/.config/msync.db'))

def library(argv=None):
    lib = msync.Library(msync.MsyncDB(library_file()))
    if argv is None or len(argv) == 0:
        msync.help.print_help(['lib'])
    elif argv[0] == 'add':
        if len(argv) == 1:
            msync.help.print_library_help(argv)
            return
        try:
            lib.add(argv[1])
        except msync.PathExistsInLibrary:
            print "Path already exists in library"
    elif argv[0] == 'check':
        if len(argv) == 1:
            argv.append(None)
        print 'The following files can not be accessed:'
        for file in sorted(lib.check(argv[1])):
            print '    %s' % file
    elif argv[0] in ('delete', 'del', 'remove', 'rm'):
        if len(argv) == 1:
            msync.help.print_library_help(argv)
            return
        if lib.remove(argv[1]) is False:
            print "error removing path from library"
    elif argv[0] in ('list', 'ls'):
        if len(argv) == 1:
            argv.append(None)
        if argv[1] is not None and argv[1].startswith('--'):
            if len(argv) == 2:
                print 'You must specify an argument with the option'
                return
            try:
                files = sorted(lib.list(option=argv[1].lstrip('-'), path=argv[2]))
            except msync.InvalidOption:
                print '%s is an invalid option' % argv[1]
                return
        else:
            files = sorted(lib.list(path=argv[1]))

        for file in files:
            print str(file.encode('utf-8'))
    elif argv[0] == 'paths':
        for path in sorted(lib.getLibraryPaths()):
            print path
    elif argv[0] in ('playlists', 'pls'):
        for pls in sorted(lib.getPlaylistsNoFiles(), key=msync.Playlist.getName):
            print pls.name
    elif argv[0] in ('rescan', 'rs', 'update', 'up'):
        if len(argv) == 1:
            argv.append(None)
        if argv[0] in ('rescan', 'rs'):
            added, removed, updated = lib.update(argv[1], rescan=True)
        else:
            added, removed, updated = lib.update(argv[1])
        if len(added) != 0:
            print 'Files added:'
            for file in added:
                print '    %s' % file
            print
        else:
            print 'No files added.'
        if len(removed) != 0:
            print 'Files removed:'
            for file in removed:
                print '    %s' % file
            print
        else:
            print 'No files removed.'
        if len(updated) != 0:
            print 'Files updated:'
            for file in updated:
                print '    %s' % file
            print
        else:
            print 'No files updated.'
    else:
        msync.help.print_help(['lib'])

def playlist(argv=None):
    if argv is None or len(argv) < 2:
        msync.help.print_help(['pl'])
        return
    db = msync.MsyncDB(library_file())
    pl = msync.Playlist(argv.pop(0))
    lib = msync.Library(db)
    pl.setDB(db)

    if argv[0] in ('create','new'):
        try:
            pl.write()
        except msync.PlaylistExists:
            print 'Playlist %s already exists' % pl.name
        return
    elif argv[0] == 'sync' and len(argv) >= 3 and \
            ( argv[1] == '--remove' or argv[1] == '-r' ):
        added, removed, updated = pl.sync(argv[2], remove=True)
        print 'Playlist deleted and %d files removed' % removed
        return

    try:
        pl.get()
    except msync.PlaylistDoesNotExist:
        print '''Playlist does not exist. Try running with the arguments 'library playlists'
to see which playlists are available, or with 'create' to create a new playlist.'''
        return

    if argv[0] == 'add':
        if len(argv) == 1:
            msync.help.print_playlist_help(argv)
            return
        
        if argv[1].startswith('--'):
            if len(argv) == 2:
                print 'You must specify an argument with the option'
                return

            try:
                files_added = pl.addByTag(argv[1].lstrip('-'), argv[2])
                pl.write()
            except msync.InvalidOption:
                print '%s is an invalid option' % argv[1]
                return
            except msync.FileNotInLibrary:
                print 'File does not exist in the library'
                return
        else:
            try:
                files_added = pl.addByFilename(argv[1])
                pl.write()
            except msync.FileNotInLibrary:
                print 'File does not exist in the library'
                return
        print '%d files added to the playlist' % files_added
    elif argv[0] in ('copy', 'cp'):
        if len(argv) == 1:
            msync.help.print_playlist_help(argv)
            return
        files_copied = pl.copy(argv[1])
        print '%d files were copied to %s' % (files_copied, argv[1])
    elif argv[0] in ('delete', 'del', 'remove', 'rm'):
        length = len(argv)
        if length == 1:
            if pl.delete():
                print 'Playlist has been deleted'
                return
        elif argv[1].startswith('--'):
            if length == 2:
                print 'You must specificy an argument with the option'
                return

            if argv[1] == '--position':
                pl.removeByPosition(int(argv[2]) - 1)
                files_removed = 1
            else:
                try:
                    files_removed = pl.removeByTag(argv[1].lstrip('-'), argv[2])
                except msync.InvalidOption:
                    print '%s is an invalid option' % argv[1]
                    return
            pl.write()
        else:
            files_removed = pl.removeByFilename(argv[1])
            pl.write()
        print '%d files removed from the playlist' % files_removed
    elif argv[0] in ('export', 'ex'):
        length = len(argv)
        if length == 1:
            msync.help.print_playlist_help(argv)
            return

        pl.export(argv[1])
    elif argv[0] == 'info':
        length = '%d:%02d' % (pl.length / 60, pl.length % 60)
        size = '%.2f MB' % (float(pl.size) / 1048576)
        print '''Name:   %s
Length: %s
Size:   %s
Tracks: %d''' % (pl.name, length, size, len(pl.playlist))
    elif argv[0] in ('list', 'ls'):
        i = 1
        file_list = pl.list()
        format_str = "(%%%dd) %%s" % len(str(len(file_list)))
        for file in file_list:
            print format_str % (i, file.filename)
            i += 1
    elif argv[0] in ('move', 'mv', 'rename', 'ren'):
        if len(argv) == 1:
            msync.help.print_playlist_help(argv)
            return
        try:
            pl.updateName(argv[1])
        except msync.PlaylistExists:
            print 'Playlist %s already exists' % argv[1]
            return
    elif argv[0] in ('position', 'pos'):
        if len(argv) < 3:
            msync.help.print_playlist_help(argv)
            return
        try:
            pl.changePosition(int(argv[1]) - 1, int(argv[2]) - 1)
        except msync.InvalidPosition:
            print 'One of these values was an invalid position in the list.'
    elif argv[0] == 'sync':
        length = len(argv)
        if length == 1:
            msync.help.print_playlist_help(argv)
            return

        image = parse_sync_options(argv[2:])

        added, removed, updated = pl.sync(argv[1], image_opts=image)
        print '%d files were added to the library' % added
        print '%d files were removed from the library' % removed
        print '%d files were updated' % updated
    else:
        msync.help.print_help(['pl'])
    
def parse_sync_options(argv):
    parser = optparse.OptionParser()
    parser.add_option('-s', '--size', action='store', type='int', dest='size', default=None)
    parser.add_option('-f', '--format', action='store', dest='fmt', default=None)

    try:
        options, args = parser.parse_args(argv)
        images = {}
        if options.size is not None:
            images['size'] = options.size
        if options.fmt is not None:
            images['format'] = options.fmt
    except Exception, strerr:
        print 'Error parsing arguments: %s' % strerr
        return {}
    
    return images

def synclist(argv):
    slist = msync.SyncList(msync.MsyncDB(library_file()))
    length = len(argv)

    if argv is None or length == 0:
        msync.help.print_help(['synclist'])
    elif argv[0] in ('delete', 'del', 'remove', 'rm'):
        if length == 1:
            msync.help.print_synclist_help(argv)
            return
        if slist.delete(argv[1]) is True:
            print 'SyncList \'%s\' has been deleted' % argv[1]
    elif argv[0] == 'info':
        (count, size, length) = slist.info()
        length = '%d:%02d' % (length / 60, length % 60)
        size = '%.2f MB' % (float(size) / 1048576)
        print '''Length: %s
Size:   %s
Tracks: %d''' % (length, size, count)
    elif argv[0] in ('list', 'ls'):
        if length == 2 and argv[1] == 'all':
            all = True
        else:
            all = False
        
        print_sl_list(slist, all)
    elif argv[0] == 'load':
        if length == 1:
            msync.help.print_synclist_help(argv)
            return
        if slist.load(argv[1]) is True:
            print 'Loaded SyncList \'%s\'' % argv[1]
    elif argv[0] == 'mark':
        if length == 1:
            msync.help.print_synclist_help(argv)
            return
        slist.mark(argv[1])
        print_sl_list(slist, True)
    elif argv[0] == 'save':
        if length == 1:
            msync.help.print_synclist_help(argv)
            return
        if slist.save(argv[1]) is True:
            print 'SyncList saved as \'%s\'' % argv[1]
    elif argv[0] == 'show':
        print 'Available SyncLists:'
        for row in slist.show():
            print '    %s' % row
    elif argv[0] == 'sync':
        if length == 1:
            msync.help.print_synclist_help(argv)
            return
        image = parse_sync_options(argv[2:])
        slist.sync(argv[1], image_opts=image)
    elif argv[0] == 'unmark':
        if length == 1:
            msync.help.print_synclist_help(argv)
            return
        slist.unmark(argv[1])
        print_sl_list(slist, True)
    else:
        msync.help.print_help(['sl'])

def print_sl_list(slist, all):
    for plist in slist.list(all):
        if all is False:
            print plist.name
        else:
            if bool(plist.marked) is True:
                ch = 'X'
            else:
                ch = ' '

            print '(%s) %s' % (ch, plist.name)

if __name__ == '__main__':
    main(sys.argv)