eyeD3 / src / eyed3 / main.py

Travis Shirk 93e88df 



















Travis Shirk 37ecbf0 


Travis Shirk 5be0db9 
Travis Shirk 8da495b 




Travis Shirk 93e88df 
Travis Shirk d36ab15 

Travis Shirk 35a72b6 
Travis Shirk 93e88df 
Travis Shirk 8fc17f3 
Travis Shirk 2b26ebf 
Travis Shirk 03ee2b4 



Travis Shirk 2b26ebf 
Travis Shirk 93e88df 










Travis Shirk 1672b4d 
Travis Shirk 8da495b 
Travis Shirk a047c17 
Travis Shirk 93e88df 
Travis Shirk 37ecbf0 
Travis Shirk a047c17 



Travis Shirk 93e88df 
Travis Shirk 1672b4d 
Travis Shirk 93e88df 




Travis Shirk 5be0db9 


Travis Shirk 93e88df 






Travis Shirk a047c17 




Travis Shirk 5be0db9 
Travis Shirk 93e88df 

Travis Shirk 64fd4f9 
Travis Shirk 7a0ed8a 
Travis Shirk d36ab15 

Travis Shirk 7a0ed8a 
Travis Shirk 64fd4f9 


Travis Shirk 7a0ed8a 
Travis Shirk 37ecbf0 
Travis Shirk 7a0ed8a 
Travis Shirk 93e88df 
Travis Shirk 64fd4f9 


Travis Shirk 7a0ed8a 








Travis Shirk 93e88df 
Travis Shirk 7a0ed8a 
Travis Shirk 93e88df 
Travis Shirk 37ecbf0 
Travis Shirk 1672b4d 
Travis Shirk c5a39fc 

Travis Shirk 1672b4d 

Travis Shirk c5a39fc 

Travis Shirk 1672b4d 

Travis Shirk 93e88df 
Travis Shirk 2b26ebf 
Travis Shirk 93e88df 


Travis Shirk 8da495b 


Travis Shirk 93e88df 

















Travis Shirk 7a0ed8a 
Travis Shirk 93e88df 

Travis Shirk 7a0ed8a 
Travis Shirk 434d971 
Travis Shirk 7a0ed8a 
















Travis Shirk b901799 




Travis Shirk f7f42af 

Travis Shirk 7a0ed8a 







Travis Shirk 64fd4f9 



Travis Shirk 8da495b 


Travis Shirk 64fd4f9 
Travis Shirk 7a0ed8a 







Travis Shirk 93e88df 
Travis Shirk 434d971 
Travis Shirk 93e88df 
Travis Shirk 434d971 








Travis Shirk 64fd4f9 

Travis Shirk 434d971 
Travis Shirk 64fd4f9 

Travis Shirk 434d971 
Travis Shirk a9efc1a 
Travis Shirk 434d971 

Travis Shirk 93e88df 
Travis Shirk 7a0ed8a 
Travis Shirk 434d971 
Travis Shirk 93e88df 
Travis Shirk 64fd4f9 
Travis Shirk 93e88df 
Travis Shirk 7a0ed8a 
Travis Shirk 35a72b6 
Travis Shirk 7a0ed8a 
Travis Shirk 35a72b6 
Travis Shirk 7a0ed8a 
Travis Shirk 35a72b6 

Travis Shirk 7a0ed8a 




Travis Shirk 1672b4d 
Travis Shirk 7a0ed8a 




Travis Shirk 35a72b6 

Travis Shirk c079d6b 

Travis Shirk 35a72b6 

Travis Shirk 7a0ed8a 
Travis Shirk 93e88df 




Travis Shirk 2b26ebf 
Travis Shirk 93e88df 








Travis Shirk 2b26ebf 
Travis Shirk 93e88df 

Travis Shirk 8da495b 

Travis Shirk 93e88df 
Travis Shirk 37ecbf0 
Travis Shirk 2b26ebf 
Travis Shirk 93e88df 




Travis Shirk c079d6b 

Travis Shirk 93e88df 

Travis Shirk c079d6b 






Travis Shirk 93e88df 


# -*- coding: utf-8 -*-
################################################################################
#  Copyright (C) 2009-2012  Travis Shirk <travis@pobox.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 2 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
#
################################################################################
from __future__ import print_function
import sys
import exceptions
import os.path
import textwrap
import eyed3
import eyed3.utils
import eyed3.utils.cli
import eyed3.plugins
import eyed3.info


DEFAULT_PLUGIN = "classic"
DEFAULT_CONFIG = eyed3.info.USER_CONFIG


