"""ARDJ, an artificial DJ.
This module contains the command line interface.  It first initializes the log
file, then, depending on the command line arguments, loads and executes the
appropriate submodule.
Commands are passed to the ardj binary as the first argument.  Command handlers
are functions prefixed with "cmd_", which have dashes replaced with
underscores.  For example, command "merge-votes" is handled by the
cmd_merge_votes function.
import glob
import logging
import os
import random
import sys
import time
import traceback
import ardj.log
def is_running_from_source():
    my_path = os.path.realpath(__file__)
    return my_path.startswith(os.path.expanduser("~/"))
def cmd_config(*args):
    """edit settings
    Opens the currently used config file in your preferred text editor."""
    import settings
        return settings.edit_cli(args)
    except Exception, e:
        print >> sys.stderr, e
def cmd_console(*args):
    """opens a console for jabber commands"""
    from ardj import console
def cmd_db_purge(*args):
    """deletes dead data from the database"""
    from ardj import database
def cmd_db_console(*args):
    """opens the database console"""
    from subprocess import Popen
    from ardj import database
    Popen(["sqlite3", "-header", database.Open().filename]).wait()
def cmd_db_init(*args):
    """initializes the database
    Initializes the configured database by executing a set of preconfigured SQL
    instructions.  This is non-destructive.  You should run this after you
    install or upgrade ardj."""
    import database
    return database.cli_init(args)
def cmd_dump_votes(prefix=None, *args):
    """dumps votes statistics"""
    from database import Vote
    daily = {}
    hourly = {}
    for vote in Vote.find_all():
        ts = time.gmtime(vote["ts"])
        if prefix is not None:
            if not time.strftime("%Y-%m-%d %H:%M:%S", ts).startswith(prefix):
        day = int(time.strftime("%w", ts)) or 7
        daily[day] = daily.get(day, 0) + 1
        hour = int(time.strftime("%H", ts))
        hourly[hour] = hourly.get(hour, 0) + 1
    def dump_votes(votes, prefix):
        total = float(sum(votes.values()))
        for k, v in votes.items():
            print "%s,%u,%u" % (prefix, k, int(v))
    dump_votes(daily, "D")
    dump_votes(hourly, "H")
def cmd_find_new_tracks(*args):
    """adds new songs from Last.fm
    See ardj.tracks.find_new_tracks() for details."""
    import tracks
    return tracks.find_new_tracks(args)
def cmd_find_new_files(*args):
    """looks for new files in the media folder"""
    import tracks
    count = tracks.MediaFolderScanner().run()
    print "Found %u new files." % count
    return True
def cmd_jabber(*args):
    """run the jabber bot"""
    from ardj import jabber, settings
    bot = jabber.Open(debug="--debug" in args)
    if bot is not None:
def cmd_export_total_listeners(*args):
    """prints overall listening statistics to stdout (CSV)"""
    from ardj import listeners
def cmd_export_yesterday_listeners(*args):
    """prints yesterday's listening statistics to stdout (CSV)"""
    from ardj import listeners
def cmd_map_listeners(*args):
    """updates the listeners map
    Updates the listener map.  See ardj.map.update_listeners() for details."""
    import map
    return map.update_listeners(args)
def cmd_merge_votes(*args):
    """merge votes according to jabber aliases"""
    from ardj import users
def cmd_say(*args):
    """renders text to voice using festival, then plays it"""
    import speech
    return speech.render_text_cli(args)
def cmd_scan_replaygain(*args):
    """calculate ReplayGain for tracks that don't have it"""
    from ardj import replaygain
def cmd_serve(*args):
    """start all subprograms, run the radio"""
    from ardj import monitor
    return monitor.run_cli(args)
def cmd_show_news_from_jamendo(*args):
    """show new tracks available at jamendo.com"""
    import jamendo
    return jamendo.print_new_tracks(args)
def cmd_start_scrobbler(*args):
    """starts the scrobbler process"""
    from ardj import scrobbler
    return scrobbler.run(*args)
def cmd_start_web_server(*args):
    """start the web server process"""
    import server
    return server.run_cli(args)
def cmd_tags(*args):
    """display tags from files"""
    if not args:
        print "Files not specified."
        return False
    from ardj import tags
    for fn in args:
        if os.path.exists(fn):
            print "Tags in %s" % fn
            for k, v in sorted(tags.get(arg).items(), key=lambda x: x[0]):
                print '  %s = %s' % (k, v)
