mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-13 23:10:15 +01:00
Ruslan Kuprieiev
f4349aa09b
This is a hidden flag (no docs needed) for a handy shortcut to pdb when we need it. E.g. binaries built by pyinstaller are much easier to debug with it.
141 lines
3.5 KiB
Python
141 lines
3.5 KiB
Python
"""Main entry point for dvc CLI."""
|
|
|
|
import errno
|
|
import logging
|
|
from contextlib import contextmanager
|
|
|
|
from dvc import analytics
|
|
from dvc.cli import parse_args
|
|
from dvc.config import ConfigError
|
|
from dvc.exceptions import DvcException, DvcParserError, NotDvcRepoError
|
|
from dvc.external_repo import clean_repos
|
|
from dvc.logger import FOOTER, disable_other_loggers
|
|
from dvc.tree.pool import close_pools
|
|
from dvc.utils import error_link
|
|
|
|
# Workaround for CPython bug. See [1] and [2] for more info.
|
|
# [1] https://github.com/aws/aws-cli/blob/1.16.277/awscli/clidriver.py#L55
|
|
# [2] https://bugs.python.org/issue29288
|
|
|
|
"".encode("idna")
|
|
|
|
logger = logging.getLogger("dvc")
|
|
|
|
|
|
@contextmanager
|
|
def profile(enable, dump):
|
|
if not enable:
|
|
yield
|
|
return
|
|
|
|
import cProfile
|
|
|
|
prof = cProfile.Profile()
|
|
prof.enable()
|
|
|
|
yield
|
|
|
|
prof.disable()
|
|
if not dump:
|
|
prof.print_stats(sort="cumtime")
|
|
return
|
|
prof.dump_stats(dump)
|
|
|
|
|
|
@contextmanager
|
|
def debug(enable):
|
|
try:
|
|
yield
|
|
return
|
|
except Exception:
|
|
if enable:
|
|
import pdb # noqa: T100
|
|
|
|
pdb.post_mortem()
|
|
raise
|
|
|
|
|
|
def main(argv=None): # noqa: C901
|
|
"""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
|
|
disable_other_loggers()
|
|
|
|
outerLogLevel = logger.level
|
|
try:
|
|
args = parse_args(argv)
|
|
|
|
level = None
|
|
if args.quiet:
|
|
level = logging.CRITICAL
|
|
elif args.verbose == 1:
|
|
level = logging.DEBUG
|
|
elif args.verbose > 1:
|
|
level = logging.TRACE
|
|
|
|
if level is not None:
|
|
logger.setLevel(level)
|
|
|
|
logger.trace(args)
|
|
|
|
with profile(enable=args.cprofile, dump=args.cprofile_dump):
|
|
with debug(args.pdb):
|
|
cmd = args.func(args)
|
|
ret = cmd.run()
|
|
except ConfigError:
|
|
logger.exception("configuration error")
|
|
ret = 251
|
|
except KeyboardInterrupt:
|
|
logger.exception("interrupted by the user")
|
|
ret = 252
|
|
except NotDvcRepoError:
|
|
logger.exception("")
|
|
ret = 253
|
|
except DvcParserError:
|
|
ret = 254
|
|
except DvcException:
|
|
ret = 255
|
|
logger.exception("")
|
|
except Exception as exc: # noqa, pylint: disable=broad-except
|
|
# pylint: disable=no-member
|
|
if isinstance(exc, OSError) and exc.errno == errno.EMFILE:
|
|
logger.exception(
|
|
"too many open files, please visit "
|
|
"{} to see how to handle this "
|
|
"problem".format(error_link("many-files")),
|
|
extra={"tb_only": True},
|
|
)
|
|
else:
|
|
from dvc.info import get_dvc_info
|
|
|
|
logger.exception("unexpected error")
|
|
|
|
dvc_info = get_dvc_info()
|
|
logger.debug("Version info for developers:\n%s", dvc_info)
|
|
|
|
logger.info(FOOTER)
|
|
ret = 255
|
|
|
|
try:
|
|
if analytics.is_enabled():
|
|
analytics.collect_and_send_report(args, ret)
|
|
|
|
return ret
|
|
finally:
|
|
logger.setLevel(outerLogLevel)
|
|
|
|
# Closing pools by-hand to prevent weird messages when closing SSH
|
|
# connections. See https://github.com/iterative/dvc/issues/3248 for
|
|
# more info.
|
|
close_pools()
|
|
|
|
# Remove cached repos in the end of the call, these are anonymous
|
|
# so won't be reused by any other subsequent run anyway.
|
|
clean_repos()
|