OLD | NEW |
(Empty) | |
| 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 |
OLD | NEW |