mirror of
https://github.com/saltstack/salt.git
synced 2025-04-10 06:41:40 +00:00

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.
103 lines
3.3 KiB
Python
Executable file
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,
|
|
)
|