示例#1
0
def main(argv=None):
    args = None
    cmd = None
    try:
        args = parse_args(argv)

        # Init loglevel early in case we'll run
        # into errors before setting it properly
        CmdBase._set_loglevel(args)

        cmd = args.func(args)

        ret = cmd.run_cmd()
    except KeyboardInterrupt as ex:
        logger.error("Interrupted by the user", ex)
        ret = 252
    except NotDvcProjectError as ex:
        logger.error("", ex)
        ret = 253
    except DvcParserError:
        ret = 254
    except Exception as ex:
        logger.error('Unexpected error', ex)
        ret = 255

    Analytics().send_cmd(cmd, args, ret)

    return ret
示例#2
0
    def setUp(self):
        super(TestLoggerQuiet, self).setUp()

        class A(object):
            quiet = True
            verbose = False

        CmdBase._set_loglevel(A())
示例#3
0
    def test_cli(self):
        class A(object):
            quiet = True
            verbose = False

        args = A()
        CmdBase.set_loglevel(args)
        self.assertEqual(logger.logger.getEffectiveLevel(), logging.CRITICAL)

        args.quiet = False
        args.verbose = True
        CmdBase.set_loglevel(args)
        self.assertEqual(logger.logger.getEffectiveLevel(), logging.DEBUG)
示例#4
0
文件: test_cli.py 项目: summon-ml/dvc
    def test(self):
        os.chdir("..")

        class A(object):
            quiet = False
            verbose = True

        args = A()
        with self.assertRaises(DvcException):
            CmdBase(args)
示例#5
0
文件: main.py 项目: wellic/dvc
def main(argv=None):
    Logger.init()

    args = parse_args(argv)

    # Init loglevel early in case we'll run
    # into errors before setting it properly
    CmdBase._set_loglevel(args)

    try:
        cmd = args.func(args)
    except Exception as ex:
        Logger.error('Initialization error', ex)
        return 255

    try:
        ret = cmd.run_cmd()
    except Exception as ex:
        Logger.error('Unexpected error', ex)
        return 254

    return ret
示例#6
0
def main(argv=None):
    """Run dvc CLI command.

    Args:
        argv: optional list of arguments to parse. sys.argv is used by default.

    Returns:
        int: command's return code.
    """
    args = None
    cmd = None
    try:
        args = parse_args(argv)

        # Init loglevel early in case we'll run
        # into errors before setting it properly
        CmdBase.set_loglevel(args)

        cmd = args.func(args)

        ret = cmd.run_cmd()
    except KeyboardInterrupt:
        logger.error("interrupted by the user")
        ret = 252
    except NotDvcProjectError:
        logger.error()
        ret = 253
    except DvcParserError:
        ret = 254
    except Exception:  # pylint: disable=broad-except
        logger.error("unexpected error")
        ret = 255

    Analytics().send_cmd(cmd, args, ret)

    return ret