def main(args, config):
    if args.list_plugins:
        _listPlugins(config)
        return 0

    args.plugin.start(args, config)

    # Process paths (files/directories)
    for p in args.paths:
        eyed3.utils.walk(args.plugin, p, excludes=args.excludes,
                         fs_encoding=args.fs_encoding)

    args.plugin.handleDone()

    return 0


def _listPlugins(config):
    from eyed3.utils.cli import GREEN, GREY, boldText

    print("")

    def header(name):
        is_default = name == DEFAULT_PLUGIN
        return (boldText("* ", c=GREEN if is_default else None) +
                boldText(name, c=None))

    all_plugins = eyed3.plugins.load(reload=True, paths=_getPluginPath(config))
    # Create a new dict for sorted display
    plugin_names = []
    for plugin in set(all_plugins.values()):
        plugin_names.append(plugin.NAMES[0])

    print("Type 'eyeD3 --plugin=<name> --help' for more help")
    print("")

    plugin_names.sort()
    for name in plugin_names:
        plugin = all_plugins[name]

        alt_names = plugin.NAMES[1:]
        alt_names = " (%s)" % ", ".join(alt_names) if alt_names else ""

        print("%s %s:" % (header(name), alt_names))
        for l in textwrap.wrap(plugin.SUMMARY,
                               initial_indent=' ' * 2,
                               subsequent_indent=' ' * 2):
            print(boldText(l, c=GREY))
        print("")


def _loadConfig(args):
    import os
    import ConfigParser

    config = None
    config_file = None

    if args.config:
        config_file = os.path.abspath(config_file)
    elif args.no_config is False:
        config_file = DEFAULT_CONFIG

    if not config_file:
        return None

    if os.path.isfile(config_file):
        try:
            config = ConfigParser.SafeConfigParser()
            config.read(config_file)
        except ConfigParser.Error as ex:
            eyed3.log.warning("User config error: " + str(ex))
            return None
    elif config_file != DEFAULT_CONFIG:
        raise IOError("User config not found: %s" % config_file)

    return config


def _getPluginPath(config):
    plugin_path = [eyed3.info.USER_PLUGINS_DIR]

    if config and config.has_option("default", "plugin_path"):
        val = config.get("default", "plugin_path")
        plugin_path += [os.path.expanduser(os.path.expandvars(d)) for d
                            in val.split(':') if val]
    return plugin_path


def profileMain(args, config):  # pragma: no cover
    '''This is the main function for profiling
    http://code.google.com/appengine/kb/commontasks.html#profiling
    '''
    import cProfile
    import pstats
    import StringIO

    eyed3.log.debug("driver profileMain")
    prof = cProfile.Profile()
    prof = prof.runctx("main(args)", globals(), locals())

    stream = StringIO.StringIO()
    stats = pstats.Stats(prof, stream=stream)
    stats.sort_stats("time")  # Or cumulative
    stats.print_stats(100)  # 80 = how many to print

    # The rest is optional.
    stats.print_callees()
    stats.print_callers()
    sys.stderr.write("Profile data:\n%s\n" % stream.getvalue())

    return 0


