SWAMPy

✎ CARVE FLESH

EDITING: alias.py

from distutils.errors import DistutilsOptionError

from setuptools.command.setopt import edit_config, option_base, config_file

def shquote(arg):
    """Quote an argument for later parsing by shlex.split()"""
    for c in '"', "'", "\\", "#":
        if c in arg:
            return repr(arg)
    if arg.split() != [arg]:
        return repr(arg)
    return arg

class alias(option_base):
    """Define a shortcut that invokes one or more commands"""

description = "define a shortcut to invoke one or more commands"
    command_consumes_arguments = True

user_options = [
        ('remove', 'r', 'remove (unset) the alias'),
    ] + option_base.user_options

boolean_options = option_base.boolean_options + ['remove']

def initialize_options(self):
        option_base.initialize_options(self)
        self.args = None
        self.remove = None

def finalize_options(self):
        option_base.finalize_options(self)
        if self.remove and len(self.args) != 1:
            raise DistutilsOptionError(
                "Must specify exactly one argument (the alias name) when "
                "using --remove"
            )

def run(self):
        aliases = self.distribution.get_option_dict('aliases')

if not self.args:
            print("Command Aliases")
            print("---------------")
            for alias in aliases:
                print("setup.py alias", format_alias(alias, aliases))
            return

elif len(self.args) == 1:
            (alias,) = self.args
            if self.remove:
                command = None
            elif alias in aliases:
                print("setup.py alias", format_alias(alias, aliases))
                return
            else:
                print("No alias definition found for %r" % alias)
                return
        else:
            alias = self.args[0]
            command = ' '.join(map(shquote, self.args[1:]))

edit_config(self.filename, {'aliases': {alias: command}}, self.dry_run)

def format_alias(name, aliases):
    source, command = aliases[name]
    if source == config_file('global'):
        source = '--global-config '
    elif source == config_file('user'):
        source = '--user-config '
    elif source == config_file('local'):
        source = ''
    else:
        source = '--filename=%r' % source
    return source + name + ' ' + command

CANCEL

Name	Type	Size	Modified	Actions
↩ ..	DIR	—	—
📄 __init__.py	PY	396 B	2023-08-18 12:07	EDIT
📄 alias.py	PY	2.3 KB	2023-08-18 12:07	EDIT
📄 bdist_egg.py	PY	16.2 KB	2023-08-18 12:07	EDIT
📄 bdist_rpm.py	PY	1.3 KB	2023-08-18 12:07	EDIT
📄 build.py	PY	6.6 KB	2023-08-18 12:07	EDIT
📄 build_clib.py	PY	4.3 KB	2023-08-18 12:07	EDIT
📄 build_ext.py	PY	17.1 KB	2023-08-18 12:07	EDIT
📄 build_py.py	PY	14.6 KB	2023-08-18 12:07	EDIT
📄 develop.py	PY	6.6 KB	2023-08-18 12:07	EDIT
📄 dist_info.py	PY	4.1 KB	2023-08-18 12:07	EDIT
📄 easy_install.py	PY	86.8 KB	2025-05-28 17:00	EDIT
📄 editable_wheel.py	PY	32.5 KB	2023-08-18 12:07	EDIT
📄 egg_info.py	PY	26.5 KB	2025-05-28 17:00	EDIT
📄 install.py	PY	5.5 KB	2023-08-18 12:07	EDIT
📄 install_egg_info.py	PY	3 KB	2025-05-28 17:00	EDIT
📄 install_lib.py	PY	4.9 KB	2025-05-28 17:00	EDIT
📄 install_scripts.py	PY	2.3 KB	2023-08-18 12:07	EDIT
📄 launcher manifest.xml	XML	628 B	2023-08-18 12:07	EDIT
📄 register.py	PY	468 B	2023-08-18 12:07	EDIT
📄 rotate.py	PY	2 KB	2023-08-18 12:07	EDIT
📄 saveopts.py	PY	657 B	2023-08-18 12:07	EDIT
📄 sdist.py	PY	6.9 KB	2023-08-18 12:07	EDIT
📄 setopt.py	PY	4.8 KB	2023-08-18 12:07	EDIT
📄 test.py	PY	7.9 KB	2023-08-18 12:07	EDIT
📄 upload.py	PY	462 B	2023-08-18 12:07	EDIT
📄 upload_docs.py	PY	7.6 KB	2023-08-18 12:07	EDIT