o

�N�gE�@szdZd
dl
Z
d
dlmZ
d
dlmZ
e
�e�
Zzd
dl	Z	dZ
Wney*


dZ
Yn
wdd�ZGd	d
�d
e�Z
dd�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
Cstsd
SdS)NFT)
�HAS_POSTGRES�rr�H/opt/saltstack/salt/lib/python3.10/site-packages/salt/pillar/postgres.py�__virtual__/s


rcs@eZ
dZd
Zedd��
Zdd�Zedd��
Z�f
dd	�Z	�Z
S)
�POSTGRESExtPillarzL
    This class receives and processes the database rows from POSTGRES.
    c


Csd
S)NZPOSTGRESr)
�clsrrr�_db_name:szPOSTGRESExtPillar._db_namec
Cs\d
ddddd�}
i}t�
di�}|
D]}||v
r%t�d|�
|
|||<q||||<q|S)zC
        Returns options used for the POSTGRES connection.
        Z	localhost�salti8)�host�user�pass�db�portZpostgreszUsing default for POSTGRES %s)Z__opts__�get�log�debug)�self�defaults�_optionsZ_opts�attrrrr�_get_options>s



�







zPOSTGRESExtPillar._get_optionsc

cs��|��}
t
j|
d
|
d|
d|
d|
dd�}|��}z3z
|V
t�d�

Wnt
jyB
}
z
t�d|j�
WYd	}~nd	}~w
wW|�	�
d	SW|�	�
d	S|�	�
w)
z)
        Yield a POSTGRES cursor
        rr
rrr)rr
�passwordZdbnamerzConnected to POSTGRES DBz Error in ext_pillar POSTGRES: %sN)
r�psycopg2�connect�cursorrrZ
DatabaseError�	exception�args�close)rr�connr�errrrr�_get_cursorSs*�





�




����zPOSTGRESExtPillar._get_cursorcst��
|
|�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)rr�kwargs�
�	__class__rrr$isz!POSTGRESExtPillar.extract_queries)�__name__�
__module__�__qualname__�__doc__�classmethodr
rrr"r$�
__classcell__rrr&rr5s




rcOst�j
||
g|�
Ri|�
�
S)
zF
    Execute queries against POSTGRES, merge and return as a dict
    )rZfetch)Z	minion_idZpillarrr%rrr�
ext_pillarqsr.)r+�logging�
contextlibrZsalt.pillar.sql_baser�	getLoggerr(rrr�ImportErrorrrr.rrrr�<module>
s




�<