golden hour

/opt/saltstack/salt/lib/python3.10/site-packages/salt/pillar

File/Folder	Size	Actions
__init__.py	50.98 KB
__pycache__	-
azureblob.py	13.88 KB
cmd_json.py	787 B
cmd_yaml.py	893 B
cmd_yamlex.py	674 B
cobbler.py	1.64 KB
confidant.py	3.29 KB
consul_pillar.py	11.61 KB
csvpillar.py	1.85 KB
digicert.py	1007 B
django_orm.py	7.74 KB
ec2_pillar.py	10.08 KB
etcd_pillar.py	4.04 KB
extra_minion_data_in_pillar.py	2.18 KB
file_tree.py	18.03 KB
foreman.py	3.66 KB
git_pillar.py	19.61 KB
gpg.py	560 B
hg_pillar.py	3.2 KB
hiera.py	846 B
http_json.py	3.37 KB
http_yaml.py	3.08 KB
libvirt.py	5.78 KB
makostack.py	21.99 KB
mongo.py	6.05 KB
mysql.py	3.76 KB
nacl.py	744 B
netbox.py	36.22 KB
neutron.py	2.41 KB
nodegroups.py	1.7 KB
pepa.py	20.99 KB
pillar_ldap.py	10.82 KB
postgres.py	2.82 KB
puppet.py	846 B
reclass_adapter.py	4 KB
redismod.py	3.28 KB
rethinkdb_pillar.py	4.71 KB
s3.py	14.4 KB
saltclass.py	1.49 KB
sql_base.py	16.51 KB
sqlcipher.py	3.42 KB
sqlite3.py	2.67 KB
stack.py	22.92 KB
svn_pillar.py	5.74 KB
varstack_pillar.py	1.1 KB
vault.py	6.54 KB
venafi.py	966 B
virtkey.py	586 B
vmware_pillar.py	16.83 KB

Edit: rethinkdb_pillar.py

"""
Provide external pillar data from RethinkDB

.. versionadded:: 2018.3.0

:depends: rethinkdb (on the salt-master)

salt master rethinkdb configuration
===================================
These variables must be configured in your master configuration file.
    * ``rethinkdb.host`` - The RethinkDB server. Defaults to ``'salt'``
    * ``rethinkdb.port`` - The port the RethinkDB server listens on.
      Defaults to ``'28015'``
    * ``rethinkdb.database`` - The database to connect to.
      Defaults to ``'salt'``
    * ``rethinkdb.username`` - The username for connecting to RethinkDB.
      Defaults to ``''``
    * ``rethinkdb.password`` - The password for connecting to RethinkDB.
      Defaults to ``''``

salt-master ext_pillar configuration
====================================

The ext_pillar function arguments are given in single line dictionary notation.

.. code-block:: yaml

ext_pillar:
    - rethinkdb: {table: ext_pillar, id_field: minion_id, field: pillar_root, pillar_key: external_pillar}

In the example above the following happens.
    * The salt-master will look for external pillars in the 'ext_pillar' table
      on the RethinkDB host
    * The minion id will be matched against the 'minion_id' field
    * Pillars will be retrieved from the nested field 'pillar_root'
    * Found pillars will be merged inside a key called 'external_pillar'

Module Documentation
====================
"""

# Import python libraries
import logging

# Import 3rd party libraries
try:
    import rethinkdb

HAS_RETHINKDB = True
except ImportError:
    HAS_RETHINKDB = False

__virtualname__ = "rethinkdb"

__opts__ = {
    "rethinkdb.host": "salt",
    "rethinkdb.port": "28015",
    "rethinkdb.database": "salt",
    "rethinkdb.username": None,
    "rethinkdb.password": None,
}

def __virtual__():
    if not HAS_RETHINKDB:
        return False
    return True

# Configure logging
log = logging.getLogger(__name__)

def ext_pillar(
    minion_id, pillar, table="pillar", id_field=None, field=None, pillar_key=None
):
    """
    Collect minion external pillars from a RethinkDB database

Arguments:

* `table`: The RethinkDB table containing external pillar information.
      Defaults to ``'pillar'``
    * `id_field`: Field in document containing the minion id.
      If blank then we assume the table index matches minion ids
    * `field`: Specific field in the document used for pillar data, if blank
      then the entire document will be used
    * `pillar_key`: The salt-master will nest found external pillars under
      this key before merging into the minion pillars. If blank, external
      pillars will be merged at top level
    """
    host = __opts__["rethinkdb.host"]
    port = __opts__["rethinkdb.port"]
    database = __opts__["rethinkdb.database"]
    username = __opts__["rethinkdb.username"]
    password = __opts__["rethinkdb.password"]

log.debug(
        "Connecting to %s:%s as user '%s' for RethinkDB ext_pillar",
        host,
        port,
        username,
    )

# Connect to the database
    conn = rethinkdb.connect(
        host=host, port=port, db=database, user=username, password=password
    )

data = None

try:

if id_field:
            log.debug(
                "ext_pillar.rethinkdb: looking up pillar. "
                "table: %s, field: %s, minion: %s",
                table,
                id_field,
                minion_id,
            )

if field:
                data = (
                    rethinkdb.table(table)
                    .filter({id_field: minion_id})
                    .pluck(field)
                    .run(conn)
                )
            else:
                data = rethinkdb.table(table).filter({id_field: minion_id}).run(conn)

else:
            log.debug(
                "ext_pillar.rethinkdb: looking up pillar. "
                "table: %s, field: id, minion: %s",
                table,
                minion_id,
            )

if field:
                data = rethinkdb.table(table).get(minion_id).pluck(field).run(conn)
            else:
                data = rethinkdb.table(table).get(minion_id).run(conn)

finally:
        if conn.is_open():
            conn.close()

if data.items:

# Return nothing if multiple documents are found for a minion
        if len(data.items) > 1:
            log.error(
                "ext_pillar.rethinkdb: ambiguous documents found for minion %s",
                minion_id,
            )
            return {}

else:
            result = data.items.pop()

if pillar_key:
            return {pillar_key: result}
        return result

else:
        # No document found in the database
        log.debug("ext_pillar.rethinkdb: no document found")
        return {}