File manager - Edit - /opt/saltstack/salt/lib/python3.10/site-packages/salt/pillar/__pycache__/postgres.cpython-310.pyc
Back
o ;jE � @ sz d Z ddlZddlmZ ddlmZ e�e�Zzddl Z dZ W n e y* dZ Y nw dd� Z G d d � d e�Z d d � ZdS ) a1 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] � N)�contextmanager)�SqlBaseExtPillarTFc C s t sdS dS )NFT)� HAS_POSTGRES� r r �H/opt/saltstack/salt/lib/python3.10/site-packages/salt/pillar/postgres.py� __virtual__/ s r c s@ e Zd ZdZedd� �Zdd� Zedd� �Z� fdd �Z � Z S ) �POSTGRESExtPillarzL This class receives and processes the database rows from POSTGRES. c C s dS )NZPOSTGRESr )�clsr r r �_db_name: s zPOSTGRESExtPillar._db_namec C s\ dddddd�}i }t �di �}|D ]}||vr%t�d|� || ||< q|| ||< q|S )zC Returns options used for the POSTGRES connection. Z localhost�salti8 )�host�user�pass�db�portZpostgresz Using default for POSTGRES %s)Z__opts__�get�log�debug)�self�defaults�_optionsZ_opts�attrr r r � _get_options> s � z POSTGRESExtPillar._get_optionsc c s� � | � � }tj|d |d |d |d |d d�}|�� }z3z |V t�d� W n tjyB } z t�d|j� W Y d }~n d }~ww W |� � d S W |� � d S |� � w ) z) Yield a POSTGRES cursor r r r r r )r r �passwordZdbnamer zConnected to POSTGRES DBz Error in ext_pillar POSTGRES: %sN) r �psycopg2�connect�cursorr r Z DatabaseError� exception�args�close)r r �connr �errr r r � _get_cursorS s* ����� �z POSTGRESExtPillar._get_cursorc s t � �||�S )z� This function normalizes the config block into a set of queries we can use. The return is a list of consistently laid out dicts. )�super�extract_queries)r r �kwargs�� __class__r r r$ i s z!POSTGRESExtPillar.extract_queries) �__name__� __module__� __qualname__�__doc__� classmethodr r r r" r$ � __classcell__r r r& r r 5 s r c O s t � j| |g|�R i |��S )zF Execute queries against POSTGRES, merge and return as a dict )r Zfetch)Z minion_idZpillarr r% r r r � ext_pillarq s r. )r+ �logging� contextlibr Zsalt.pillar.sql_baser � getLoggerr( r r r � ImportErrorr r r. r r r r �<module> s � <
| ver. 1.4 |
Github
|
.
| PHP 8.2.30 | Generation time: 0 |
proxy
|
phpinfo
|
Settings