salt/scripts/salt-extend
Erik Johnson bb03f4a0ee Update scripts/* to explicitly reference Python 3
Since the master branch no longer supports Python 2, and many distros
still default /usr/bin/python to Python 2, this commit changes the
scripts to explicitly call Python 3.
2020-05-07 16:58:40 -07:00

103 lines
3.3 KiB
Python
Executable file

#!/usr/bin/env python3
"""
Quickstart for creating an/or extending the functionality of your SaltStack installation
usage: salt-extend [-h] [--extension EXTENSION]
[--salt-directory SALT_DIRECTORY] [--name NAME]
[--description DESCRIPTION] [--no-merge]
optional arguments:
-h, --help show this help message and exit
--extension EXTENSION, -e EXTENSION
Extension type, e.g. 'module', 'state'.
--salt-directory SALT_DIRECTORY, -o SALT_DIRECTORY
Directory where your salt installation is kept
(defaults to .).
--name NAME, -n NAME Module name.
--description DESCRIPTION, -d DESCRIPTION
Short description of what the module does.
--no-merge Don't merge the module into the salt directory, keep
in a temp location
"""
import logging
import sys
from salt.scripts import salt_extend
PY26 = sys.version_info[0] == 2 and sys.version_info[1] == 6
def _parse_args_argparse():
import argparse
parser = argparse.ArgumentParser(
description="Quickly boilerplate an extension to SaltStack"
)
parser.add_argument(
"--extension", "-e", help="Extension type, e.g. 'module', 'state'."
)
parser.add_argument(
"--salt-directory",
"-o",
help="Directory where your salt installation is kept (defaults to .).",
)
parser.add_argument("--name", "-n", help="Module name.")
parser.add_argument(
"--description", "-d", help="Short description of what the module does."
)
parser.add_argument(
"--no-merge",
help="Don't merge the module into the salt directory, keep in a temp location",
action="store_true",
)
parser.add_argument(
"--debug",
help="Display detailed logs whilst applying templates",
action="store_true",
)
return parser.parse_args()
def _parse_args_optparse():
from optparse import OptionParser
parser = OptionParser(usage="Quickly boilerplate an extension to SaltStack")
parser.add_option(
"--extension", "-e", help="Extension type, e.g. 'module', 'state'."
)
parser.add_option(
"--salt-directory",
"-o",
help="Directory where your salt installation is kept (defaults to .).",
)
parser.add_option("--name", "-n", help="Module name.")
parser.add_option(
"--description", "-d", help="Short description of what the module does."
)
parser.add_option(
"--no-merge",
help="Don't merge the module into the salt directory, keep in a temp location",
action="store_true",
)
parser.add_option(
"--debug",
help="Display detailed logs whilst applying templates",
action="store_true",
)
return parser.parse_args()
if __name__ == "__main__":
if PY26:
(args, _) = _parse_args_optparse()
else:
args = _parse_args_argparse()
if args.debug:
logging.basicConfig(level=logging.DEBUG)
salt_extend(
extension=args.extension,
name=args.name,
description=args.description,
salt_dir=args.salt_directory,
merge=not args.no_merge,
)