You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

main_parser.py 2.9KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104
  1. """A single place for constructing and exposing the main parser
  2. """
  3. import os
  4. import sys
  5. from pip import __version__
  6. from pip._internal.cli import cmdoptions
  7. from pip._internal.cli.parser import (
  8. ConfigOptionParser, UpdatingDefaultsHelpFormatter,
  9. )
  10. from pip._internal.commands import (
  11. commands_dict, get_similar_commands, get_summaries,
  12. )
  13. from pip._internal.exceptions import CommandError
  14. from pip._internal.utils.misc import get_prog
  15. from pip._internal.utils.typing import MYPY_CHECK_RUNNING
  16. if MYPY_CHECK_RUNNING:
  17. from typing import Tuple, List # noqa: F401
  18. __all__ = ["create_main_parser", "parse_command"]
  19. def create_main_parser():
  20. # type: () -> ConfigOptionParser
  21. """Creates and returns the main parser for pip's CLI
  22. """
  23. parser_kw = {
  24. 'usage': '\n%prog <command> [options]',
  25. 'add_help_option': False,
  26. 'formatter': UpdatingDefaultsHelpFormatter(),
  27. 'name': 'global',
  28. 'prog': get_prog(),
  29. }
  30. parser = ConfigOptionParser(**parser_kw)
  31. parser.disable_interspersed_args()
  32. pip_pkg_dir = os.path.abspath(os.path.join(
  33. os.path.dirname(__file__), "..", "..",
  34. ))
  35. parser.version = 'pip %s from %s (python %s)' % (
  36. __version__, pip_pkg_dir, sys.version[:3],
  37. )
  38. # add the general options
  39. gen_opts = cmdoptions.make_option_group(cmdoptions.general_group, parser)
  40. parser.add_option_group(gen_opts)
  41. # so the help formatter knows
  42. parser.main = True # type: ignore
  43. # create command listing for description
  44. command_summaries = get_summaries()
  45. description = [''] + ['%-27s %s' % (i, j) for i, j in command_summaries]
  46. parser.description = '\n'.join(description)
  47. return parser
  48. def parse_command(args):
  49. # type: (List[str]) -> Tuple[str, List[str]]
  50. parser = create_main_parser()
  51. # Note: parser calls disable_interspersed_args(), so the result of this
  52. # call is to split the initial args into the general options before the
  53. # subcommand and everything else.
  54. # For example:
  55. # args: ['--timeout=5', 'install', '--user', 'INITools']
  56. # general_options: ['--timeout==5']
  57. # args_else: ['install', '--user', 'INITools']
  58. general_options, args_else = parser.parse_args(args)
  59. # --version
  60. if general_options.version:
  61. sys.stdout.write(parser.version) # type: ignore
  62. sys.stdout.write(os.linesep)
  63. sys.exit()
  64. # pip || pip help -> print_help()
  65. if not args_else or (args_else[0] == 'help' and len(args_else) == 1):
  66. parser.print_help()
  67. sys.exit()
  68. # the subcommand name
  69. cmd_name = args_else[0]
  70. if cmd_name not in commands_dict:
  71. guess = get_similar_commands(cmd_name)
  72. msg = ['unknown command "%s"' % cmd_name]
  73. if guess:
  74. msg.append('maybe you meant "%s"' % guess)
  75. raise CommandError(' - '.join(msg))
  76. # all the args without the subcommand
  77. cmd_args = args[:]
  78. cmd_args.remove(cmd_name)
  79. return cmd_name, cmd_args