Current File : //proc/thread-self/root/usr/lib/python2.7/site-packages/salt/modules/postgres.pyo
�
���^c @@ s� d Z d d l m Z m Z m Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l
Z
d d l Z y d d l Z e
Z Wn e k
r� e Z n Xd d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z m Z d d l m Z d d l m Z d d l m Z d d l m! Z! e j" e# � Z$ e
Z% d Z& d
Z' d Z( d Z) e& e' e( e) f Z* i
d
d 6d d 6d d 6d d 6d d 6d d 6d d 6d d 6d d 6d d 6d! d" 6d# d$ 6d% d& 6Z+ e, d} � Z- i d/ d+ 6d d( 6d d) 6d0 d* 6d1 d' 6d2 d- 6d d. 6Z. d3 � Z/ d4 � Z0 d d d d d d5 � Z2 d6 d d d7 d d d e d8 � Z3 d d d d d d d9 � Z4 d d d d d d d: � Z5 d d d d d; � Z6 d< � Z7 d d d d d d d= � Z8 d d d d d d e d> � Z9 d d d d d d d? � Z: d d d d d d d@ � Z; dA dB � Z< d d d d d d d d d d d d dC � Z= d d d d d d d e d dD � Z> d d d d d d dE � Z? d d d d d d dF � Z@ d d d d d d dG � ZA d d d d d d d d dH � ZB d d d d d d d d d d dI �
ZC d d d d d d dJ � ZD d d d d d d e dK � ZE d d d d d d e dL � ZF d d d d d d dM � ZG d dN dO e dP � ZH e% dQ � ZI dO dR d d d d d d d d d d d d dS � ZJ d d d d d d d d d d d d d d d dR d d dT � ZK d d d d d d d d d d d d d d d d d dU � ZL d d d d d d dR d d d d d d d d d d d dV � ZM d d d d d d d d d d d d d d d d d dW � ZN d d d d d d dX � ZO d d d d d d dY � ZP d d d d d d dZ � ZQ d d d d d d d[ � ZR d d d d d d d\ � ZS d d d d d d d] � ZT d^ � ZU d d d d d d d_ � ZV d d d d d d d d d` � ZW d d d d d d d d d da � ZX d d d d d d d d d d db �
ZY d d d d d d dc � ZZ d d d d d d d d d d d d d d d dd � Z[ d d d d d d d d d d d d d d d de � Z\ d d d d d d df � Z] d d d d d dg � Z^ d d d d d d dh � Z_ d d d d d di � Z` d d d d d dj � Za d d d d d dk � Zb d d d d d dl � Zc d d d d d dm � Zd d d d d d dn � Ze d d d d d do � Zf d d d d d dp � Zg dq � Zh dr d d d d d d ds � Zi dt � Zj du � Zk dv � Zl dr d d d d d d dw � Zm d d dr d d d d d d dx � Zn d d dr d d d d d d dy � Zo d dr d d d d d d dz � Zp d6 d d d7 d d e d d{ � Zq d| � Zr d S(~ u:
Module to provide Postgres compatibility to salt.
:configuration: In order to connect to Postgres, certain configuration is
required in /etc/salt/minion on the relevant minions. Some sample configs
might look like::
postgres.host: 'localhost'
postgres.port: '5432'
postgres.user: 'postgres' -> db user
postgres.pass: ''
postgres.maintenance_db: 'postgres'
The default for the maintenance_db is 'postgres' and in most cases it can
be left at the default setting.
This data can also be passed into pillar. Options passed into opts will
overwrite options passed into pillar
:note: This module uses MD5 hashing which may not be compliant with certain
security audits.
:note: When installing postgres from the official postgres repos, on certain
linux distributions, either the psql or the initdb binary is *not*
automatically placed on the path. Add a configuration to the location
of the postgres bin's path to the relevant minion for this module::
postgres.bins_dir: '/usr/pgsql-9.5/bin/'
i ( t absolute_importt unicode_literalst print_functionN( t CommandExecutionErrort SaltInvocationError( t LooseVersion( t six( t zip( t StringIOu EXTENSION NOT INSTALLEDu EXTENSION INSTALLEDu EXTENSION TO UPGRADEu EXTENSION TO MOVEu INSERTu au CREATEu Cu TRUNCATEu Du CONNECTu cu TRIGGERu tu SELECTu ru USAGEu Uu TEMPORARYu Tu UPDATEu wu EXECUTEu Xu
REFERENCESu xu DELETEu du GRANTu *u schemau
tablespaceu languageu sequenceu tableu groupu databaseu functionu arwdDxtu rwUu UCu CTcc C@ s_ d g } t s t SxE | D]= } t j j j | � s t | � sW t d j | � f Sq q Wt S( u�
Only load this module if the psql bin exist.
initdb bin might also be used, but its presence will be detected on runtime.
u psqlu {0} was not found( t HAS_CSVt Falset saltt utilst patht whicht _find_pg_binaryt formatt True( R t util( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt __virtual__t s
c C@ s^ t d d � } t j j j | � } | sV | rZ t j j j t j j | | � � Sn | Sd S( uc
... versionadded:: 2016.3.2
Helper function to locate various psql related binaries
u
config.optionu postgres.bins_dirN( t __salt__R R R
R t ost join( R t
pg_bin_dirt util_bin( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyR � s %c
C@ s i t d 6t d 6} | d k r� | s9 t d d � } n | sO | j d � r� d t d k rh d } q� d t d k r� d
} q� d } q� n | d k r� | } n | r� | | d <n | d k r� t d d
� } n | d k r�t j j j d t � } t j j j
| d � �� } | j t j j j
d j | sA| j d � rGd n | | rV| n d | re| n d | � � � t d | | d � i | d 6| d <Wd QXn t d | d t | �} | j d d � d k r�t j d � n | d k rt d | � rt j d � n | S( u�
Helper function to call psql, because the password requirement
makes this too much code to be repeated in each function below
u reset_system_localeu clean_envu
config.optionu
postgres.hostu /u FreeBSDu os_familyu pgsqlu OpenBSDu _postgresqlu postgresu runasu
postgres.passt textu wu {0}:{1}:*:{2}:{3}u localhostu *u
file.chownu u
PGPASSFILEu envNu cmd.run_allt python_shellu retcodei u% Error connecting to Postgresql serveru file.removeu Remove PGPASSFILE failed( R
R t NoneR t
startswitht
__grains__R R t filest mkstempt fopent writet stringutilst to_strR t gett logt errort warning(
t cmdt runast passwordt hostt portt usert kwargst
pgpassfilet fp_t ret( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt _run_psql� sD
u passwordu UTF8c C@ sf | d k rG d t d k r% d } qG d t d k r> d } qG d } n | d k r\ | } n t d � } | s} t d � � n | d j | � d
j | � d j | � d j | � g }
| d k r� |
j d
j | � � n | d k r|
j d � |
j | � n | r|
j d � n | d k r�t j j j d t
� } t j j j | d � �@ } | j t j j
j d j | � � � t d | | d � Wd QX|
j d j | � g � n t d | d t
� }
d j g |
D] } t j | � ^ q�� } t d | d t |
�} | j d d � d k r5t j d � n | d k rbt d | � rbt j d � n | S(! u(
Helper function to call initdb
u FreeBSDu os_familyu pgsqlu OpenBSDu _postgresqlu postgresu initdbu initdb executable not found.u --pgdata={0}u --username={0}u
--auth={0}u --encoding={0}u --locale={0}u -Xu --data-checksumsR u wu {0}u
file.chownu Nu --pwfile={0}R) t clean_envu u cmd.run_allR u retcodei u- Error initilizing the postgres data directoryu file.removeu Removal of PGPASSFILE failed( R R R R R t appendR R R R R R R! R" R# R t extendt dictR t pipest quoteR
R$ R% R& R' ( t namet authR- R* t encodingt localeR) t waldirt checksumst _INITDB_BINR( R/ R0 R. t ct cmdstrR1 ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt _run_initdb� sL
%+ c
C@ s� d } t d | d d | d | d | d | d | �} t | d | d | d | d | d | �} x% t j j j | d
d � D] } | SWd S(
u�
Return the version of a Postgres server.
CLI Example:
.. code-block:: bash
salt '*' postgres.version
uH SELECT setting FROM pg_catalog.pg_settings WHERE name = 'server_version'u -cu -tR+ R- R, t maintenance_dbR* R) u stdoutu
N( t _psql_cmdR2 R R t itertoolst split(
R- R+ R, RC R* R) t queryR( R1 t line( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt version s '#c C@ sO t | d | d | d | d | d | �} | r: t | � St j d � d Sd S( u�
Returns the server version properly parsed and int casted for internal use.
If the Postgres server does not respond, None will be returned.
R+ R, RC R* R) uM Attempt to parse version of Postgres server failed. Is the server responding?N( RI t
_LooseVersionR% R' R ( R- R+ R, RC R* R) t psql_version( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt _parsed_version# s
c C@ st | s t d d � } n | s2 t d d � } n | sK t d d � } n | sd t d d � } n | | | | f S( uz
Returns a tuple of (user, host, port, db) with config, pillar, or default
values assigned to missing values.
u
config.optionu
postgres.useru
postgres.hostu
postgres.portu postgres.maintenance_db( R ( R- R+ R, RC ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt _connection_defaults< s c O@ s� t | j d � | j d � | j d � | j d � � \ } } } } t d � } | d d d d g } | r | d
| g 7} n | r� | d | g 7} n | r� | d t j | � g 7} n | s� d
} n | j d | g � | j | � | S( u�
Return string with fully composed psql command.
Accepts optional keyword arguments: user, host, port and maintenance_db,
as well as any number of positional arguments to be added to the end of
the command.
u useru hostu portu maintenance_dbu psqlu
--no-alignu
--no-readlineu --no-psqlrcu
--no-passwordu
--usernameu --hostu --portu postgresu --dbname( RM R$ R R t text_typeR5 ( t argsR. R- R+ R, RC t _PSQL_BINR( ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyRD M s,
c C@ sR t d | d | d | d | | � } t | d | d | d | d | d | �} | S( NR+ R- R, RC R) R* ( RD R2 ( R( R+ R, RC R* R) R- t rcmdt cmdret( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt _psql_prepare_and_runm s 'c C@ s: g } d j | j � j d � � } | r< d j | � } n t d d d | g d | d | d | d
| d | d | �}
|
d
d k r� | St |
d � } i } x{ t j | d t j j j
d � d t j j j
d � �D]@ }
|
s� q� n | s |
} q� n | j t t
| |
� � � q� W| r6| d d !} n | S( u�
Run an SQL-Query and return the results as a list. This command
only supports SELECT statements. This limitation can be worked around
with a query like this:
WITH updated AS (UPDATE pg_authid SET rolconnlimit = 2000 WHERE
rolname = 'rolename' RETURNING rolconnlimit) SELECT * FROM updated;
query
The query string.
user
Database username, if different from config or default.
host
Database host, if different from config or default.
port
Database port, if different from the config or default.
maintenance_db
The database to run the query against.
password
User password, if different from the config or default.
runas
User to run the command as.
write
Mark query as READ WRITE transaction.
CLI Example:
.. code-block:: bash
salt '*' postgres.psql_query 'select * from pg_stat_activity'
u$ COPY ({0}) TO STDOUT WITH CSV HEADERu ;u6 START TRANSACTION READ WRITE; {0}; COMMIT TRANSACTION;u -vu datestyle=ISO,MDYu -cR) R+ R- R, RC R* u retcodei u stdoutt delimiteru ,t quotecharu "i����( R t stript rstripRS R t csvt readerR R R" R# R4 R6 R ( RG R- R+ R, RC R* R) R! R1 t csv_queryRR t csv_filet headert row( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt
psql_query} s6 ( c
C@ st i } d } t | d | d | d | d | d | d | �} x1 | D]) } | | | d <| | d j d � qC W| S( u�
Return dictionary with information about databases of a Postgres server.
CLI Example:
.. code-block:: bash
salt '*' postgres.db_list
u6 SELECT datname as "Name", pga.rolname as "Owner", pg_encoding_to_char(encoding) as "Encoding", datcollate as "Collate", datctype as "Ctype", datacl as "Access privileges", spcname as "Tablespace" FROM pg_database pgd, pg_roles pga, pg_tablespace pgts WHERE pga.oid = pgd.datdba AND pgts.oid = pgd.dattablespaceR) R+ R- R, RC R* u Name( R^ t pop(
R- R+ R, RC R* R) R1 RG t rowsR] ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt db_list� s
c
C@ s7 t d | d | d | d | d | d | � } | | k S( u�
Checks if a database exists on the Postgres server.
CLI Example:
.. code-block:: bash
salt '*' postgres.db_exists 'dbname'
R- R+ R, RC R* R) ( Ra ( R9 R- R+ R, RC R* R) t databases( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt db_exists� s u 'c C@ sM | d k r d S| | k r7 t d j | | � � � n d j d | d | � S( Nu' Unsupported character {0} in value: {1}u {quote}{value}{quote}R8 t value( R R R ( Rd R8 ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt _quote_ddl_value s c
C@ s3 d j | � }
t j j j d t | d � f d t |
d � f d | f d t | � f d t | � f d t | � f g � } g } x< | j � D]. \ } } | d k r� | | d | g 7} q� q� W| r� | j d
d � |
d j | � 7}
n t
d
|
g d | d | d | d | d | d | �} | d d
k S( u�
Adds a databases to the Postgres server.
CLI Example:
.. code-block:: bash
salt '*' postgres.db_create 'dbname'
salt '*' postgres.db_create 'dbname' template=template_postgis
u CREATE DATABASE "{0}"u
TABLESPACEu "u OWNERu TEMPLATEu ENCODINGu
LC_COLLATEu LC_CTYPEu =i u WITHu u -cR- R+ R, RC R* R) u retcodeN( R R R t odictt OrderedDictRe t itemsR t insertR RS ( R9 R- R+ R, RC R* t
tablespaceR; t
lc_collatet lc_ctypet ownert templateR) RG t with_argst with_chunkst keyRd R1 ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt db_create
s( c
C@ s� t | | f � s t S| rR | rR t | | d | d | d | d | d | �}
n� g } | rz | j d j | | � � n | r� | j d j | | � � n xD | D]< } t d | g d | d | d | d | d | d | �}
q� W|
d
d k r� t St S( u�
Change tablespace or/and owner of database.
CLI Example:
.. code-block:: bash
salt '*' postgres.db_alter dbname owner=otheruser
R- R+ R, R* R) u# ALTER DATABASE "{0}" OWNER TO "{1}"u) ALTER DATABASE "{0}" SET TABLESPACE "{1}"u -cRC u retcodei ( t anyR t owner_toR4 R RS R
(
R9 R- R+ R, RC R* Rj Rm t
owner_recurseR) R1 t queriesRG ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt db_alterF s0
c C@ sS d j | � } t d | g d | d | d | d | d | d | �} | d d
k S( u�
Removes a databases from the Postgres server.
CLI Example:
.. code-block:: bash
salt '*' postgres.db_remove 'dbname'
u DROP DATABASE "{0}"u -cR- R+ R, R) RC R* u retcodei ( R RS ( R9 R- R+ R, RC R* R) RG R1 ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt db_remover s
c
C@ sx i } d } t d | d | d | d | d | d | d | �} x1 | D]) } | | | d <| | d j d � qG W| S(
u�
Return dictionary with information about tablespaces of a Postgres server.
CLI Example:
.. code-block:: bash
salt '*' postgres.tablespace_list
.. versionadded:: 2015.8.0
u� SELECT spcname as "Name", pga.rolname as "Owner", spcacl as "ACL", spcoptions as "Opts", pg_tablespace_location(pgts.oid) as "Location" FROM pg_tablespace pgts, pg_roles pga WHERE pga.oid = pgts.spcowneru postgres.psql_queryR) R+ R- R, RC R* u Name( R R_ (
R- R+ R, RC R* R) R1 RG R` R] ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt tablespace_list� s
c
C@ s7 t d | d | d | d | d | d | � } | | k S( u�
Checks if a tablespace exists on the Postgres server.
CLI Example:
.. code-block:: bash
salt '*' postgres.tablespace_exists 'dbname'
.. versionadded:: 2015.8.0
R- R+ R, RC R* R) ( Ry ( R9 R- R+ R, RC R* R) t tablespaces( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt tablespace_exists� s c
C@ s� d }
d } | r$ d j | � }
n | ry g t j | � D] \ } }
d j | |
� ^ q: } d j d j | � � } n d j | |
| | � } t d | g d | d | d
| d | d | d
| �} | d d k S( u�
Adds a tablespace to the Postgres server.
CLI Example:
.. code-block:: bash
salt '*' postgres.tablespace_create tablespacename '/path/datadir'
.. versionadded:: 2015.8.0
u u OWNER "{0}"u {0} = {1}u WITH ( {0} )u , u. CREATE TABLESPACE "{0}" {1} LOCATION '{2}' {3}u -cR- R+ R, RC R* R) u retcodei ( R R t iteritemsR RS ( R9 t locationt optionsRm R- R+ R, RC R* R) t owner_queryt
options_queryt kt vt optionstextRG R1 ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt tablespace_create� s 4 c C@ s t | | | | g � s t Sg } | rD | j d j | | � � n | rf | j d j | | � � n | r� | j d j | | j � d | j � d � � n | r� | j d j | | � � n xX | D]P } t d | g d | d | d | d
| d | d |
�}
|
d
d k r� t Sq� Wt S( u�
Change tablespace name, owner, or options.
CLI Example:
.. code-block:: bash
salt '*' postgres.tablespace_alter tsname new_owner=otheruser
salt '*' postgres.tablespace_alter index_space new_name=fast_raid
salt '*' postgres.tablespace_alter test set_option="{'seq_page_cost': '1.1'}"
salt '*' postgres.tablespace_alter tsname reset_option=seq_page_cost
.. versionadded:: 2015.8.0
u& ALTER TABLESPACE "{0}" RENAME TO "{1}"u% ALTER TABLESPACE "{0}" OWNER TO "{1}"u& ALTER TABLESPACE "{0}" SET ({1} = {2})i u" ALTER TABLESPACE "{0}" RESET ({1})u -cR- R+ R, RC R* R) u retcode( Rs R R4 R t keyst valuesRS R
( R9 R- R+ R, RC R* t new_namet new_ownert
set_optiont reset_optionR) Rv RG R1 ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt tablespace_alter� s. '
c C@ sS d j | � } t d | g d | d | d | d | d | d | �} | d d
k S( u�
Removes a tablespace from the Postgres server.
CLI Example:
.. code-block:: bash
salt '*' postgres.tablespace_remove tsname
.. versionadded:: 2015.8.0
u DROP TABLESPACE "{0}"u -cR- R+ R, R) RC R* u retcodei ( R RS ( R9 R- R+ R, RC R* R) RG R1 ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt tablespace_remove s
c @ s i } t d | d | d | d | d | d | � } | r~ | t d � k rT d } n d } | t d
� k ru d }
q� d }
n t j d � t S� f d
� } d j d | d � d | d � | d � g � j |
| � } t | d | d | d | d | d | d | �}
d � } x� |
D]� } i } x! d% D] } | | | � | | <q-WxM d& D]E } y! t j j | | d � | | <WqQt
k
r�d' | | <qQXqQW| d | d <� r�| d | d <n t t
j | d! j d" � g � � d# | d! <| | | d$ <qW| S(( u�
Return a dict with information about users of a Postgres server.
Set return_password to True to get password hash in the result.
CLI Example:
.. code-block:: bash
salt '*' postgres.user_list
R- R+ R, RC R* R) u 9.1u pg_roles.rolreplicationu NULLu 9.5u pg_roles.rolcatupdateuB Could not retrieve Postgres version. Is Postgresql server running?c @ s � r
| Sd S( Nu ( ( t s( t return_password( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt <lambda>R s u u� SELECT pg_roles.rolname as "name",pg_roles.rolsuper as "superuser", pg_roles.rolinherit as "inherits privileges", pg_roles.rolcreaterole as "can create roles", pg_roles.rolcreatedb as "can create databases", {0} as "can update system catalogs", pg_roles.rolcanlogin as "can login", {1} as "replication", pg_roles.rolconnlimit as "connections", (SELECT array_agg(pg_roles2.rolname) FROM pg_catalog.pg_auth_members JOIN pg_catalog.pg_roles pg_roles2 ON (pg_auth_members.roleid = pg_roles2.oid) WHERE pg_auth_members.member = pg_roles.oid) as "groups",pg_roles.rolvaliduntil::timestamp(0) as "expiry time", pg_roles.rolconfig as "defaults variables" uB , COALESCE(pg_shadow.passwd, pg_authid.rolpassword) as "password" u FROM pg_roles u4 LEFT JOIN pg_authid ON pg_roles.oid = pg_authid.oid u8 LEFT JOIN pg_shadow ON pg_roles.oid = pg_shadow.usesysidc S@ s0 | | d k r t S| | d k r( t Sd Sd S( uW
Returns the boolean value of the key, instead of 't' and 'f' strings.
u tu fN( R R
R ( t rowdictRq ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt get_bools s
u superuseru inherits privilegesu can create rolesu can create databasesu can update system catalogsu can loginu replicationu connectionsu expiry timeu %Y-%m-%d %H:%M:%Su defaults variablesu passwordu groupsu {}i u name( u superuseru inherits privilegesu can create rolesu can create databasesu can update system catalogsu can loginu replicationu connections( u expiry timeN( RL RJ R% R& R
R R R^ t datetimet strptimet
ValueErrorR t listRX RY RV ( R- R+ R, RC R* R) R� R1 t vert replication_columnt rolcatupdate_columnt _xRG R` R� R] t retrowRq t date_key( ( R� s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt user_list. sd
-c C@ sm t d | d | d | d | d | d | d | � } y | j | d � SWn t k
rh t j d � d SXd S(
u�
Return a dict with information about users of a Postgres server.
Set return_password to True to get password hash in the result.
CLI Example:
.. code-block:: bash
salt '*' postgres.role_get postgres
R- R+ R, RC R* R) R� u6 Could not retrieve Postgres role. Is Postgres running?N( R� R$ R t AttributeErrorR% R& ( R9 R- R+ R, RC R* R) R� t all_users( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt role_get� s
c C@ s: t t | d | d | d | d | d | d | d t �� S( u�
Checks if a user exists on the Postgres server.
CLI Example:
.. code-block:: bash
salt '*' postgres.user_exists 'username'
R- R+ R, RC R* R) R� ( t boolR� R
( R9 R- R+ R, RC R* R) ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt user_exists� s
u NOu c C@ s| | sx | d k r | } n | d k rZ | rB d j | | � } qZ d j | | | � } n | rx d j | | � } qx n | S( Nu {0} {1}u
{0} {2}{1}( R R ( t stringt testt flagt condt prefixt addtxtt skip( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt _add_role_flag� s c C@ sz | d k r t j | � } n | rv | rv | j d � rv d j t j t j j j
d j | | � � � j � � } n | S( uQ
pgsql passwords are md5 hashes of the string: 'md5{password}{rolename}'
u md5u md5{0}u {0}{1}N( R R RN R R t hashlibt md5R R R" t to_bytest hexdigest( t roleR* t encrypted( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt _maybe_encrypt_password� s 6u rolec C@ sc | d k r$ | d k r$ t } q$ n | d k r] | d k rE t } n | d k r] t } q] n | d k rr t } n t } d } d } | d k r� t | t j � r� t | � p� t | t � s� t } n t | t j � rt | � rd j t | | j
d d � d | �� } n t |
t j � rMt |
� rMd j |
j
d d � � } n t } t | � r�t | � t | d � k r�t } n i d d
6| d 6i d d
6| d 6i d
d
6| d 6i d d
6| d 6| d 6i d d
6| d 6i d d
6| d 6i d d
6t | � d 6t j | � d 6| d k d 6i d d
6| d k oDt | � d 6| pZt | t � d 6| d 6d d 6i d d
6t | � d 6| d 6| d 6i d d
6t |
� d 6|
d k d 6| d 6f
} x | D] } t | | � } q�W| j
d � r| j
d d � } n |
r_t |
t � r-d j |
� }
n x/ |
j d � D] } d j | | | � } q=Wn | S( Nu useru groupu u '{0}'u 'u ''R� u superuseru INHERITu flagu testu CREATEDBu
CREATEROLEu SUPERUSERu skipu REPLICATIONu LOGINu CONNECTION LIMITu addtxtu ENCRYPTEDu condu UNu prefixu PASSWORDu VALID UNTILu WITHu WITHu ,u {0}; GRANT "{1}" TO "{2}"( u useru group( R R R
t _DEFAULT_PASSWORDS_ENCRYPTIONt
isinstanceR t string_typesR� R R� t replaceRN R� t endswithR� R RF ( R9 t sub_cmdt typ_R� t logint connlimitt inheritt createdbt createrolest superusert groupst replicationt rolepasswordt valid_untilt db_rolet skip_passwdt escaped_passwordt escaped_valid_untilt skip_superusert flagst datat group( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt _role_cmd_args� sz (
c C@ s� t | | | | | d | d | �rA t j d | j � | � t Sd j | � } d j | t | d | d | d |
d | d
| d | d | d
| d | d |
d | d | � � } t d | g d | d | d | d | d | d | �} | d d k S( u|
Creates a Postgres role. Users and Groups are both roles in postgres.
However, users can login, groups cannot.
R* R) u %s '%s' already existsu CREATE ROLE "{0}" WITHu {0} {1}R� R� R� R� R� R� R� R� R� R� R� R� u -cR+ R- R, RC u retcodei ( R� R% t infot
capitalizeR
R R� RS ( R9 R- R+ R, RC R* R� R� R� R� R� R� R� R� R� R� R� R� R) R� R1 ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt _role_createC s0 c & C@ sv t | d d d | d | d | d | d | d | d | d
| d | d |
d
| d | d |
d | d | d | d | �S( u
Creates a Postgres user.
CLI Examples:
.. code-block:: bash
salt '*' postgres.user_create 'username' user='user' \
host='hostname' port='port' password='password' \
rolepassword='rolepassword' valid_until='valid_until'
R� u userR- R+ R, RC R* R� R� R� R� R� R� R� R� R� R� R� R) ( R� ( t usernameR- R+ R, RC R* R� R� R� R� R� R� R� R� R� R� R� R) ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt user_createy s& c C@ s t | d | d | d | d | d | d | d t �} t | � s_ t j d | j � | � t Sd j | � } d
j | t | d | d |
d
| d | d | d | d |
d | d | d | d | d | � � } t d | g d | d | d | d | d | d | �} | d d k S( u"
Updates a postgres role.
R- R+ R, RC R* R) R� u %s '%s' could not be foundu ALTER ROLE "{0}" WITHu {0} {1}R� R� R� R� R� R� R� R� R� R� R� R� u -cu retcodei ( R� R
R� R% R� R� R R� RS ( R9 R- R+ R, RC R* R� R� R� R� R� R� R� R� R� R� R� R� R) R� R� R1 ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt _role_update� s@ c & C@ sv t | d | d | d | d | d | d d d |
d | d
| d | d | d
| d | d |
d | d | d | d | �S( u
Updates a Postgres user.
CLI Examples:
.. code-block:: bash
salt '*' postgres.user_update 'username' user='user' \
host='hostname' port='port' password='password' \
rolepassword='rolepassword' valid_until='valid_until'
R- R+ R, RC R* R� u userR� R� R� R� R� R� R� R� R� R� R� R) ( R� ( R� R- R+ R, RC R* R� R� R� R� R� R� R� R� R� R� R� R) ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt user_update� s& c C@ s� t | | | | | d | d | �s8 t j d | � t Sd j | � } t d | g d | d | d | d | d | d | �t | | | | | d | d | �s� t St j d
| � t Sd S( u1
Removes a role from the Postgres Server
R* R) u User '%s' does not existu DROP ROLE "{0}"u -cR+ R- R, RC u Failed to delete user '%s'.N( R� R% R� R
R RS R ( R9 R- R+ R, RC R* R) R� ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt _role_remove s
c
C@ s� g } d } t | d | d | d | d | d | d | �} i } x7 | D]/ } d | k rI d | k rI | | | d <qI qI W| S(
u�
List available postgresql extensions
CLI Example:
.. code-block:: bash
salt '*' postgres.available_extensions
u( select * from pg_available_extensions();R- R+ R, RC R* R) u default_versionu name( R^ (
R- R+ R, RC R* R) t extsRG R1 R] ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt available_extensions6 s
c
C@ s� g } d } t | d | d | d | d | d | d | �} i } x7 | D]/ } d | k rI d | k rI | | | d <qI qI W| S(
u�
List installed postgresql extensions
CLI Example:
.. code-block:: bash
salt '*' postgres.installed_extensions
ug select a.*, b.nspname as schema_name from pg_extension a, pg_namespace b where a.extnamespace = b.oid;R- R+ R, RC R* R) u
extversionu extname( R^ (
R- R+ R, RC R* R) R� RG R1 R] ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt installed_extensionsU s
c
C@ s7 t d | d | d | d | d | d | � j | d � S( u�
Get info about an available postgresql extension
CLI Example:
.. code-block:: bash
salt '*' postgres.get_available_extension plpgsql
R- R+ R, RC R* R) N( R� R$ R ( R9 R- R+ R, RC R* R) ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt get_available_extensiont s c
C@ s7 t d | d | d | d | d | d | � j | d � S( u�
Get info about an installed postgresql extension
CLI Example:
.. code-block:: bash
salt '*' postgres.get_installed_extension plpgsql
R- R+ R, RC R* R) N( R� R$ R ( R9 R- R+ R, RC R* R) ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt get_installed_extension� s c
C@ s` t d | d | d | d | d | d | � } | j � g | D] } | j � ^ q= k r\ t St S( u�
Test if a specific extension is available
CLI Example:
.. code-block:: bash
salt '*' postgres.is_available_extension
R- R+ R, RC R* R) ( R� t lowerR R
( R9 R- R+ R, RC R* R) R� t a( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt is_available_extension� s c C@ s t | � t | � k S( u�
Compare versions of extensions using salt.utils.versions.LooseVersion.
Returns ``True`` if version a is lesser than b.
( RJ ( R� t b( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt _pg_is_older_ext_ver� s c C@ s: t | d | d | d | d | d | d | �} t | � S( u�
Test if a specific extension is installed
CLI Example:
.. code-block:: bash
salt '*' postgres.is_installed_extension
R- R+ R, RC R* R) ( R� R� ( R9 R- R+ R, RC R* R) t
installed_ext( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt is_installed_extension� s c C@ s� t | d | d | d | d | d | d | �} t g }
| r� t g }
| d k r t | j d | � | � r |
j t � n | d k r� | j d d � d
k r� | j d | � | k r� |
j t � q� n |
S(
u�
Get lifecycle information about an extension
CLI Example:
.. code-block:: bash
salt '*' postgres.create_metadata adminpack
R- R+ R, RC R* R) u
extversionu extrelocatableu fu tu schema_nameN( R� t _EXTENSION_NOT_INSTALLEDt _EXTENSION_INSTALLEDR R� R$ R4 t _EXTENSION_TO_UPGRADEt _EXTENSION_TO_MOVE( R9 t ext_versiont schemaR- R+ R, RC R* R) R� R1 ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt create_metadata� s* c
C@ sh | d k r t } n | d k r* t } n | d k r? t } n d g }
| r^ |
j d � n |
j | � | r� |
j d � n | r� |
j d � n |
j d � d j |
� } t | d | d | d | d
| d | d | �rt d
| g d | d | d | d | d
| d | �n t | d | d | d | d
| d | d | � } | sdt j d | � n | S( u�
Drop an installed postgresql extension
CLI Example:
.. code-block:: bash
salt '*' postgres.drop_extension 'adminpack'
u DROP EXTENSIONu IF EXISTSu CASCADEu RESTRICTu ;u R- R+ R, RC R* R) u -cu Failed to drop ext: %sN( R R R
R4 R R� RS R% R� (
R9 t if_existst restrictt cascadeR- R+ R, RC R* R) RO R( R1 ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt drop_extension sJ
c C@ s� | d k r t } n t | d | d | d | d | d | d | d | d |
�} t | k } t | d | d | d | d | d | d |
�}
|
r0| syd g } | r� | j d
� n | j d j | � � g } | r� | j d j | � � n | r| j d
j | � � n | r1| j d j | � � n | rT| j d � | j | � n | j d � d j | � j � } nw g } | r�t
| k r�| j d j | | � � n | r�t | k r�| j d j | | � � n d j | � j � } | r0t d | g d |
d | d | d | d | d | �q0n t | d | d | d | d | d | d | d | d |
�} t } x/ t
D]' } | | k ry| t k ryt } qyqyW| s�t j d | � n | S( u�
Install a postgresql extension
CLI Example:
.. code-block:: bash
salt '*' postgres.create_extension 'adminpack'
R� R� R- R+ R, RC R* R) u CREATE EXTENSIONu
IF NOT EXISTSu "{0}"u SCHEMA "{0}"u VERSION {0}u FROM {0}u WITHu ;u u' ALTER EXTENSION "{0}" SET SCHEMA "{1}";u$ ALTER EXTENSION "{0}" UPDATE TO {1};u -cu Failed to create ext: %sN( R R R� R� R� R4 R R5 R RV R� R� RS t _EXTENSION_FLAGSR� R
R% R� ( R9 t
if_not_existsR� R� t from_versionR- R+ R, RC R* R) t mtdatat installedt installableRO t sargsR( R1 t i( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt create_extensionT s�
c C@ s. t | d | d | d | d | d | d | �S( u�
Removes a user from the Postgres server.
CLI Example:
.. code-block:: bash
salt '*' postgres.user_remove 'username'
R- R+ R, RC R* R) ( R� ( R� R- R+ R, RC R* R) ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt user_remove� s c " C@ sj t | d | d d d | d | d | d | d | d | d
| d | d |
d
| d | d |
d | d | �S( u?
Creates a Postgres group. A group is postgres is similar to a user, but
cannot login.
CLI Example:
.. code-block:: bash
salt '*' postgres.group_create 'groupname' user='user' \
host='hostname' port='port' password='password' \
rolepassword='rolepassword'
R- R� u groupR+ R, RC R* R� R� R� R� R� R� R� R� R� R) ( R� ( t groupnameR- R+ R, RC R* R� R� R� R� R� R� R� R� R� R) ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt group_create� s" c " C@ sj t | d | d | d | d | d | d | d d d | d
| d |
d | d
| d | d |
d | d | �S( u�
Updates a postgres group
CLI Examples:
.. code-block:: bash
salt '*' postgres.group_update 'username' user='user' \
host='hostname' port='port' password='password' \
rolepassword='rolepassword'
R- R+ R, RC R* R� R� u groupR� R� R� R� R� R� R� R� R) ( R� ( R� R- R+ R, RC R* R� R� R� R� R� R� R� R� R� R) ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt group_update� s" c C@ s. t | d | d | d | d | d | d | �S( u�
Removes a group from the Postgres server.
CLI Example:
.. code-block:: bash
salt '*' postgres.group_remove 'groupname'
R- R+ R, RC R* R) ( R� ( R� R- R+ R, RC R* R) ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt group_remove& s c C@ s- t j � } | j d � | j d j | | � � d d d d d f } x| | D]t \ } }
t |
d
| d | d | d | d | d | �} x5 | D]- } | j | j d | d | d � d � q� WqN W| j d � | j � t j | j d � t d | j g d
| d | d | d | d | d | �}
|
S( u�
Set the owner of all schemas, functions, tables, views and sequences to
the given username.
CLI Example:
.. code-block:: bash
salt '*' postgres.owner_to 'dbname' 'username'
u begin;
u% alter database "{0}" owner to "{1}";
u" alter schema {n} owner to {owner};uF select quote_ident(schema_name) as n from information_schema.schemata;u! alter table {n} owner to {owner};u� select quote_ident(table_schema)||'.'||quote_ident(table_name) as n from information_schema.tables where table_schema not in ('pg_catalog', 'information_schema');u$ alter function {n} owner to {owner};u� select p.oid::regprocedure::text as n from pg_catalog.pg_proc p join pg_catalog.pg_namespace ns on p.pronamespace=ns.oid where ns.nspname not in ('pg_catalog', 'information_schema') and not p.proisagg;u% alter aggregate {n} owner to {owner};u� select p.oid::regprocedure::text as n from pg_catalog.pg_proc p join pg_catalog.pg_namespace ns on p.pronamespace=ns.oid where ns.nspname not in ('pg_catalog', 'information_schema') and p.proisagg;u$ alter sequence {n} owner to {owner};ul select quote_ident(sequence_schema)||'.'||quote_ident(sequence_name) as n from information_schema.sequences;R- R+ R, RC R* R) Rm t nu nu
u commit;
i� u -f( u" alter schema {n} owner to {owner};uF select quote_ident(schema_name) as n from information_schema.schemata;( u! alter table {n} owner to {owner};u� select quote_ident(table_schema)||'.'||quote_ident(table_name) as n from information_schema.tables where table_schema not in ('pg_catalog', 'information_schema');( u$ alter function {n} owner to {owner};u� select p.oid::regprocedure::text as n from pg_catalog.pg_proc p join pg_catalog.pg_namespace ns on p.pronamespace=ns.oid where ns.nspname not in ('pg_catalog', 'information_schema') and not p.proisagg;( u% alter aggregate {n} owner to {owner};u� select p.oid::regprocedure::text as n from pg_catalog.pg_proc p join pg_catalog.pg_namespace ns on p.pronamespace=ns.oid where ns.nspname not in ('pg_catalog', 'information_schema') and p.proisagg;( u$ alter sequence {n} owner to {owner};ul select quote_ident(sequence_schema)||'.'||quote_ident(sequence_name) as n from information_schema.sequences;(
t tempfilet NamedTemporaryFileR! R R^ t flushR t chmodR9 RS ( t dbnamet ownernameR- R+ R, R* R) t sqlfileRv t fmtRG R1 R] RR ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyRt ? s>
/
c
C@ s� t | | d | d | d | d | d | �rD t j d | | � t Sd j | � } | d k rt d j | | � } n t d | g d | d
| d | d | d
| d | �} | d d k S( u
Creates a Postgres schema.
CLI Example:
.. code-block:: bash
salt '*' postgres.schema_create dbname name owner='owner' \
user='user' \
db_user='user' db_password='password'
db_host='hostname' db_port='port'
R- t db_usert db_passwordt db_hostt db_portu '%s' already exists in '%s'u CREATE SCHEMA "{0}"u {0} AUTHORIZATION "{1}"u -cR* R, R+ RC R) u retcodei N( t
schema_existsR% R� R
R R RS (
R� R9 Rm R- R R R R R� R1 ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt
schema_create� s c C@ s� t | | d d d | d | d | d | �sD t j d | | � t Sd j | � } t d | g d | d
| d | d | d | d
| �t | | | d | d | d | d | �s� t St j d | � t Sd S( us
Removes a schema from the Postgres server.
CLI Example:
.. code-block:: bash
salt '*' postgres.schema_remove dbname schemaname
dbname
Database name we work on
schemaname
The schema's name we'll remove
user
System user all operations should be performed on behalf of
db_user
database username if different from config or default
db_password
user password if any password for a specified user
db_host
Database host if different from config or default
db_port
Database port if different from config or default
R- R R R R u" Schema '%s' does not exist in '%s'u DROP SCHEMA "{0}"u -cR) RC R+ R, R* u Failed to delete schema '%s'.N( R R R% R� R
R RS R ( R� R9 R- R R R R R� ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt
schema_remove� s" % c C@ s1 t t | | d | d | d | d | d | �� S( up
Checks if a schema exists on the Postgres server.
CLI Example:
.. code-block:: bash
salt '*' postgres.schema_exists dbname schemaname
dbname
Database name we query on
name
Schema name we look for
user
The system user the operation should be performed on behalf of
db_user
database username if different from config or default
db_password
user password if any password for a specified user
db_host
Database host if different from config or default
db_port
Database port if different from config or default
R- R R R R ( R� t
schema_get( R� R9 R- R R R R ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyR � s "c C@ sd t | d | d | d | d | d | �} y | j | d � SWn t k
r_ t j d � t SXd S( up
Return a dict with information about schemas in a database.
CLI Example:
.. code-block:: bash
salt '*' postgres.schema_get dbname name
dbname
Database name we query on
name
Schema name we look for
user
The system user the operation should be performed on behalf of
db_user
database username if different from config or default
db_password
user password if any password for a specified user
db_host
Database host if different from config or default
db_port
Database port if different from config or default
R- R R R R u8 Could not retrieve Postgres schema. Is Postgres running?N( t schema_listR$ R R� R% R& R
( R� R9 R- R R R R t all_schemas( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyR s !
c C@ s� i } d j d g � } t | d | d | d | d | d | d | �} xA | D]9 } i }
x d D] } | | |
| <qb W|
| | d <qO W| S(
uL
Return a dict with information about schemas in a Postgres database.
CLI Example:
.. code-block:: bash
salt '*' postgres.schema_list dbname
dbname
Database name we query on
user
The system user the operation should be performed on behalf of
db_user
database username if different from config or default
db_password
user password if any password for a specified user
db_host
Database host if different from config or default
db_port
Database port if different from config or default
u u� SELECT pg_namespace.nspname as "name",pg_namespace.nspacl as "acl", pg_roles.rolname as "owner" FROM pg_namespace LEFT JOIN pg_roles ON pg_roles.oid = pg_namespace.nspowner R) R+ R- R, RC R* u owneru aclu name( u owneru acl( R R^ ( R� R- R R R R R1 RG R` R] R� Rq ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyR C s
c
C@ sc i } d } t | d | d | d | d | d | d | �} x | D] } | d | | d <qC W| S( uI
.. versionadded:: 2016.3.0
Return a list of languages in a database.
CLI Example:
.. code-block:: bash
salt '*' postgres.language_list dbname
maintenance_db
The database to check
user
database username if different from config or default
password
user password if any password for a specified user
host
Database host if different from config or default
port
Database port if different from config or default
runas
System user all operations should be performed on behalf of
u) SELECT lanname AS "Name" FROM pg_languageR) R+ R- R, RC R* u Name( R^ (
RC R- R+ R, R* R) R1 RG R` R] ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt
language_list} s %
c C@ s4 t | d | d | d | d | d | �} | | k S( u}
.. versionadded:: 2016.3.0
Checks if language exists in a database.
CLI Example:
.. code-block:: bash
salt '*' postgres.language_exists plpgsql dbname
name
Language to check for
maintenance_db
The database to check in
user
database username if different from config or default
password
user password if any password for a specified user
host
Database host if different from config or default
port
Database port if different from config or default
runas
System user all operations should be performed on behalf of
R- R+ R, R* R) ( R ( R9 RC R- R+ R, R* R) t languages( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt language_exists� s
* c C@ sy t | | � r& t j d | | � t Sd j | � } t d | g d | d | d | d | d | d | �} | d
d k S( u�
.. versionadded:: 2016.3.0
Installs a language into a database
CLI Example:
.. code-block:: bash
salt '*' postgres.language_create plpgsql dbname
name
Language to install
maintenance_db
The database to install the language in
user
database username if different from config or default
password
user password if any password for a specified user
host
Database host if different from config or default
port
Database port if different from config or default
runas
System user all operations should be performed on behalf of
u Language %s already exists in %su CREATE LANGUAGE {0}u -cR- R+ R, RC R* R) u retcodei ( R
R% R� R
R RS ( R9 RC R- R+ R, R* R) RG R1 ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt language_create� s ( c C@ sy t | | � s& t j d | | � t Sd j | � } t d | g d | d | d | d | d | d | �} | d
d k S( u�
.. versionadded:: 2016.3.0
Removes a language from a database
CLI Example:
.. code-block:: bash
salt '*' postgres.language_remove plpgsql dbname
name
Language to remove
maintenance_db
The database to install the language in
user
database username if different from config or default
password
user password if any password for a specified user
host
Database host if different from config or default
port
Database port if different from config or default
runas
System user all operations should be performed on behalf of
u Language %s does not exist in %su DROP LANGUAGE {0}u -cR- R+ R, R) RC R* u retcodei ( R
R% R� R
R RS ( R9 RC R- R+ R, R* R) RG R1 ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt language_remove s ( c
C@ s� | d k rB d j d d d d d d d d
g � j | | � } n�| d k r� d j d d d d d d d d
g � j | | � } nM| d
k r� d j d d d d g � j | � } n| d k r� d j d d d d d d d g � j | | � } n� | d k r)d j d d d d g � j | � } n� | d k r\d j d d d d g � j | � } nu | d! k r�d j d" d# d$ d% g � j | � } nB | d& k r�d j d' d( d) d* d+ d, d- d. d/ g � j | � } n | S(0 u<
Generate the SQL required for specific object type
u tableu u SELECT relacl AS nameu FROM pg_catalog.pg_class cu JOIN pg_catalog.pg_namespace nu ON n.oid = c.relnamespaceu WHERE nspname = '{0}'u AND relname = '{1}'u AND relkind = 'r'u ORDER BY relnameu sequenceu AND relkind = 'S'u schemau SELECT nspacl AS nameu FROM pg_catalog.pg_namespaceu ORDER BY nspnameu functionu SELECT proacl AS nameu FROM pg_catalog.pg_proc pu ON n.oid = p.pronamespaceu% AND p.oid::regprocedure::text = '{1}'u ORDER BY proname, proargtypesu
tablespaceu SELECT spcacl AS nameu FROM pg_catalog.pg_tablespaceu WHERE spcname = '{0}'u ORDER BY spcnameu languageu SELECT lanacl AS nameu FROM pg_catalog.pg_languageu WHERE lanname = '{0}'u ORDER BY lannameu databaseu SELECT datacl AS nameu FROM pg_catalog.pg_databaseu WHERE datname = '{0}'u ORDER BY datnameu groupu SELECT rolname, admin_optionu! FROM pg_catalog.pg_auth_members mu JOIN pg_catalog.pg_roles ru ON m.member=r.oidu WHERE m.roleid INu (SELECT oidu FROM pg_catalog.pg_rolesu WHERE rolname='{0}')u ORDER BY rolname( R R ( R9 t object_typet prependRG ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt _make_privileges_list_queryX s� u publicc C@ s� | d k r6 d j d d d d g � j | | � } n\| d k r{ d j d d d
d d d
d d d g � j | | � } n| d k r� d j d d d
d d g � j | � } n� | d k r� d j d d d d d d d g � j | | � } n� | d k r&d j d d d
d d g � j | � } nl | d k r\d j d d d
d d! g � j | � } n6 | d" k r�d j d d# d
d$ d% g � j | � } n t | d& | d' | d( | d) | d* | d+ | �}
y |
d, d- } Wn t k
r�d. } n X| S(/ u/
Return the owner of a postgres object
u tableu u SELECT tableowner AS nameu FROM pg_tablesu WHERE schemaname = '{0}'u AND tablename = '{1}'u sequenceu SELECT rolname AS nameu FROM pg_catalog.pg_class cu JOIN pg_roles ru ON c.relowner = r.oidu JOIN pg_catalog.pg_namespace nu ON n.oid = c.relnamespaceu WHERE relkind='S'u AND nspname='{0}'u AND relname = '{1}'u schemau FROM pg_namespace nu ON n.nspowner = r.oidu WHERE nspname = '{0}'u functionu FROM pg_catalog.pg_proc pu ON n.oid = p.pronamespaceu% AND p.oid::regprocedure::text = '{1}'u ORDER BY proname, proargtypesu
tablespaceu FROM pg_tablespace tu ON t.spcowner = r.oidu WHERE spcname = '{0}'u languageu FROM pg_language lu ON l.lanowner = r.oidu WHERE lanname = '{0}'u databaseu FROM pg_database du ON d.datdba = r.oidu WHERE datname = '{0}'R) R+ R- R, RC R* i u nameN( R R R^ t
IndexErrorR ( R9 R R RC R- R+ R, R* R) RG R` R1 ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt _get_object_owner� s�
c C@ s� | d k r� g t | D] } t | ^ q } | j d � | t k r^ t d j | � � � n t | � j t | � � s� t d j | | � � � q� n | r� t d � � n d S( u*
Validate the supplied privileges
u groupu ALLu! Invalid object_type: {0} providedu1 Invalid privilege(s): {0} provided for object {1}u= The privileges option should not be set for object_type groupN( t _PRIVILEGE_TYPE_MAPt _PRIVILEGES_MAPR4 t _PRIVILEGES_OBJECTSR R t sett issubset( R t privst
privilegest permt _perms( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt _validate_privileges
s
c C@ sI | j � } | d k r d n | } t j d | j � � } | | | f S( u
Format options
u u \s?,\s?N( R� R t reRF t upper( R R t _privs( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt _mod_priv_opts
s c C@ sy i } d } xf | D]^ } | d k r@ t | t | <t | } q | d k rY t | | <q t | t | <t | } q W| S( u
Process part
u *N( R R
R R ( t permst _tmpt previousR ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt _process_priv_part%
s
c C@ sW | j � } t | | | � } | t k rB t d j | � � � n t | d | d | d | d | d | d | �}
i } x� |
D]� } | d k r"| d }
|
j d
� }
|
j d � } x� | D][ } | j d � \ } } | j d
� \ } } | d k rd } n t | � } | | | <q� Wq | d d k r;t } n t
} | | | d <q W| S( uH
.. versionadded:: 2016.3.0
Return a list of privileges for the specified object.
CLI Example:
.. code-block:: bash
salt '*' postgres.privileges_list table_name table maintenance_db=db_name
name
Name of the object for which the permissions should be returned
object_type
The object type, which can be one of the following:
- table
- sequence
- schema
- tablespace
- language
- database
- group
- function
prepend
Table and Sequence object types live under a schema so this should be
provided if the object is not under the default `public` schema
maintenance_db
The database to connect to
user
database username if different from config or default
password
user password if any password for a specified user
host
Database host if different from config or default
port
Database port if different from config or default
runas
System user all operations should be performed on behalf of
u! Invalid object_type: {0} providedR) R+ R- R, RC R* u groupu nameu {}u ,u /u =u u publicu admin_optionu tu rolname( R� R R R R R^ RV RF R& R R
( R9 R R RC R- R+ R, R* R) RG R` R1 R] t resultt partst partt
perms_partt _t rolenameR# R$ t admin_option( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt privileges_list8
s> :
c C@ s� t | | � \ } } } t | | | � | d k r� t | | d | d | d | d | d | d |
d | �}
|
d k r� | |
k r� t Sn t | | d | d | d | d | d | d |
d | �} | | k r�| d k r� | r� | | } n t } | St | } | r3t d � | D� � } | | | k } ns g | D] } t | ^ q:} d
| k r�| j � | | j
� j � k } n% t | � j t | | j
� � � } | Sn t
S( uQ
.. versionadded:: 2016.3.0
Check if a role has the specified privileges on an object
CLI Example:
.. code-block:: bash
salt '*' postgres.has_privileges user_name table_name table \
SELECT,INSERT maintenance_db=db_name
name
Name of the role whose privileges should be checked on object_type
object_name
Name of the object on which the check is to be performed
object_type
The object type, which can be one of the following:
- table
- sequence
- schema
- tablespace
- language
- database
- group
- function
privileges
Comma separated list of privileges to check, from the list below:
- INSERT
- CREATE
- TRUNCATE
- CONNECT
- TRIGGER
- SELECT
- USAGE
- TEMPORARY
- UPDATE
- EXECUTE
- REFERENCES
- DELETE
- ALL
grant_option
If grant_option is set to True, the grant option check is performed
prepend
Table and Sequence object types live under a schema so this should be
provided if the object is not under the default `public` schema
maintenance_db
The database to connect to
user
database username if different from config or default
password
user password if any password for a specified user
host
Database host if different from config or default
port
Database port if different from config or default
runas
System user all operations should be performed on behalf of
u groupR RC R- R+ R, R* R) c s@ s | ] } t | t f Vq d S( N( R R ( t .0R ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pys <genexpr> s u ALLN( R" R R R R R. R R6 R t sortR� R R R
( R9 t object_nameR R t grant_optionR RC R- R+ R, R* R) R! Rm t _privilegest retvalR R# R ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt has_privileges�
s8 T
% c C@ s' t | | � \ } } } t | | | � t | | | | d | d | d | d | d | d |
d | �r� t j d | | | � t Sd j | � }
| d k r� d j | | � } n- | d
k r� d j | � } n d j | � } | re| d k rd j | | � } q�| d k rD| j � d k rDd j |
| j � | | � } q�d j |
| j � | | � } n~ | d k r�d j | | � } n] | d k r�| j � d k r�d j |
| j � | | � } n d j |
| j � | | � } t d | g d | d | d | d | d |
d | �} | d d k S( uU
.. versionadded:: 2016.3.0
Grant privileges on a postgres object
CLI Example:
.. code-block:: bash
salt '*' postgres.privileges_grant user_name table_name table \
SELECT,UPDATE maintenance_db=db_name
name
Name of the role to which privileges should be granted
object_name
Name of the object on which the grant is to be performed
object_type
The object type, which can be one of the following:
- table
- sequence
- schema
- tablespace
- language
- database
- group
- function
privileges
Comma separated list of privileges to grant, from the list below:
- INSERT
- CREATE
- TRUNCATE
- CONNECT
- TRIGGER
- SELECT
- USAGE
- TEMPORARY
- UPDATE
- EXECUTE
- REFERENCES
- DELETE
- ALL
grant_option
If grant_option is set to True, the recipient of the privilege can
in turn grant it to others
prepend
Table and Sequence object types live under a schema so this should be
provided if the object is not under the default `public` schema
maintenance_db
The database to connect to
user
database username if different from config or default
password
user password if any password for a specified user
host
Database host if different from config or default
port
Database port if different from config or default
runas
System user all operations should be performed on behalf of
R RC R- R+ R, R* R) u9 The object: %s of type: %s already has privileges: %s setu ,u tableu sequenceu {0}."{1}"u functionu {0}u "{0}"u groupu$ GRANT {0} TO "{1}" WITH ADMIN OPTIONu ALLu> GRANT {0} ON ALL {1}S IN SCHEMA {2} TO "{3}" WITH GRANT OPTIONu/ GRANT {0} ON {1} {2} TO "{3}" WITH GRANT OPTIONu GRANT {0} TO "{1}"u, GRANT {0} ON ALL {1}S IN SCHEMA {2} TO "{3}"u GRANT {0} ON {1} {2} TO "{3}"u -cu retcodei ( u tableu sequence( u tableu sequence( u tableu sequence(
R" R R5 R% R� R
R R R RS ( R9 R1 R R R2 R RC R- R+ R, R* R) R! t _grantst on_partRG R1 ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt privileges_grant sP U
c C@ s: t | | � \ } } } t | | | � t | | | | d | d | d | d | d | d | d |
�s� t j d | | | � t Sd j | � } | d k r� d j | | � }
n | }
| d
k r� d j | | � } n d j | | j � |
| � } t d | g d | d | d | d | d | d |
�} | d d k S( u3
.. versionadded:: 2016.3.0
Revoke privileges on a postgres object
CLI Example:
.. code-block:: bash
salt '*' postgres.privileges_revoke user_name table_name table \
SELECT,UPDATE maintenance_db=db_name
name
Name of the role whose privileges should be revoked
object_name
Name of the object on which the revoke is to be performed
object_type
The object type, which can be one of the following:
- table
- sequence
- schema
- tablespace
- language
- database
- group
- function
privileges
Comma separated list of privileges to revoke, from the list below:
- INSERT
- CREATE
- TRUNCATE
- CONNECT
- TRIGGER
- SELECT
- USAGE
- TEMPORARY
- UPDATE
- EXECUTE
- REFERENCES
- DELETE
- ALL
maintenance_db
The database to connect to
user
database username if different from config or default
password
user password if any password for a specified user
host
Database host if different from config or default
port
Database port if different from config or default
runas
System user all operations should be performed on behalf of
R RC R- R+ R, R* R) u; The object: %s of type: %s does not have privileges: %s setu ,u tableu sequenceu {0}.{1}u groupu REVOKE {0} FROM {1}u REVOKE {0} ON {1} {2} FROM {3}u -cu retcodei ( u tableu sequence(
R" R R5 R% R� R
R R R RS ( R9 R1 R R R RC R- R+ R, R* R) R! R6 R7 RG R1 ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt privileges_revoke� s0 L
c
C@ s^ t | � r t j d | � t St | d | d | d | d | d | d | �} | d d k S(
u�
.. versionadded:: 2016.3.0
Initializes a postgres data directory
CLI Example:
.. code-block:: bash
salt '*' postgres.datadir_init '/var/lib/pgsql/data'
name
The name of the directory to initialize
auth
The default authentication method for local connections
password
The password to set for the postgres user
user
The database superuser name
encoding
The default encoding for new databases
locale
The default locale for new databases
waldir
The transaction log (WAL) directory (default is to keep WAL
inside the data directory)
.. versionadded:: 2019.2.0
checksums
If True, the cluster will be created with data page checksums.
.. note:: Data page checksums are supported since PostgreSQL 9.3.
.. versionadded:: 2019.2.0
runas
The system user the operation should be performed on behalf of
u %s already existsR: R- R* R; R< R) u retcodei ( t datadir_existsR% R� R
RB (
R9 R: R- R* R; R< R= R> R) R1 ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt datadir_init s 7 c C@ sL t j j | d � } t j j | d � } t j j | � oK t j j | � S( u�
.. versionadded:: 2016.3.0
Checks if postgres data directory has been initialized
CLI Example:
.. code-block:: bash
salt '*' postgres.datadir_exists '/var/lib/pgsql/data'
name
Name of the directory to check
u
PG_VERSIONu postgresql.conf( R R
R t isfile( R9 t
_version_filet _config_file( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyR: V s ( u schemau
tablespaceu languageu sequenceu tableu groupu databaseu function(s t __doc__t
__future__R R R R� t loggingR� R R R7 R� RX R R t ImportErrorR
t salt.utils.filesR t salt.utils.itertoolst salt.utils.odictt salt.utils.patht salt.utils.stringutilst salt.exceptionsR R t salt.utils.versionsR RJ t salt.extR t salt.ext.six.movesR R t getLoggert __name__R% R� R� R� R� R� R� R t frozensetR R R R R R2 RB RI RL RM RD RS R^ Ra Rc Re Rr Rw Rx Ry R{ R� R� R� R� R� R� R� R� R� R� R� R� R� R� R� R� R� R� R� R� R� R� R� R� R� R� R� R� Rt R R R R R R R
R R R R R R" R& R. R5 R8 R9 R; R: ( ( ( s9 /usr/lib/python2.7/site-packages/salt/modules/postgres.pyt <module> sJ
3 ; !
Q !
- * ! " , e
M % " - !
( 3 Q ! G"9(+: 4 - 5 3 R R ] s � f >
Mini Shell