Current File : //usr/lib/python2.7/site-packages/salt/modules/mssql.pyo
�
���^c @@ s� d Z d d l m Z m Z m Z d d l m Z m Z d d l j j
Z
y d d l Z e Z
Wn e k
ry e Z
n Xi d d 6d d 6d d
6d d 6d d
6e d 6Z d � Z d � Z d e f d � � YZ d � Z d � Z d � Z d � Z d d d � Z d � Z d � Z d � Z d d d � Z d � Z d d � Z d d d d d � Z d � Z! d d d! � Z" d" � Z# d d d d d d# � Z$ d$ � Z% d S(% u�
Module to provide MS SQL Server compatibility to salt.
:depends: - FreeTDS
- pymssql Python module
:configuration: In order to connect to MS SQL Server, certain configuration is
required in minion configs/pillars on the relevant minions. Some sample
pillars might look like::
mssql.server: 'localhost'
mssql.port: 1433
mssql.user: 'sysdba'
mssql.password: 'Some preferable complex password'
mssql.database: ''
The default for the port is '1433' and for the database is '' (empty string);
in most cases they can be left at the default setting.
Options that are directly passed into functions will overwrite options from
configs or pillars.
i ( t absolute_importt print_functiont unicode_literals( t JSONEncodert loadsNu localhostu serveri� u portu sysdbau useru u passwordu databaseu as_dictc C@ s t r
t St d f S( uC
Only load this module if all imports succeeded bin exists
uY The mssql execution module cannot be loaded: the pymssql python library is not available.( t HAS_ALL_IMPORTSt Truet False( ( ( s6 /usr/lib/python2.7/site-packages/salt/modules/mssql.pyt __virtual__0 s c K@ sh i } xR d D]J } | | k r0 | | | | <q
t d d | t j | d � � | | <q
Wt j | � S(
Nu serveru portu useru passwordu databaseu as_dictu
config.optionu mssql.( u serveru portu useru passwordu databaseu as_dict( t __salt__t _DEFAULTSt gett Nonet pymssqlt connect( t kwargst connection_argst arg( ( s6 /usr/lib/python2.7/site-packages/salt/modules/mssql.pyt _get_connection9 s
+t
_MssqlEncoderc B@ s e Z d � Z RS( c C@ s
t j | � S( N( t sixt text_type( t selft o( ( s6 /usr/lib/python2.7/site-packages/salt/modules/mssql.pyt defaultE s ( t __name__t
__module__R ( ( ( s6 /usr/lib/python2.7/site-packages/salt/modules/mssql.pyR C s c K@ sw yJ t | � j � } | j | � t t � j i | j � d 6� � d SWn& t k
rr } d t j | � f f SXd S( u
Run a SQL query and return query result as list of tuples, or a list of dictionaries if as_dict was passed, or an empty list if no data is available.
CLI Example:
.. code-block:: bash
salt minion mssql.tsql_query 'SELECT @@version as version' as_dict=True
u resultsetu Could not run the queryN( u Could not run the query(
R t cursort executeR R t encodet fetchallt ExceptionR R ( t queryR t curt err( ( s6 /usr/lib/python2.7/site-packages/salt/modules/mssql.pyt
tsql_queryI s
+c K@ s
t d | � S( u
Return the version of a MS SQL server.
CLI Example:
.. code-block:: bash
salt minion mssql.version
u SELECT @@version( R# ( R ( ( s6 /usr/lib/python2.7/site-packages/salt/modules/mssql.pyt version] s
c K@ s* g t d d t | �D] } | d ^ q S( u�
Return the database list created on a MS SQL server.
CLI Example:
.. code-block:: bash
salt minion mssql.db_list
u SELECT name FROM sys.databasest as_dicti ( R# R ( R t row( ( s6 /usr/lib/python2.7/site-packages/salt/modules/mssql.pyt db_listj s
c K@ s" t t d j | � | � � d k S( u�
Find if a specific database exists on the MS SQL server.
CLI Example:
.. code-block:: bash
salt minion mssql.db_exists database_name='DBNAME'
u6 SELECT database_id FROM sys.databases WHERE NAME='{0}'i ( t lenR# t format( t
database_nameR ( ( s6 /usr/lib/python2.7/site-packages/salt/modules/mssql.pyt db_existsw s u NONEc K@ s� | d k r d Sd j | | � } | rB | d d j | � 7} n d } zT y0 t | � } | j t � | j � j | � Wn t k
r� } d j | � SXWd | r� | j t � | j
� n Xt S(
u�
Creates a new database.
Does not update options of existing databases.
new_database_options can only be a list of strings
CLI Example:
.. code-block:: bash
salt minion mssql.db_create DB_NAME
u NONEu PARTIALu* CONTAINMENT can be one of NONE and PARTIALu( CREATE DATABASE [{0}] CONTAINMENT = {1} u WITH u , u Could not create the login: {0}N( u NONEu PARTIAL( R) t joinR R t
autocommitR R R R R t close( t databaset containmentt new_database_optionsR t sqlt connt e( ( s6 /usr/lib/python2.7/site-packages/salt/modules/mssql.pyt db_create� s"
c K@ s� y� t | | � r� | d k r� t | � } | j t � | j � } | j d j | � � | j d j | � � | j t � | j � t St SWn t k
r� } d j | � SXd S(
u�
Drops a specific database from the MS SQL server.
It will not drop any of 'master', 'model', 'msdb' or 'tempdb'.
CLI Example:
.. code-block:: bash
salt minion mssql.db_remove database_name='DBNAME'
u masteru modelu msdbu tempdbu: ALTER DATABASE {0} SET SINGLE_USER WITH ROLLBACK IMMEDIATEu DROP DATABASE {0}u Could not find the database: {0}N( u masteru modelu msdbu tempdb(
R+ R R- R R R R) R R. R ( R* R R3 R! R4 ( ( s6 /usr/lib/python2.7/site-packages/salt/modules/mssql.pyt db_remove� s
c K@ s t d d d t | � S( up
Lists database roles.
CLI Example:
.. code-block:: bash
salt minion mssql.role_list
R u sp_helproleR% ( R# R ( R ( ( s6 /usr/lib/python2.7/site-packages/salt/modules/mssql.pyt role_list� s c K@ s+ t t d d j | � d t | � � d k S( u~
Checks if a role exists.
CLI Example:
.. code-block:: bash
salt minion mssql.role_exists db_owner
R u sp_helprole "{0}"R% i ( R( R# R) R ( t roleR ( ( s6 /usr/lib/python2.7/site-packages/salt/modules/mssql.pyt role_exists� s c K@ s� | s g } n d j | � } | r: | d j | � 7} n d } z� y` t | � } | j t � | j � j | � x- | D]% } | j � j d j | | � � qy WWn t k
r� } d j | � SXWd | r� | j t � | j � n Xt S( u�
Creates a new database role.
If no owner is specified, the role will be owned by the user that
executes CREATE ROLE, which is the user argument or mssql.user option.
grants is list of strings.
CLI Example:
.. code-block:: bash
salt minion mssql.role_create role=product01 owner=sysdba grants='["SELECT", "INSERT", "UPDATE", "DELETE", "EXECUTE"]'
u CREATE ROLE {0}u AUTHORIZATION {0}u GRANT {0} TO [{1}]u Could not create the role: {0}N(
R) R R R- R R R R R R. ( R8 t ownert grantsR R2 R3 t grantR4 ( ( s6 /usr/lib/python2.7/site-packages/salt/modules/mssql.pyt role_create� s&
'
c K@ s~ yZ t | � } | j t � | j � } | j d j | � � | j t � | j � t SWn t k
ry } d j | � SXd S( u�
Remove a database role.
CLI Example:
.. code-block:: bash
salt minion mssql.role_create role=test_role01
u
DROP ROLE {0}u Could not create the role: {0}N( R R- R R R R) R. R ( R8 R R3 R! R4 ( ( s6 /usr/lib/python2.7/site-packages/salt/modules/mssql.pyt role_remove s
c K@ sh | r d j | | � } n y) t t d d j | � | � � d k SWn t k
rc } d j | � SXd S( u�
Find if a login exists in the MS SQL server.
domain, if provided, will be prepended to login
CLI Example:
.. code-block:: bash
salt minion mssql.login_exists 'LOGIN'
u {0}\{1}R u/ SELECT name FROM sys.syslogins WHERE name='{0}'i u Could not find the login: {0}N( R) R( R# R ( t logint domainR R4 ( ( s6 /usr/lib/python2.7/site-packages/salt/modules/mssql.pyt login_exists s )c
K@ s� t | � t | � k r t St | | | � r2 t S| rM d j | | � } n | s\ g } n | sk g } n d j | � } | r� | d 7} nG t | t j � r� | j d d j | � � n | j d d j | � � | r� | d d j | � 7} n d } z� y` t
| � } | j t � | j
� j | � x- | D]% } | j
� j d j | | � � q3WWn t k
r|} d
j | � SXWd | r�| j t � | j � n Xt S( uK
Creates a new login. Does not update password of existing logins. For
Windows authentication, provide ``new_login_domain``. For SQL Server
authentication, prvide ``new_login_password``. Since hashed passwords are
*varbinary* values, if the ``new_login_password`` is 'int / long', it will
be considered to be HASHED.
new_login_roles
a list of SERVER roles
new_login_options
a list of strings
CLI Example:
.. code-block:: bash
salt minion mssql.login_create LOGIN_NAME database=DBNAME [new_login_password=PASSWORD]
u {0}\{1}u CREATE LOGIN [{0}] u FROM WINDOWS i u PASSWORD=0x{0:x} HASHEDu PASSWORD=N'{0}'u WITH u , u( ALTER SERVER ROLE [{0}] ADD MEMBER [{1}]u Could not create the login: {0}N( t boolR RA R) t
isinstanceR t
integer_typest insertR, R R R- R R R R R. (
R? t new_login_passwordt new_login_domaint new_login_rolest new_login_optionsR R2 R3 R8 R4 ( ( s6 /usr/lib/python2.7/site-packages/salt/modules/mssql.pyt login_create. s@
'
c K@ s~ yZ t | � } | j t � | j � } | j d j | � � | j t � | j � t SWn t k
ry } d j | � SXd S( uy
Removes an login.
CLI Example:
.. code-block:: bash
salt minion mssql.login_remove LOGINNAME
u DROP LOGIN [{0}]u Could not remove the login: {0}N( R R- R R R R) R R. R ( R? R R3 R! R4 ( ( s6 /usr/lib/python2.7/site-packages/salt/modules/mssql.pyt login_removei s
c K@ sS | r d j | | � } n | r. | | d <n t t d d j | � | � � d k S( u�
Find if an user exists in a specific database on the MS SQL server.
domain, if provided, will be prepended to username
CLI Example:
.. code-block:: bash
salt minion mssql.user_exists 'USERNAME' [database='DBNAME']
u {0}\{1}u databaseR u* SELECT name FROM sysusers WHERE name='{0}'i ( R) R( R# ( t usernameR@ R/ R ( ( s6 /usr/lib/python2.7/site-packages/salt/modules/mssql.pyt user_exists s
c K@ s* g t d d t | �D] } | d ^ q S( u�
Get the user list for a specific database on the MS SQL server.
CLI Example:
.. code-block:: bash
salt minion mssql.user_list [database='DBNAME']
u9 SELECT name FROM sysusers where issqluser=1 or isntuser=1R% i ( R# R ( R R& ( ( s6 /usr/lib/python2.7/site-packages/salt/modules/mssql.pyt user_list� s
c K@ s� | r | r d St | | | � r0 d j | � S| ri d j | | � } | r` d j | | � n | } n | r| | | d <n | s� g } n | s� g } n d j | � } | r� | d j | � 7} n
| d 7} | r� | d d j | � 7} n d } z� y` t | � } | j t � | j � j | � x- | D]% } | j � j d
j | | � � q.WWn t k
rw}
d j |
� SXWd | r�| j t
� | j � n Xt S(
u2
Creates a new user. If login is not specified, the user will be created
without a login. domain, if provided, will be prepended to username.
options can only be a list of strings
CLI Example:
.. code-block:: bash
salt minion mssql.user_create USERNAME database=DBNAME
u" domain cannot be set without loginu User {0} already existsu {0}\{1}u databaseu CREATE USER [{0}] u FOR LOGIN [{0}]u WITHOUT LOGINu WITH u , u! ALTER ROLE [{0}] ADD MEMBER [{1}]u Could not create the user: {0}N( RM R) R, R R R- R R R R R R. ( RL R? R@ R/ t rolest optionsR R2 R3 R8 R4 ( ( s6 /usr/lib/python2.7/site-packages/salt/modules/mssql.pyt user_create� sB
!
'
c K@ s� d | k r t SyZ t | � } | j t � | j � } | j d j | � � | j t � | j � t SWn t k
r� } d j | � SXd S( u�
Removes an user.
CLI Example:
.. code-block:: bash
salt minion mssql.user_remove USERNAME database=DBNAME
u databaseu
DROP USER {0}u Could not create the user: {0}N( R R R- R R R R) R. R ( RL R R3 R! R4 ( ( s6 /usr/lib/python2.7/site-packages/salt/modules/mssql.pyt user_remove� s
(& t __doc__t
__future__R R R t jsonR R t salt.ext.sixt extR R
R R t ImportErrorR R
R R R R# R$ R' R+ R R5 R6 R7 R9 R= R> RA RJ RK RM RN RQ RR ( ( ( s6 /usr/lib/python2.7/site-packages/salt/modules/mssql.pyt <module> sF
! % ;
6
Mini Shell