venv/lib/python3.11/site-packages/setuptools/command/alias.py

   1 from distutils.errors import DistutilsOptionError
   2
   3 from setuptools.command.setopt import edit_config, option_base, config_file
   4
   5
   6 def shquote(arg):
   7     """Quote an argument for later parsing by shlex.split()"""
   8     for c in '"', "'", "\\", "#":
   9         if c in arg:
  10             return repr(arg)
  11     if arg.split() != [arg]:
  12         return repr(arg)
  13     return arg
  14
  15
  16 class alias(option_base):
  17     """Define a shortcut that invokes one or more commands"""
  18
  19     description = "define a shortcut to invoke one or more commands"
  20     command_consumes_arguments = True
  21
  22     user_options = [
  23         ('remove', 'r', 'remove (unset) the alias'),
  24     ] + option_base.user_options
  25
  26     boolean_options = option_base.boolean_options + ['remove']
  27
  28     def initialize_options(self):
  29         option_base.initialize_options(self)
  30         self.args = None
  31         self.remove = None
  32
  33     def finalize_options(self):
  34         option_base.finalize_options(self)
  35         if self.remove and len(self.args) != 1:
  36             raise DistutilsOptionError(
  37                 "Must specify exactly one argument (the alias name) when "
  38                 "using --remove"
  39             )
  40
  41     def run(self):
  42         aliases = self.distribution.get_option_dict('aliases')
  43
  44         if not self.args:
  45             print("Command Aliases")
  46             print("---------------")
  47             for alias in aliases:
  48                 print("setup.py alias", format_alias(alias, aliases))
  49             return
  50
  51         elif len(self.args) == 1:
  52             (alias,) = self.args
  53             if self.remove:
  54                 command = None
  55             elif alias in aliases:
  56                 print("setup.py alias", format_alias(alias, aliases))
  57                 return
  58             else:
  59                 print("No alias definition found for %r" % alias)
  60                 return
  61         else:
  62             alias = self.args[0]
  63             command = ' '.join(map(shquote, self.args[1:]))
  64
  65         edit_config(self.filename, {'aliases': {alias: command}}, self.dry_run)
  66
  67
  68 def format_alias(name, aliases):
  69     source, command = aliases[name]
  70     if source == config_file('global'):
  71         source = '--global-config '
  72     elif source == config_file('user'):
  73         source = '--user-config '
  74     elif source == config_file('local'):
  75         source = ''
  76     else:
  77         source = '--filename=%r' % source
  78     return source + name + ' ' + command