File Manager

003 File Manager

Current Path: /usr/local/lib/python3.8/site-packages/salt/pillar

usr / local / lib / python3.8 / site-packages / salt / pillar /

📁 ..
📄 __init__.py(48.72 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.12 KB)
📄 etcd_pillar.py(2.36 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.19 KB)
📄 hiera.py(846 B)
📄 http_json.py(2.43 KB)
📄 http_yaml.py(2.43 KB)
📄 libvirt.py(5.78 KB)
📄 makostack.py(21.99 KB)
📄 mongo.py(5.6 KB)
📄 mysql.py(3.76 KB)
📄 nacl.py(744 B)
📄 netbox.py(29.5 KB)
📄 neutron.py(2.41 KB)
📄 nodegroups.py(1.7 KB)
📄 pepa.py(20.98 KB)
📄 pillar_ldap.py(10.95 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(15.38 KB)
📄 sqlcipher.py(3.42 KB)
📄 sqlite3.py(2.67 KB)
📄 stack.py(22.07 KB)
📄 svn_pillar.py(5.74 KB)
📄 varstack_pillar.py(1.1 KB)
📄 vault.py(4.03 KB)
📄 venafi.py(966 B)
📄 virtkey.py(586 B)
📄 vmware_pillar.py(16.83 KB)

Editing: postgres.py

"""
Retrieve Pillar data by doing a postgres query

.. versionadded:: 2017.7.0

:maturity: new
:depends: psycopg2
:platform: all

Complete Example
================

.. code-block:: yaml

postgres:
      user: 'salt'
      pass: 'super_secret_password'
      db: 'salt_db'

ext_pillar:
      - postgres:
          fromdb:
            query: 'SELECT col1,col2,col3,col4,col5,col6,col7
                      FROM some_random_table
                     WHERE minion_pattern LIKE %s'
            depth: 5
            as_list: True
            with_lists: [1,3]
"""

import logging
from contextlib import contextmanager

from salt.pillar.sql_base import SqlBaseExtPillar

# Set up logging
log = logging.getLogger(__name__)

try:
    import psycopg2

HAS_POSTGRES = True
except ImportError:
    HAS_POSTGRES = False

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

class POSTGRESExtPillar(SqlBaseExtPillar):
    """
    This class receives and processes the database rows from POSTGRES.
    """

@classmethod
    def _db_name(cls):
        return "POSTGRES"

def _get_options(self):
        """
        Returns options used for the POSTGRES connection.
        """
        defaults = {
            "host": "localhost",
            "user": "salt",
            "pass": "salt",
            "db": "salt",
            "port": 5432,
        }
        _options = {}
        _opts = __opts__.get("postgres", {})
        for attr in defaults:
            if attr not in _opts:
                log.debug("Using default for POSTGRES %s", attr)
                _options[attr] = defaults[attr]
                continue
            _options[attr] = _opts[attr]
        return _options

@contextmanager
    def _get_cursor(self):
        """
        Yield a POSTGRES cursor
        """
        _options = self._get_options()
        conn = psycopg2.connect(
            host=_options["host"],
            user=_options["user"],
            password=_options["pass"],
            dbname=_options["db"],
            port=_options["port"],
        )
        cursor = conn.cursor()
        try:
            yield cursor
            log.debug("Connected to POSTGRES DB")
        except psycopg2.DatabaseError as err:
            log.exception("Error in ext_pillar POSTGRES: %s", err.args)
        finally:
            conn.close()

def extract_queries(self, args, kwargs):  # pylint: disable=useless-super-delegation
        """
        This function normalizes the config block into a set of queries we
        can use.  The return is a list of consistently laid out dicts.
        """
        return super().extract_queries(args, kwargs)

def ext_pillar(minion_id, pillar, *args, **kwargs):
    """
    Execute queries against POSTGRES, merge and return as a dict
    """
    return POSTGRESExtPillar().fetch(minion_id, pillar, *args, **kwargs)

003 File Manager

Editing: postgres.py

Upload File

Create Folder