Current File : //usr/lib/python2.7/site-packages/salt/modules/mongodb.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 m
Z d d l m
Z d d l m Z y d d l Z e Z Wn e k
r� e Z n Xe j e � Z d � Z d d d d d d d � Z d
� Z d d d d d d � Z d d d d d d � Z d d d d d d
� Z d � Z d d d d d d d � Z! d d d d d d d � Z" d d d d d d d � Z# d d d d d d d � Z$ d d d d d d d d � Z% d d d d d d d � Z& d d d d d d � Z' d d d d d d � Z( d d d d d d � Z) d d d d d d d � Z* d d d d d d d � Z+ d d d d d d d d � Z, d d d d d d d d d � Z- d S( u�
Module to provide MongoDB functionality to Salt
:configuration: This module uses PyMongo, and accepts configuration details as
parameters as well as configuration settings::
mongodb.host: 'localhost'
mongodb.port: 27017
mongodb.user: ''
mongodb.password: ''
This data can also be passed into pillar. Options passed into opts will
overwrite options passed into pillar.
i ( t absolute_importt print_functiont unicode_literalsN( t LooseVersion( t get_error_message( t sixc C@ s t r
d St d f Sd S( u7
Only load this module if pymongo is installed
u mongodbuT The mongodb execution module cannot be loaded: the pymongo library is not available.N( t HAS_MONGODBt False( ( ( s8 /usr/lib/python2.7/site-packages/salt/modules/mongodb.pyt __virtual__' s u adminc C@ s� | s t d d � } n | s2 t d d � } n | sK t d d � } n | sd t d d � } n | ss | } n yV t j d | d | � } t j j | | � } | r� | r� | j | | d | �n Wn( t j j k
r� t j d | � t
SX| S(
uv
Returns a tuple of (user, host, port) with config, pillar, or default
values assigned to missing values.
u
config.optionu mongodb.useru mongodb.passwordu mongodb.hostu mongodb.portt hostt portt sourceu Error connecting to database %s( t __salt__t pymongot MongoClientt databaset Databaset authenticatet errorst PyMongoErrort logt errorR ( t usert passwordR R
R t authdbt connt mdb( ( s8 /usr/lib/python2.7/site-packages/salt/modules/mongodb.pyt _connect1 s& c C@ s^ y. t | t j � r- t j j j | � } n Wn) t k
rY } t j d | � | � n X| S( uF
Potentially interprets a string as JSON for usage with mongo
u Could not parse objects: %s(
t
isinstanceR t string_typest saltt utilst jsont loadst
ValueErrorR R ( t objectst err( ( s8 /usr/lib/python2.7/site-packages/salt/modules/mongodb.pyt _to_dictM s
c C@ sw t | | | | d | �} | s% d Sy t j d � | j � SWn0 t j j k
rr } t j | � t j | � SXd S( u�
List all MongoDB databases
CLI Example:
.. code-block:: bash
salt '*' mongodb.db_list <user> <password> <host> <port>
R u# Failed to connect to mongo databaseu Listing databasesN(
R R t infot database_namesR
R R R R t text_type( R R R R
R R R$ ( ( s8 /usr/lib/python2.7/site-packages/salt/modules/mongodb.pyt db_list[ s
c C@ s; t | | | | d | �} t | t j � r1 t S| | k S( u�
Checks if a database exists in MongoDB
CLI Example:
.. code-block:: bash
salt '*' mongodb.db_exists <name> <user> <password> <host> <port>
R ( R) R R R R ( t nameR R R R
R t dbs( ( s8 /usr/lib/python2.7/site-packages/salt/modules/mongodb.pyt db_existsq s
c C@ s� t | | | | d | �} | s% d Sy! t j d | � | j | � Wn6 t j j k
r~ } t j d | | � t j | � SXt
S( u�
Remove a MongoDB database
CLI Example:
.. code-block:: bash
salt '*' mongodb.db_remove <name> <user> <password> <host> <port>
R u# Failed to connect to mongo databaseu Removing database %su* Removing database %s failed with error: %s( R R R& t
drop_databaseR
R R R R R( t True( R* R R R R
R R R$ ( ( s8 /usr/lib/python2.7/site-packages/salt/modules/mongodb.pyt db_remove� s
c C@ s | j d � d S( Nu buildInfou version( t command( R ( ( s8 /usr/lib/python2.7/site-packages/salt/modules/mongodb.pyt _version� s c
C@ s� t | | | | d | �} | sJ d j | | � } t j | � t | f Sy# t j j | | � } t | � SWn3 t j j
k
r� } t j d | � t j | � SXd S( u�
Get MongoDB instance version
CLI Example:
.. code-block:: bash
salt '*' mongodb.version <user> <password> <host> <port> <database>
R u- Failed to connect to MongoDB database {0}:{1}u# Listing users failed with error: %sN(
R t formatR R R R
R R R1 R R R R( (
R R R R
R R R t err_msgR R$ ( ( s8 /usr/lib/python2.7/site-packages/salt/modules/mongodb.pyt version� s
c C@ s� t | | | | d | �} | sJ d j | | � } t j | � t | f St j j | | � } y | j d | � d SWn9 t j j
k
r� }
t j d |
� t t j |
� f SXd S( u�
Get single user from MongoDB
CLI Example:
.. code-block:: bash
salt '*' mongodb.user_find <name> <user> <password> <host> <port> <database> <authdb>
R u- Failed to connect to MongoDB database {0}:{1}u usersInfou usersu# Listing users failed with error: %sN(
R R2 R R R R
R R R0 R R R R( ( R* R R R R
R R R R3 R R$ ( ( s8 /usr/lib/python2.7/site-packages/salt/modules/mongodb.pyt user_find� s
c C@ s? t | | | | d | �} | s% d Sy� t j d � t j j | | � } g } t | � } t | � t d � k r� x� | j d � d D]) } | j i | d d 6| d d 6� q� WnH xE | j
j j � D]1 } | j i | d d 6| j
d d
� d 6� q� W| SWn3 t j j k
r:}
t j d |
� t j |
� SXd S(
u�
List users of a MongoDB database
CLI Example:
.. code-block:: bash
salt '*' mongodb.user_list <user> <password> <host> <port> <database>
R u# Failed to connect to mongo databaseu
Listing usersu 2.6u usersInfou usersu useru rolesu readOnlyu Noneu# Listing users failed with error: %sN( R R R& R
R R R1 t
_LooseVersionR0 t appendt systemt userst findt getR R R R R( ( R R R R
R R R R t outputt mongodb_versionR$ ( ( s8 /usr/lib/python2.7/site-packages/salt/modules/mongodb.pyt user_list� s*
c C@ se t | | | | | | � } t | t j � r1 d Sx- | D]% } | t | � j d � k r8 t Sq8 Wt S( u�
Checks if a user exists in MongoDB
CLI Example:
.. code-block:: bash
salt '*' mongodb.user_exists <name> <user> <password> <host> <port> <database>
u# Failed to connect to mongo databaseu user( R> R R R t dictR; R. R ( R* R R R R
R R R9 ( ( s8 /usr/lib/python2.7/site-packages/salt/modules/mongodb.pyt user_exists� s
c C@ s� t | | | | d | �} | s% d S| s4 g } n y? t j d | � t j j | | � }
|
j | | d | �Wn6 t j j k
r� } t j d | | � t
j | � SXt S( u�
Create a MongoDB user
CLI Example:
.. code-block:: bash
salt '*' mongodb.user_create <user_name> <user_password> <roles> <user> <password> <host> <port> <database>
R u# Failed to connect to mongo databaseu Creating user %st rolesu* Creating database %s failed with error: %s(
R R R& R
R R t add_userR R R R R( R. ( R* t passwdR R R R
R R RA R R R$ ( ( s8 /usr/lib/python2.7/site-packages/salt/modules/mongodb.pyt user_create s c
C@ s� t | | | | � } | s d Sy6 t j d | � t j j | | � } | j | � Wn6 t j j k
r� } t j d | | � t
j | � SXt S( u�
Remove a MongoDB user
CLI Example:
.. code-block:: bash
salt '*' mongodb.user_remove <name> <user> <password> <host> <port> <database>
u# Failed to connect to mongo databaseu Removing user %su* Creating database %s failed with error: %s(
R R R& R
R R t remove_userR R R R R( R. (
R* R R R R
R R R R R$ ( ( s8 /usr/lib/python2.7/site-packages/salt/modules/mongodb.pyt user_remove+ s c
C@ s� y t | � } Wn t k
r$ d SXt | | | | | | � } t | t j � rV d Sx� | D]~ } | t | � j d � k r] xV | D]N } t | t � s� i | d 6| d 6} n | t | � j d g � k r� t Sq� Wt Sq] Wt S( u�
Checks if a user of a MongoDB database has specified roles
CLI Examples:
.. code-block:: bash
salt '*' mongodb.user_roles_exists johndoe '["readWrite"]' dbname admin adminpwd localhost 27017
.. code-block:: bash
salt '*' mongodb.user_roles_exists johndoe '[{"role": "readWrite", "db": "dbname" }, {"role": "read", "db": "otherdb"}]' dbname admin adminpwd localhost 27017
u Roles provided in wrong formatu# Failed to connect to mongo databaseu useru roleu dbu roles(
R% t ExceptionR> R R R R? R; R R. (
R* RA R R R R R
R R9 t role( ( s8 /usr/lib/python2.7/site-packages/salt/modules/mongodb.pyt user_roles_existsE s
c C@ s� t | | | | d | �} | s% d Sy t | � } Wn t k
rI d SXyB t j d | | � t j j | | � } | j d | d | �Wn9 t j j
k
r� }
t j d | | |
� t j
|
� SXt S( u�
Grant one or many roles to a MongoDB user
CLI Examples:
.. code-block:: bash
salt '*' mongodb.user_grant_roles johndoe '["readWrite"]' dbname admin adminpwd localhost 27017
.. code-block:: bash
salt '*' mongodb.user_grant_roles janedoe '[{"role": "readWrite", "db": "dbname" }, {"role": "read", "db": "otherdb"}]' dbname admin adminpwd localhost 27017
R u# Failed to connect to mongo databaseu Roles provided in wrong formatu Granting roles %s to user %su grantRolesToUserRA u2 Granting roles %s to user %s failed with error: %s( R R% RG R R& R
R R R0 R R R R R( R. ( R* RA R R R R R
R R R R$ ( ( s8 /usr/lib/python2.7/site-packages/salt/modules/mongodb.pyt user_grant_rolesk s
c C@ s� t | | | | d | �} | s% d Sy t | � } Wn t k
rI d SXyB t j d | | � t j j | | � } | j d | d | �Wn9 t j j
k
r� }
t j d | | |
� t j
|
� SXt S( u�
Revoke one or many roles to a MongoDB user
CLI Examples:
.. code-block:: bash
salt '*' mongodb.user_revoke_roles johndoe '["readWrite"]' dbname admin adminpwd localhost 27017
.. code-block:: bash
salt '*' mongodb.user_revoke_roles janedoe '[{"role": "readWrite", "db": "dbname" }, {"role": "read", "db": "otherdb"}]' dbname admin adminpwd localhost 27017
R u# Failed to connect to mongo databaseu Roles provided in wrong formatu Revoking roles %s from user %su revokeRolesFromUserRA u4 Revoking roles %s from user %s failed with error: %s( R R% RG R R& R
R R R0 R R R R R( R. ( R* RA R R R R R
R R R R$ ( ( s8 /usr/lib/python2.7/site-packages/salt/modules/mongodb.pyt user_revoke_roles� s
c
C@ s� t | | | | | | � } | s% d Sy t | � } Wn t k
rK } | SXyQ t j d | | | � t j j | | � }
t |
| � } | j | � } | SWn- t j
j k
r� } t j d | | � | SXd S( u
Insert an object or list of objects into a collection
CLI Example:
.. code-block:: bash
salt '*' mongodb.insert '[{"foo": "FOO", "bar": "BAR"}, {"foo": "BAZ", "bar": "BAM"}]' mycollection <user> <password> <host> <port> <database>
u# Failed to connect to mongo databaseu Inserting %r into %s.%su) Inserting objects %r failed with error %sN(
R R% RG R R& R
R R t getattrt insertR R R (
R# t
collectionR R R R
R R R R$ R t colt ids( ( s8 /usr/lib/python2.7/site-packages/salt/modules/mongodb.pyRM � s c C@ s� t | | | | | | � } | s% d St j | � } t j d | � } t | � d k r` d d S| d d | d <d | d | d <g }
xB | D]: } y t | � } |
j | � Wq� t k
r� } | SXq� W|
d }
|
d } t | | d | | | | | | � } t
| t � sd
St | � d k r1d St | � d k rGd Sym t j
d
|
| | � t j j | | � } t | | � } | j |
i | d 6� } | j } d j | � SWn- t j j k
r�} t j d | | � | SXd S( ul
Update an object into a collection
http://api.mongodb.com/python/current/api/pymongo/collection.html#pymongo.collection.Collection.update_one
.. versionadded:: 2016.11.0
CLI Example:
.. code-block:: bash
salt '*' mongodb.update_one '{"_id": "my_minion"} {"bar": "BAR"}' mycollection <user> <password> <host> <port> <database>
u# Failed to connect to mongo databaseu }\s+{i u& Your request does not contain a valid u' '{_"id": "my_id"} {"my_doc": "my_val"}'i u }u {i uG The find result is not well formatted. An error appears; cannot update.u9 Did not find any result. You should try an insert before.u1 Too many results. Please try to be more specific.u Updating %r into %s.%su $setu {0} objects updatedu' Updating object %s failed with error %sN( R R R( t ret splitt lenR% R7 RG R: R t listR R& R
R R RL t
update_onet modified_countR2 R R R ( R# RN R R R R
R R R t objst documentt objR$ t _id_fieldt _update_doct test_fR RO RP t nb_mod( ( s8 /usr/lib/python2.7/site-packages/salt/modules/mongodb.pyRU � sV
c
C@ s� t | | | | | | � } | s% d Sy t | � } Wn t k
rK } | SXyT t j d | | � t j j | | � }
t |
| � } | j | � } t
| � SWn* t j j k
r� } t j
d | � | SXd S( u�
Find an object or list of objects in a collection
CLI Example:
.. code-block:: bash
salt '*' mongodb.find mycollection '[{"foo": "FOO", "bar": "BAR"}]' <user> <password> <host> <port> <database>
u# Failed to connect to mongo databaseu Searching for %r in %su' Searching objects failed with error: %sN( R R% RG R R& R
R R RL R: RT R R R (
RN t queryR R R R
R R R R$ R RO t ret( ( s8 /usr/lib/python2.7/site-packages/salt/modules/mongodb.pyR: s i c C@ s� t | | | | | | � } | s% d Sy t | � } Wn t k
rQ }
t |
� SXya t j d | | � t j j | | � } t | | � } | j
| d | �}
d j |
d � SWn6 t j j
k
r� }
t j d t |
� � t |
� SXd S( u
Remove an object or list of objects into a collection
CLI Example:
.. code-block:: bash
salt '*' mongodb.remove mycollection '[{"foo": "FOO", "bar": "BAR"}, {"foo": "BAZ", "bar": "BAM"}]' <user> <password> <host> <port> <database>
u# Failed to connect to mongo databaseu Removing %r from %st wu {0} objects removedu nu& Removing objects failed with error: %sN( R R% RG t _get_error_messageR R& R
R R RL t removeR2 R R R ( RN R^ R R R R
R R` R R R$ R RO R_ ( ( s8 /usr/lib/python2.7/site-packages/salt/modules/mongodb.pyRb 5 s (. t __doc__t
__future__R R R t loggingRQ t salt.utils.jsonR t salt.utils.versionsR R6 t salt.exceptionsR Ra t salt.extR R
R. R t ImportErrorR t getLoggert __name__R R t NoneR R% R) R, R/ R1 R4 R5 R> R@ RD RF RI RJ RK RM RU R: Rb ( ( ( s8 /usr/lib/python2.7/site-packages/salt/modules/mongodb.pyt <module> sV
( % " " D
Mini Shell