def cmd_add_incoming_tracks(*args):
    """moves tracks from the incoming folder to the database"""
    from ardj import tracks
    files = tracks.find_incoming_files(delay=0, verbose=True)
    success = tracks.add_incoming_files(files)
    print "Added %u files to the database." % len(success)
    return True
def cmd_print_current_track(*args):
    """prints the currently playing track"""
    from ardj import tracks
    import json
    print json.dumps(tracks.Track.get_last())
def cmd_print_next_track(*args):
    """names a file to play next"""
        from ardj import tracks, database, settings
        track = tracks.get_next_track()
        if track is not None:
            print track["filepath"].encode("utf-8")
    except Exception, e:
        print >> sys.stderr, "ERROR: %s" % e
    count = tracks.count_available()
    if not count:
        music_dir = settings.get_music_dir()
        print >> sys.stderr, "WARNING: there are NO tracks in the database.  Put some files in %s, then run 'ardj find-new-files'." % music_dir
        print >> sys.stderr, "WARNING: could not pick a track to play.  Details can be found in %s." % settings.getpath("log", "syslog")
    if is_running_from_source():
        d = os.path.dirname
        pattern = os.path.join(d(d(d(__file__))), "share/audio/*.ogg")
        pattern = "/usr/share/ardj/failure/*.ogg"
    files = glob.glob(pattern)
    if files:
        print >> sys.stderr, "WARNING: playing a pre-packaged file."
        print random.choice(files)
def cmd_twit(msg, *args):
    """sends a message to twitter"""
    from ardj import twitter
        print twitter.send_message(msg.decode("utf-8"))
    except twitter.ConfigError, e:
        print >> sys.stderr, e
        return False
def cmd_twit_replies(*args):
    """shows replies to your account"""
    from ardj import twitter
        for (name, text) in reversed(twitter.get_replies()):
            print '%s: %s' % (name, text)
    except twitter.ConfigError, e:
        print >> sys.stderr, e
        return False
def cmd_twit_replies_speak(*args):
    """render replies using festival and queue for playing"""
    from ardj import twitter
        replies = twitter.get_replies()
        if len(replies):
            nick, text = replies[0]
            print twitter.speak_message(nick, text.split(' ', 1)[1], play=True)
    except twitter.ConfigError, e:
        print >> sys.stderr, e
        return False
def cmd_update_database(*args):
    """remove tracks with no files, add new ones"""
    from ardj import database
    from ardj import tracks
def cmd_update_schedule(*args):
    """looks for events in the Last.fm database"""
    from ardj import database, tout
    tout.update_schedule(refresh="--refresh" in args)
def cmd_update_track_weights(*args):
    """shift current weights to real weights"""
    from ardj import database, tracks
def cmd_update_track_lengths(*args):
    """update track lengths from files (maintenance)"""
    from ardj import database, tracks
    ids = [int(n) for n in args if n.isdigit()]
def cmd_xmpp_send(*args):
    """send a Jabber message"""
    if len(args) < 1:
        print "Usage: ardj xmpp-send \"message text\" [recipient_jid]"
    recipient = None
    if len(args) > 1:
        recipient = args[1].decode("utf-8")
    from database import Message, commit
    Message(message=args[0].decode("utf-8"), re=recipient).put()
def cmd_dedup_tracks(*args):
    """combines duplicate tracks"""
    from ardj import tracks
    from ardj.database import commit
    count = tracks.dedup_by_filename(verbose=True)
    if count:
        print "Removed %u duplicate tracks." % count
        print "No duplicate tracks found."
def cmd_download_artist_schedule(*args):
    """queues retrieving more tracks by the specified artists"""
    from ardj.database import DownloadRequest, commit
    for arg in args[1:]:
        if DownloadRequest.find_by_artist(arg) is None:
            DownloadRequest(artist=arg, owner="console").put()
def cmd_download_artist(*args):
    """downloads songs by an artist (immediately)"""
    from ardj.tracks import find_new_tracks
def cmd_db_stats(*args):
    """shows database statistics"""
    from ardj.database import Track
    tracks = Track.find_all()
    count = len(tracks)
    length = sum([t.get("length", 0) for t in tracks])
    print "%u tracks, %.1f hours." % (count, length / 60 / 60)
def cmd_queue_flush(*args):
    """delete everything from the queue"""
    from ardj.database import Queue, commit
