File manager - Edit - /opt/saltstack/salt/lib/python3.10/site-packages/salt/modules/__pycache__/mongodb.cpython-310.pyc
Back
o ;j�v � @ s� d Z ddlZddlZddlZddlmZ ddlm Z zddl Z dZ W n e y- dZ Y nw e� e�Zdd� Z d7d d �Zd d � Zd8dd�Zd8dd�Zd8dd�Zdd� Z d7dd�Z d7dd�Z d7dd�Z d7d d �Z d9d d �Z d7d d!�Z d8d"d#�Z d8d$d%�Z d8d&d'�Z d7d(d)�Z d7d*d+�Z! d7d,d-�Z" d7d.d/�Z# d7d0d1�Z$ d:d2d3�Z% 4 d;d5d6�Z&dS )<a� 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. � N)�get_error_message)�VersionTFc C s t rdS dS )z7 Only load this module if pymongo is installed Zmongodb)FzTThe mongodb execution module cannot be loaded: the pymongo library is not available.)� HAS_MONGODB� r r �H/opt/saltstack/salt/lib/python3.10/site-packages/salt/modules/mongodb.py� __virtual__! s r �adminc C s� | st d d�} |st d d�}|st d d�}|s t d d�}|s$|}z#tj||d�}tj�||�}| rB|rE|j| ||d� W |S W |S W |S tjjyY t� d|� Y d S w ) zv Returns a tuple of (user, host, port) with config, pillar, or default values assigned to missing values. z config.optionz mongodb.userzmongodb.passwordz mongodb.hostz mongodb.port)�host�port)�sourcez Error connecting to database %sF) Z__salt__�pymongoZ MongoClient�database�DatabaseZ authenticate�errors� PyMongoError�log�error)�user�passwordr r r �authdb�conn�mdbr r r �_connect/ s0 ��� �r c C sL zt | t�rtjj�| �} W | S W | S ty% } zt�d|� � d}~ww )zF Potentially interprets a string as JSON for usage with mongo zCould not parse objects: %sN) � isinstance�str�salt�utils�json�loads� ValueErrorr r )�objects�errr r r �_to_dictM s �� ��r" c C sj t | ||||d�}|s dS z t�d� |�� W S tjjy4 } zt�|� t|�W Y d}~S d}~ww )a List all MongoDB databases user The user to connect to MongoDB as. Default is None. password The password to use to connect to MongoDB as. Default is None. host The host where MongoDB is running. Default is None. port The host where MongoDB is running. Default is None. authdb The MongoDB database to use for authentication. Default is None. CLI Example: .. code-block:: bash salt '*' mongodb.db_list <user> <password> <host> <port> �r �#Failed to connect to mongo databasezListing databasesN) r r �infoZlist_database_namesr r r r r )r r r r r r r! r r r �db_list[ s ��r& c C s( t |||||d�}t|t�rdS | |v S )a] Checks if a database exists in MongoDB name The name of the database to check for. user The user to connect to MongoDB as. Default is None. password The password to use to connect to MongoDB as. Default is None. host The host where MongoDB is running. Default is None. port The host where MongoDB is running. Default is None. authdb The MongoDB database to use for authentication. Default is None. CLI Example: .. code-block:: bash salt '*' mongodb.db_exists <name> <user> <password> <host> <port> r# F)r& r r )�namer r r r r Zdbsr r r � db_exists� s r( c C sv t |||||d�}|s dS zt�d| � |�| � W dS tjjy: } zt�d| |� t|�W Y d}~S d}~ww )aM Remove a MongoDB database name The name of the database to remove. user The user to connect to MongoDB as. Default is None. password The password to use to connect to MongoDB as. Default is None. host The host where MongoDB is running. Default is None. port The host where MongoDB is running. Default is None. authdb The MongoDB database to use for authentication. Default is None. CLI Example: .. code-block:: bash salt '*' mongodb.db_remove <name> <user> <password> <host> <port> r# r$ zRemoving database %sz*Removing database %s failed with error: %sNT) r r r% Z drop_databaser r r r r )r' r r r r r r r! r r r � db_remove� s ���r) c C s | � d�d S )NZ buildInfo�version)�command)r r r r �_version� s r, c C s� t | ||||d�}|s d|� d|� �}t�|� d|fS z tj�||�}t|�W S tjjyF } zt�d| � t | �W Y d} ~ S d} ~ ww )a Get MongoDB instance version user The user to connect to MongoDB as. Default is None. password The password to use to connect to MongoDB as. Default is None. host The host where MongoDB is running. Default is None. port The host where MongoDB is running. Default is None. authdb The MongoDB database to use for authentication. Default is None. CLI Example: .. code-block:: bash salt '*' mongodb.version <user> <password> <host> <port> <database> r# �&Failed to connect to MongoDB database �:F�#Listing users failed with error: %sN) r r r r r r r, r r r ) r r r r r r r �err_msgr r! r r r r* � s ��r* c C s� t |||||d�}|s d|� d|� �}t�|� d|fS tj�||�} z | �d| �d W S tjjyL } zt�d| � dt | �fW Y d} ~ S d} ~ ww ) a� Get single user from MongoDB name The name of the user to find. user The user to connect to MongoDB as. Default is None. password The password to use to connect to MongoDB as. Default is None. host The host where MongoDB is running. Default is None. port The host where MongoDB is running. Default is None. database The MongoDB database to use when looking for the user. Default is ``admin``. authdb The MongoDB database to use for authentication. Default is None. CLI Example: .. code-block:: bash salt '*' mongodb.user_find <name> <user> <password> <host> <port> <database> <authdb> r# r- r. F� usersInfo�usersr/ N) r r r r r r r+ r r r ) r' r r r r r r r r0 r r! r r r � user_find� s ! ��r3 c C s� t | ||||d�}|s dS zMt�d� tj�||�}g }t|�} t| �td�krA|�d�d D ]} |� | d | d d �� q/|W S |j j � � D ]} |� | d | � d d �d �� qG|W S tjjyx } zt�d | � t| �W Y d} ~ S d} ~ ww )a~ List users of a MongoDB database user The user to connect to MongoDB as. Default is None. password The password to use to connect to MongoDB as. Default is None. host The host where MongoDB is running. Default is None. port The host where MongoDB is running. Default is None. database The MongoDB database to use when listing users. Default is ``admin``. authdb The MongoDB database to use for authentication. Default is None. CLI Example: .. code-block:: bash salt '*' mongodb.user_list <user> <password> <host> <port> <database> r# r$ z Listing usersz2.6r1 r2 r �roles)r r4 �readOnly�None)r r5 r/ N)r r r% r r r r, r r+ �append�systemr2 �find�getr r r r ) r r r r r r r r �outputZmongodb_versionr! r r r � user_list* s, �� ��r< c C sF t ||||||�}t|t�rdS |D ]}| t|��d�kr dS qdS )a� Checks if a user exists in MongoDB user The user to connect to MongoDB as. Default is None. password The password to use to connect to MongoDB as. Default is None. host The host where MongoDB is running. Default is None. port The host where MongoDB is running. Default is None. database The MongoDB database to use when checking if the user exists. Default is ``admin``. authdb The MongoDB database to use for authentication. Default is None. CLI Example: .. code-block:: bash salt '*' mongodb.user_exists <name> <user> <password> <host> <port> <database> r$ r TF)r< r r �dictr: )r' r r r r r r r2 r r r � user_existsb s �r> c s� t |||||d�} | s dS |sg }� fdd�|D �} zt�d| � tj�| � �} | jd| || d� W d S tjjyO } zt� d| | � W Y d } ~ d S d } ~ ww ) au Create a MongoDB user name The name of the user to create. passwd The password for the user that is being created. user The user to connect to MongoDB as. Default is None. password The password to use to connect to MongoDB as. Default is None. host The host where MongoDB is running. Default is None. port The host where MongoDB is running. Default is None. database The MongoDB database to use when checking if the user exists. Default is ``admin``. authdb The MongoDB database to use for authentication. Default is None. roles The roles that should be associated with the user. Default is None. CLI Example: .. code-block:: bash salt '*' mongodb.user_create <user_name> <user_password> <roles> <user> <password> <host> <port> <database> r# r$ c s g | ]}|� d ��qS )��roleZdbr )�.0Z_role�r r r � <listcomp>� s z user_create.<locals>.<listcomp>zCreating user %sZ createUser)�pwdr4 z&Creating user %s failed with error: %sNFT� r r r% r r r r+ r r r ) r' �passwdr r r r r r r4 r Z_rolesr r! r rB r � user_create� s / ���rG c C s| t ||||�}|s dS zt�d| � tj�||�}|�d| � W dS tjjy= } zt� d| | � W Y d} ~ dS d} ~ ww )a_ Remove a MongoDB user name The name of the user that should be removed. user The user to connect to MongoDB as. Default is None. password The password to use to connect to MongoDB as. Default is None. host The host where MongoDB is running. Default is None. port The host where MongoDB is running. Default is None. authdb The MongoDB database to use for authentication. Default is None. CLI Example: .. code-block:: bash salt '*' mongodb.user_remove <name> <user> <password> <host> <port> <database> r$ zRemoving user %sZdropUserz&Removing user %s failed with error: %sNFTrE ) r' r r r r r r r r r! r r r � user_remove� s ���rH c C s� zt |�}W n ty Y dS w t||||||�}t|t�r!dS |D ]+}| t|��d�krN|D ]} t| t�s<| |d�} | t|��dg �vrJ dS q0 dS q#dS )a Checks if a user of a MongoDB database has specified roles name The name of the user to check for the specified roles. roles The roles to check are associated with the specified user. database The database to check has the specified roles for the specified user. user The user to connect to MongoDB as. Default is None. password The password to use to connect to MongoDB as. Default is None. host The host where MongoDB is running. Default is None. port The host where MongoDB is running. Default is None. authdb The MongoDB database to use for authentication. Default is None. 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 � Roles provided in wrong formatr$ r r? r4 FT)r" � Exceptionr< r r r= r: ) r' r4 r r r r r r r2 r@ r r r �user_roles_exists� s& ( � �� rK c C � t |||||d�}|s dS zt|�}W n ty Y dS w zt�d|| � tj�||�} | jd| |d� W d S tj j yW } zt� d|| | � t | �W Y d} ~ S d} ~ ww ) a� Grant one or many roles to a MongoDB user name The user to grant the specified roles to. roles The roles to grant to the specified user. database The database to great the roles against for the specified user. user The user to connect to MongoDB as. Default is None. password The password to use to connect to MongoDB as. Default is None. host The host where MongoDB is running. Default is None. port The host where MongoDB is running. Default is None. authdb The MongoDB database to use for authentication. Default is None. 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# r$ rI z Granting roles %s to user %sZgrantRolesToUser�r4 z2Granting roles %s to user %s failed with error: %sNT� r r" rJ r r% r r r r+ r r r r � r' r4 r r r r r r r r r! r r r �user_grant_roles9 s( ( �����rP c C rL ) a� Revoke one or many roles to a MongoDB user user The user to connect to MongoDB as. Default is None. roles The roles to revoke from the specified user. database The database to revoke the roles from for the specified user. password The password to use to connect to MongoDB as. Default is None. host The host where MongoDB is running. Default is None. port The host where MongoDB is running. Default is None. authdb The MongoDB database to use for authentication. Default is None. 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# r$ rI z Revoking roles %s from user %sZrevokeRolesFromUserrM z4Revoking roles %s from user %s failed with error: %sNTrN rO r r r �user_revoke_rolesw s( % �����rQ c C � t ||||||�}|s dS zt�d|| � tj�||�}|�| � W dS tjjyA } zt� d|| | � | W Y d} ~ S d} ~ ww )a� .. versionadded:: 3006.0 Create a collection in the specified database. collection The name of the collection to create. user The user to connect to MongoDB as. Default is None. password The password to use to connect to MongoDB as. Default is None. host The host where MongoDB is running. Default is None. port The host where MongoDB is running. Default is None. authdb The MongoDB database to use for authentication. Default is None. CLI Example: .. code-block:: bash salt '*' mongodb.collection_create mycollection <user> <password> <host> <port> <database> r$ zCreating %s.%s�.Creating collection %r.%r failed with error %sNT) r r r% r r r Zcreate_collectionr r r � � collectionr r r r r r r r r! r r r �collection_create� � ' ����rV c C rR )a� .. versionadded:: 3006.0 Drop a collection in the specified database. collection The name of the collection to drop. user The user to connect to MongoDB as. Default is None. password The password to use to connect to MongoDB as. Default is None. host The host where MongoDB is running. Default is None. port The host where MongoDB is running. Default is None. authdb The MongoDB database to use for authentication. Default is None. CLI Example: .. code-block:: bash salt '*' mongodb.collection_drop mycollection <user> <password> <host> <port> <database> r$ zDropping %s.%srS NT) r r r% r r r Zdrop_collectionr r r rT r r r �collection_drop� rW rX c C sp t | |||||�}|s dS ztj�||�}|�� }W |S tjjy7 } zt�d| � | W Y d} ~ S d} ~ ww )an .. versionadded:: 3006.0 List the collections available in the specified database. user The user to connect to MongoDB as. Default is None. password The password to use to connect to MongoDB as. Default is None. host The host where MongoDB is running. Default is None. port The host where MongoDB is running. Default is None. authdb The MongoDB database to use for authentication. Default is None. CLI Example: .. code-block:: bash salt '*' mongodb.collections_list mycollection <user> <password> <host> <port> <database> r$ z(Listing collections failed with error %sN) r r r r Zlist_collection_namesr r r r ) r r r r r r r r �retr! r r r �collections_list s # � ��rZ c C s� t ||||||�}|s dS zt| �} W n ty' } z| W Y d} ~ S d} ~ ww z t�d| ||� tj�||�} t| |�} | � | �} | j W S tj j yb } zt� d| | � | W Y d} ~ S d} ~ ww )a" Insert an object or list of objects into a collection objects The objects to insert into the collection, should be provided as a list. collection The collection to insert the objects into. user The user to connect to MongoDB as. Default is None. password The password to use to connect to MongoDB as. Default is None. host The host where MongoDB is running. Default is None. port The host where MongoDB is running. Default is None. authdb The MongoDB database to use for authentication. Default is None. CLI Example: .. code-block:: bash salt '*' mongodb.insert '[{"foo": "FOO", "bar": "BAR"}, {"foo": "BAZ", "bar": "BAM"}]' mycollection <user> <password> <host> <port> <database> r$ NzInserting %r into %s.%sz)Inserting objects %r failed with error %s)r r" rJ r r% r r r �getattrZ insert_manyZ acknowledgedr r r ) r rU r r r r r r r r! r �col�idsr r r �insertP s( ) �� ��r^ c C s� t ||||||�}|s dS t| �} t�d| �} t| �dkr dS | d d | d< d| d | d< g } | D ]#} z t| �} | �| � W q4 tyW } z | W Y d } ~ S d } ~ ww | d } | d }t|| d ||||||�}t |t �std S |sxd S t|�dkr�d S z%t � d | ||� t j�||�}t||�}|�| d|i�}|j}|� d�W S t jjy� } zt �d| | � | W Y d } ~ S d } ~ ww )a� 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 objects The objects to update in the collection, should be provided as a list. collection The collection to insert the objects into. user The user to connect to MongoDB as. Default is None. password The password to use to connect to MongoDB as. Default is None. host The host where MongoDB is running. Default is None. port The host where MongoDB is running. Default is None. authdb The MongoDB database to use for authentication. Default is None. CLI Example: .. code-block:: bash salt '*' mongodb.update_one '{"_id": "my_minion"} {"bar": "BAR"}' mycollection <user> <password> <host> <port> <database> r$ z}\s+{� zMYour request does not contain a valid '{_"id": "my_id"} {"my_doc": "my_val"}'r �}�{� NzGThe find result is not well formatted. An error appears; cannot update.z9Did not find any result. You should try an insert before.z1Too many results. Please try to be more specific.zUpdating %r into %s.%sz$setz objects updatedz'Updating object %s failed with error %s)r r �re�split�lenr" r7 rJ r9 r �listr r% r r r r[ � update_oneZmodified_countr r r )r rU r r r r r r r ZobjsZdocument�objr! Z _id_fieldZ _update_docZtest_fr r\ r] Znb_modr r r rg � sP , ��� ��rg c C s t ||||||�}|s dS zt|�}W n ty' } z| W Y d} ~ S d} ~ ww zBt�d|| � tj�||�} t| | �} t |t �r\g } |D ]} | � | �}dd� |D �}| � |� qE| W S | � |�}dd� |D �} | W S tj jy� } zt�d| � | W Y d} ~ S d} ~ ww )a� Find an object or list of objects in a collection collection The collection to find the objects in. query The query to use when locating objects in the collection. user The user to connect to MongoDB as. Default is None. password The password to use to connect to MongoDB as. Default is None. host The host where MongoDB is running. Default is None. port The host where MongoDB is running. Default is None. authdb The MongoDB database to use for authentication. Default is None. CLI Example: .. code-block:: bash salt '*' mongodb.find mycollection '[{"foo": "FOO", "bar": "BAR"}]' <user> <password> <host> <port> <database> r$ NzSearching for %r in %sc S � g | ]}|�qS r r �rA Z_resr r r rC $ � zfind.<locals>.<listcomp>c S ri r r rj r r r rC ( rk z'Searching objects failed with error: %s)r r" rJ r r% r r r r[ r rf r9 �extendr r r )rU �queryr r r r r r r r! r r\ rY �_query�resZ_retr r r r9 � s8 ) �� � ��r9 rb c C s* t ||||||�} | s dS zt|�}W n ty) } z t| �W Y d} ~ S d} ~ ww zJt�d|| � tj�| |�} t | | �} d} t |t �r]|D ]}t d|�D ] }| � |�}| |j7 } qNqGnt d|�D ] }| � |�}| |j7 } qb| � d�W S tjjy� } zt�dt| �� t| �W Y d} ~ S d} ~ ww )a� Remove an object or list of objects from a collection collection The collection to remove objects from based on the query. query Query to determine which objects to remove. user The user to connect to MongoDB as. Default is None. password The password to use to connect to MongoDB as. Default is None. host The host where MongoDB is running. Default is None. port The host where MongoDB is running. Default is None. database The database where the collection is. w The number of matches to remove from the collection. authdb The MongoDB database to use for authentication. Default is None. CLI Example: .. code-block:: bash salt '*' mongodb.remove mycollection '[{"foo": "FOO", "bar": "BAR"}, {"foo": "BAZ", "bar": "BAM"}]' <user> <password> <host> <port> <database> r$ NzRemoving %r from %sr z objects removedz&Removing objects failed with error: %s)r r" rJ �_get_error_messager r% r r r r[ r rf �rangeZ delete_one� deleted_countr r r )rU rm r r r r r �wr r r! r r\ rr rn �countro r r r �remove/ s< 0 �� �� ��ru )NNNNr N)NNNNN)NNNNr NN)NNNNNr N)NNNNNr rb N)'�__doc__�loggingrc Zsalt.utils.jsonr Zsalt.exceptionsr rp Zsalt.utils.versionsr r r � ImportError� getLogger�__name__r r r r" r& r( r) r, r* r3 r< r> rG rH rK rP rQ rV rX rZ r^ rg r9 ru r r r r �<module> s� � � % $* �* �0 �9 �- �C �- �@ �? �= �9 �8 �3 �@ �_ �G�
| ver. 1.4 |
Github
|
.
| PHP 8.2.30 | Generation time: 0 |
proxy
|
phpinfo
|
Settings