def parseCommandLine(cmd_line_args=None):
    from eyed3.utils.cli import ArgumentParser

    def makeParser():
        p = ArgumentParser(prog=eyed3.info.NAME, add_help=True)
        p.add_argument("paths", metavar="PATH", nargs="*",
                       help="Files or directory paths")
        p.add_argument("--exclude", action="append", metavar="PATTERN",
                       dest="excludes",
                       help="A regular expression for path exclusion. May be "
                            "specified multiple times.")
        p.add_argument("-L", "--plugins", action="store_true", default=False,
                       dest="list_plugins", help="List all available plugins")
        p.add_argument("-P", "--plugin", action="store", dest="plugin",
                       default=None, metavar="NAME",
                       help="Specify which plugin to use. The default is '%s'" %
                            DEFAULT_PLUGIN)
        p.add_argument("-C", "--config", action="store", dest="config",
                       default=None, metavar="FILE",
                       help="Supply a configuration file. The default is "
                            "'%s', although even that is optional." %
                            DEFAULT_CONFIG)
        p.add_argument("--backup", action="store_true", dest="backup",
                       help="Plugins should honor this option such that "
                            "a backup is made of any file modified. The backup "
                            "is made in same directory with a '.orig' "
                            "extension added.")
        p.add_argument("-Q", "--quiet", action="store_true", dest="quiet",
                       default=False, help="A hint to plugins to output less.")
        p.add_argument("--fs-encoding", action="store",
                       dest="fs_encoding", default=eyed3.LOCAL_FS_ENCODING,
                       metavar="ENCODING",
                       help="Use the specified file system encoding for "
                            "filenames.  Default as it was detected is '%s' "
                            "but this option is still useful when reading "
                            "from mounted file systems." %
                            eyed3.LOCAL_FS_ENCODING)
        p.add_argument("--no-config", action="store_true", dest="no_config",
                       help="Do not load the default user config '%s'. "
                            "The -c/--config options are still honored if "
                            "present." % DEFAULT_CONFIG)
        p.add_argument("--no-color", action="store_true", dest="no_color",
                       help="Do not load the default user config '%s'. "
                            "Suppress color codes in console output.")

        # Debugging options
        group = p.debug_arg_group
        group.add_argument("--profile", action="store_true", default=False,
                           dest="debug_profile",
                           help="Run using python profiler.")
        group.add_argument("--pdb", action="store_true", dest="debug_pdb",
                           help="Drop into 'pdb' when errors occur.")
        return p

    cmd_line_args = list(cmd_line_args) if cmd_line_args else list(sys.argv[1:])

    # Remove any options not related to plugin/config for first parse. These
    # determine the parser for the next stage.
    stage_one_args = []
    idx, auto_append = 0, False
    while idx < len(cmd_line_args):
        opt = cmd_line_args[idx]
        if auto_append:
            stage_one_args.append(opt)
            auto_append = False

        if opt in ("-C", "--config", "-P", "--plugin", "--no-config"):
            stage_one_args.append(opt)
            if opt != "--no-config":
                auto_append = True
        elif (opt.startswith("-C=") or opt.startswith("--config=") or
                opt.startswith("-P=") or opt.startswith("--plugin=")):
            stage_one_args.append(opt)
        idx += 1

    parser = makeParser()
    args = parser.parse_args(stage_one_args)

    config = _loadConfig(args)

    if args.plugin:
        # Plugin on the command line takes precedence over config.
        plugin_name = args.plugin
    elif config and config.has_option("default", "plugin"):
        # Get default plugin from config or use DEFAULT_CONFIG
        plugin_name = config.get("default", "plugin")
        if not plugin_name:
            plugin_name = DEFAULT_PLUGIN
    else:
        plugin_name = DEFAULT_PLUGIN
    assert(plugin_name)

    PluginClass = eyed3.plugins.load(plugin_name, paths=_getPluginPath(config))
    if PluginClass is None:
        eyed3.utils.cli.printError("Plugin not found: %s" % plugin_name)
        parser.exit(1)
    plugin = PluginClass(parser)

    if config and config.has_option("default", "options"):
        cmd_line_args.extend(config.get("default", "options").split())
    if config and config.has_option(plugin_name, "options"):
        cmd_line_args.extend(config.get(plugin_name, "options").split())

    # Reparse the command line including options from the config.
    args = parser.parse_args(args=cmd_line_args)

    args.plugin = plugin
    eyed3.log.debug("command line args: %s", args)
    eyed3.log.debug("plugin is: %s", plugin)

    return args, parser, config


if __name__ == "__main__":  # pragma: no cover
    retval = 1

    # We should run against the same install
    eyed3.require(eyed3.info.VERSION)

    try:
        args, _, config = parseCommandLine()

        for fp in [sys.stdout, sys.stderr]:
            color = not args.no_color and os.isatty(fp.fileno())
            eyed3.utils.cli.enableColorOutput(fp, color)

        mainFunc = main if args.debug_profile is False else profileMain
        retval = mainFunc(args, config)
    except KeyboardInterrupt:
        retval = 0
    except IOError as ex:
        eyed3.utils.cli.printError(ex)
    except exceptions.Exception as ex:
        eyed3.utils.cli.printError("Uncaught exception: %s\n" % str(ex))
        eyed3.log.exception(ex)

        if args.debug_pdb:
            try:
                import ipdb as pdb
            except ImportError:
                import pdb

            e, m, tb = sys.exc_info()
            pdb.post_mortem(tb)
    finally:
        sys.exit(retval)

# vim: set ft=python:
Tip: Filter by directory path e.g. /media app.js to search for public/media/app.js.
Tip: Use camelCasing e.g. ProjME to search for ProjectModifiedEvent.java.
Tip: Filter by extension type e.g. /repo .js to search for all .js files in the /repo directory.
Tip: Separate your search with spaces e.g. /ssh pom.xml to search for src/ssh/pom.xml.
Tip: Use ↑ and ↓ arrow keys to navigate and return to view the file.
Tip: You can also navigate files with Ctrl+j (next) and Ctrl+k (previous) and view the file with Ctrl+o.
Tip: You can also navigate files with Alt+j (next) and Alt+k (previous) and view the file with Alt+o.