Current File : //usr/lib/python2.7/site-packages/salt/pillar/postgres.pyo
�
���^c @@ s� d Z d d l m Z m Z m Z d d l m Z d d l Z d d l m Z e j
e � Z y d d l
Z
e Z Wn e k
r� e Z n Xd � Z d e f d � � YZ d � Z d S(
u1
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]
i ( t absolute_importt print_functiont unicode_literals( t contextmanagerN( t SqlBaseExtPillarc C@ s t s
t St S( N( t HAS_POSTGRESt Falset True( ( ( s8 /usr/lib/python2.7/site-packages/salt/pillar/postgres.pyt __virtual__3 s t POSTGRESExtPillarc B@ s> e Z d Z e d � � Z d � Z e d � � Z d � Z RS( uL
This class receives and processes the database rows from POSTGRES.
c C@ s d S( Nu POSTGRES( ( t cls( ( s8 /usr/lib/python2.7/site-packages/salt/pillar/postgres.pyt _db_name= s c C@ s� i d d 6d d 6d d 6d d 6d d 6} i } t j d i � } xL | D]D } | | k r~ t j d
| � | | | | <qH n | | | | <qH W| S( uC
Returns options used for the POSTGRES connection.
u localhostu hostu saltu useru passu dbi8 u portu postgresu Using default for POSTGRES %s( t __opts__t gett logt debug( t selft defaultst _optionst _optst attr( ( s8 /usr/lib/python2.7/site-packages/salt/pillar/postgres.pyt _get_optionsA s
c c@ s� | j � } t j d | d d | d d | d d | d d | d
� } | j � } zF y | Vt j d � Wn) t j k
r� } t j d | j � n XWd
| j � Xd
S( u)
Yield a POSTGRES cursor
t hostu hostt useru usert passwordu passt dbnameu dbt portu portu Connected to POSTGRES DBu Error in ext_pillar POSTGRES: %sN(
R t psycopg2t connectt cursorR R t
DatabaseErrort exceptiont argst close( R R t connR t err( ( s8 /usr/lib/python2.7/site-packages/salt/pillar/postgres.pyt _get_cursorT s
c C@ s t t | � j | | � S( u�
This function normalizes the config block into a set of queries we
can use. The return is a list of consistently laid out dicts.
( t superR t extract_queries( R R t kwargs( ( s8 /usr/lib/python2.7/site-packages/salt/pillar/postgres.pyR& h s ( t __name__t
__module__t __doc__t classmethodR R R R$ R& ( ( ( s8 /usr/lib/python2.7/site-packages/salt/pillar/postgres.pyR 9 s
c O@ s t � j | | | | � S( uF
Execute queries against POSTGRES, merge and return as a dict
( R t fetch( t minion_idt pillarR R' ( ( s8 /usr/lib/python2.7/site-packages/salt/pillar/postgres.pyt
ext_pillarp s ( R* t
__future__R R R t
contextlibR t loggingt salt.pillar.sql_baseR t getLoggerR( R R R R t ImportErrorR R R R/ ( ( ( s8 /usr/lib/python2.7/site-packages/salt/pillar/postgres.pyt <module> s
7
Mini Shell