def cmd_fix_artist_names(*args):
    """correct names according to Last.fm"""
    from ardj.scrobbler import LastFM
    from ardj.database import Track, commit
    cli = LastFM().authorize()
    if cli is None:
        print "Last.fm authentication failed."
        return False
    names = Track.get_artist_names()
    print "Correcting %u artists." % len(names)
    for name in names:
        new_name = cli.get_corrected_name(name)
        if new_name is not None and new_name != name:
            logging.info("Correcting artist name \"%s\" to \"%s\"" % (name.encode("utf-8"), new_name.encode("utf-8")))
            Track.rename_artist(name, new_name)
def cmd_lastfm_track_tags(artist_name, track_title):
    """show track tags from last.fm
    Usage: lastfm-track-tags "artist name" "track title"
    from ardj.scrobbler import LastFM
    cli = LastFM()
    print u", ".join(cli.get_track_tags(artist_name.decode("utf-8"), track_title.decode("utf-8")))
def cmd_lastfm_find_tags(*args):
    """Adds lastfm:* tags to tracks that don't have them."""
    from ardj.tracks import add_missing_lastfm_tags
def cmd_mark_hitlist(*args):
    """marks best tracks with the \"hitlist\" tag"""
    from ardj import database
def cmd_mark_liked_by(label, *jids):
    """applies a label to tracks liked by all specified jids"""
    import database
    import tracks
    count = tracks.add_label_to_tracks_liked_by(label, jids, "console")
    print "Found %u tracks." % count
    return True
def cmd_mark_long(*args):
    """marks long tracks with the "long" label"""
    from ardj.tracks import mark_long
    print "Average length: %s, total long tracks: %u." % mark_long()
def cmd_mark_orphans(*args):
    """marks tracks that don't belong to a playlist with the "orphan" label"""
    from ardj import database
def cmd_mark_recent(*args):
    """marks 100 recently added tracks with the "recent" label"""
    from ardj import database
def cmd_play(*args):
    """sets urgent playlist for next hour"""
    from ardj import database, tracks
    tracks.set_urgent(" ".join(args).decode("utf-8"))
def cmd_help(*args):
    """shows this help screen"""
    commands = []
    for k, v in globals().items():
        if k.startswith("cmd_"):
            command = k[4:].replace("_", "-")
            commands.append((command, getattr(v, "__doc__", "").split("\n")[0].strip()))
    length = max([len(x[0]) for x in commands])
    print "Usage: ardj command\nAvailable commands:"
    fmt = '  %%-%us   %%s' % length
    for cmd, doc in sorted(commands):
        print fmt % (cmd, doc)
    return False
def find_handler(name):
    """Finds the right command handler.
    The best handler is the one with the exactly matching name.  If that one
    does not exist, all command handlers starting wigh the specified string are
    selected.  If more than one exsits, an error message is displayed."""
    commands = globals()
    cmd_name = "cmd_" + name.replace("-", "_")
    if cmd_name in commands:
        return commands[cmd_name]
    similar = []
    for k, v in commands.items():
        if k.startswith(cmd_name):
    if len(similar) == 1:
        return similar[0]
    return None
def run(args):
    """The main command line interface entry point.
    Initializes the log file then looks for the specified command handler and
    executes it.  If no command was given or no handler could be found,
    displays a help screen.
    Returns 1 (i.e., an error) if the handler returned either False or None,
    otherwise returns zero (clear)."""
    if "UPSTART_JOB" in os.environ:
    if "--zsh" in args:
        arguments = []
        for k, v in sorted(globals().items()):
            if k.startswith("cmd_"):
                name = k[4:].replace("_", "-")
                info = v.__doc__.split("\n")[0].replace("'", "\\`").replace("\"", "\\\"")
                arguments.append("%s\\:'%s'" % (name, info))
        lst = " ".join(arguments)
        print "#compdef ardj\n# generated by ardj --zsh\n_arguments \"1:Select command:((%s))\"" % lst
    # Strip global modifiers.
    while len(args) > 1 and args[1].startswith("-"):
        del args[1]
    if len(args) >= 2:
        handler = find_handler(args[1])
        if handler is not None:
                if handler(*args[2:]) in (None, True):
            except KeyboardInterrupt:
            except Exception, e:
                ardj.log.log_error('ERROR handling command %s: %s' % (args[1], e), e)
                print "ERROR: %s" % e
                print traceback.format_exc(e)