mirror of
https://github.com/saltstack/salt.git
synced 2025-04-10 06:41:40 +00:00
134 lines
3.4 KiB
Python
134 lines
3.4 KiB
Python
#!/usr/bin/env python
|
|
'''
|
|
Welcome to the Salt repl which exposes the execution environment of a minion in
|
|
a pre-configured Python shell
|
|
|
|
__opts__, __salt__, __grains__, and __pillar__ are available.
|
|
|
|
Jinja can be tested with full access to the above structures in the usual way:
|
|
|
|
JINJA("""\\
|
|
I am {{ salt['cmd.run']('whoami') }}.
|
|
|
|
{% if otherstuff %}
|
|
Some other stuff here
|
|
{% endif %}
|
|
""", otherstuff=True)
|
|
|
|
A history file is maintained in ~/.saltsh_history.
|
|
completion behavior can be customized via the ~/.inputrc file.
|
|
|
|
'''
|
|
# pylint: disable=file-perms
|
|
|
|
|
|
import atexit
|
|
import builtins
|
|
import os
|
|
import pprint # pylint: disable=unused-import
|
|
import readline
|
|
import sys
|
|
from code import InteractiveConsole
|
|
|
|
import jinja2
|
|
|
|
import salt.client
|
|
import salt.config
|
|
import salt.loader
|
|
import salt.output
|
|
import salt.pillar
|
|
import salt.runner
|
|
|
|
# pylint: disable=unused-import
|
|
# These are imported to be available in the spawned shell
|
|
import salt.utils.yaml
|
|
|
|
# pylint: enable=unused-import
|
|
|
|
HISTFILE = "{HOME}/.saltsh_history".format(**os.environ)
|
|
|
|
|
|
def savehist():
|
|
"""
|
|
Save the history file
|
|
"""
|
|
readline.write_history_file(HISTFILE)
|
|
|
|
|
|
def get_salt_vars():
|
|
"""
|
|
Return all the Salt-usual double-under data structures for a minion
|
|
"""
|
|
# Create the Salt __opts__ variable
|
|
__opts__ = salt.config.client_config(
|
|
os.environ.get("SALT_MINION_CONFIG", "/etc/salt/minion")
|
|
)
|
|
|
|
# Populate grains if it hasn't been done already
|
|
if "grains" not in __opts__ or not __opts__["grains"]:
|
|
__opts__["grains"] = salt.loader.grains(__opts__)
|
|
|
|
# file_roots and pillar_roots should be set in the minion config
|
|
if "file_client" not in __opts__ or not __opts__["file_client"]:
|
|
__opts__["file_client"] = "local"
|
|
|
|
# ensure we have a minion id
|
|
if "id" not in __opts__ or not __opts__["id"]:
|
|
__opts__["id"] = "saltsh_mid"
|
|
|
|
# Populate template variables
|
|
__salt__ = salt.loader.minion_mods(__opts__)
|
|
__grains__ = __opts__["grains"]
|
|
|
|
if __opts__["file_client"] == "local":
|
|
__pillar__ = salt.pillar.get_pillar(
|
|
__opts__,
|
|
__grains__,
|
|
__opts__.get("id"),
|
|
__opts__.get("saltenv"),
|
|
).compile_pillar()
|
|
else:
|
|
__pillar__ = {}
|
|
|
|
# pylint: disable=invalid-name,unused-variable,possibly-unused-variable
|
|
def JINJA(x, **y):
|
|
return jinja2.Template(x).render(
|
|
grains=__grains__, salt=__salt__, opts=__opts__, pillar=__pillar__, **y
|
|
)
|
|
|
|
# pylint: enable=invalid-name,unused-variable,possibly-unused-variable
|
|
|
|
return locals()
|
|
|
|
|
|
def main():
|
|
"""
|
|
The main entry point
|
|
"""
|
|
salt_vars = get_salt_vars()
|
|
|
|
def salt_outputter(value):
|
|
"""
|
|
Use Salt's outputters to print values to the shell
|
|
"""
|
|
if value is not None:
|
|
builtins._ = value
|
|
salt.output.display_output(value, "", salt_vars["__opts__"])
|
|
|
|
sys.displayhook = salt_outputter
|
|
|
|
# Set maximum number of items that will be written to the history file
|
|
readline.set_history_length(300)
|
|
|
|
if os.path.exists(HISTFILE):
|
|
readline.read_history_file(HISTFILE)
|
|
|
|
atexit.register(savehist)
|
|
atexit.register(lambda: sys.stdout.write("Salt you later!\n"))
|
|
|
|
saltrepl = InteractiveConsole(locals=salt_vars)
|
|
saltrepl.interact(banner=__doc__)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|