Current File : //usr/lib/python2.7/site-packages/salt/modules/solrcloud.pyo
�
���^c
@@ s� d Z d d l m Z m Z m Z d d l Z d d l m Z d d l j j
Z
d d l m Z e j
e � Z d d d d d
d d d
d d g
Z d d g Z d d d d d g Z d d d g Z d d g Z d d g Z d
g Z d d d d d d g Z d � Z d d � Z d! � Z d" � Z d# � Z d$ � Z d% � Z d& � Z d d' � Z! d( � Z" d) � Z# d* � Z$ d d+ � Z% d d, � Z& d d- � Z' d. � Z( d/ � Z) d0 � Z* d S(1 u�
Module for solrcloud configuration
.. versionadded:: 2017.7.0
For now, module is limited to http-exposed API. It doesn't implement config upload via Solr zkCli
i ( t absolute_importt unicode_literalst print_functionN( t SaltInvocationError( t sixu configu schemau dataDiru configSetu
propertiesu coreNodeNameu ulogDiru shardu
collectionu rolesu transientu
loadOnStartupu collection.configNameu router.fieldu asyncu ruleu snitchu numShardsu replicationFactoru maxShardsPerNodeu autoAddReplicasu createNodeSet.shuffleu shardsu
createNodeSetc C@ s d S( Nu solrcloud( ( ( ( s: /usr/lib/python2.7/site-packages/salt/modules/solrcloud.pyt __virtual__/ s u http://localhost:8983/solr/c K@ s� t | t j � s! t d � � n | d d k r> | d } n t j | | d d d t d t | �} d | k r� | d d
k r� t d j d | d
| � � � q� t d j d t j | d � d | d
| d | d � � � n | d Sd S( u�
Internal function to query solrcloud
:param url: relative solr URL
:param solr_url: solr base URL
:param kwargs: additional args passed to http.query call
:return: Query JSON answer converted to python dict
:rtype: dict
u solr_url must be a stringi����u /t decode_typeu jsont decodet raise_erroru erroru statusi� uU Got a 404 when trying to contact solr at {solr_url}{url}. Please check your solr URL.t solr_urlt urlu; Got a {status} error when calling {solr_url}{url} : {error}t statust erroru dictN(
t
isinstanceR t string_typest
ValueErrort httpt queryt TrueR t formatt text_type( R
R t kwargst query_result( ( s: /usr/lib/python2.7/site-packages/salt/modules/solrcloud.pyt _query3 s"
( c C@ s� d } x� t j | � D]� \ } } | t k ry t | t � sT t d | d � � n | d | d | ro d n d } q | t k r� t | t j � s� t d | d � � n | d | d | } q | d t j | � d t j | � } q W| S(
u9
Internal function to validate core properties
u u Option "u " value must be an booleanu
&property.u =u trueu falseu' In option "properties", core property "u " value must be a string( R t iteritemst BOOL_PROPS_LISTR
t boolR t STRING_PROPS_LISTR R ( t
propertiest props_stringt prop_namet
prop_value( ( s: /usr/lib/python2.7/site-packages/salt/modules/solrcloud.pyt _validate_core_properties[ s % ,c C@ s� d } x�t j | � D]�\ } } | t k rp t | t j � sW t d | d � � n | d | d | } q | t k r� t | t j � s� t d | d � � n | d | d t j | � } q | t k r t | t
� s� t d | d � � n | d | d | rd n d } q | t k rrt | t � sPt d | d
� � n | d | d d j
| � } q | t k r�t | t � s�t d | d � � n | t | � } q t d
| d � � q W| S( u=
Internal function to validate collections options
u u Option "u " value must be a stringu &u =u " value must be an intu " value must be an booleanu trueu falseu! " value must be a list of stringsu , u " value must be an dictu Unknown option "u "( R R t STRING_OPTIONS_LISTR
R R t INT_OPTIONS_LISTt
integer_typesR t BOOL_OPTIONS_LISTR t LIST_OPTIONS_LISTt listt joint DICT_OPTIONS_LISTt dictR ( t optionst options_stringt option_namet option_value( ( s: /usr/lib/python2.7/site-packages/salt/modules/solrcloud.pyt _validate_collection_optionsv s0 " % " c C@ s t S( u�
Get collection option list that can only be defined at creation
CLI Example:
.. code-block:: bash
salt '*' solrcloud.collection_creation_options
( t CREATION_ONLY_OPTION( ( ( s: /usr/lib/python2.7/site-packages/salt/modules/solrcloud.pyt collection_creation_options� s c K@ s t d | � d S( u�
Get cluster status
Additional parameters (kwargs) may be passed, they will be proxied to http.query
CLI Example:
.. code-block:: bash
salt '*' solrcloud.cluster_status
u. admin/collections?action=CLUSTERSTATUS&wt=jsonu cluster( R ( R ( ( s: /usr/lib/python2.7/site-packages/salt/modules/solrcloud.pyt cluster_status� s c K@ sG t | t j � s! t d � � n t | � } d | k oF | | d k S( u�
Check alias existence
Additional parameters (kwargs) may be passed, they will be proxied to http.query
CLI Example:
.. code-block:: bash
salt '*' solrcloud.alias_exists my_alias
u Alias name must be a stringu aliases( R
R R R R1 ( t
alias_nameR t cluster( ( s: /usr/lib/python2.7/site-packages/salt/modules/solrcloud.pyt alias_exists� s c K@ s� t | t j � s! t d � � n g t j t | � d � D]. } d | d k r; | d | d d f ^ q; } g g | D] } | | d k ry | ^ qy D] } | d ^ q� } | S( u�
Get collection list for an alias
Additional parameters (kwargs) may be passed, they will be proxied to http.query
CLI Example:
.. code-block:: bash
salt '*' solrcloud.alias_get my_alias
u Alias name must be a stringu collectionsu aliasesi i ( R
R R R R R1 ( R2 R t k_vt collection_aliasest k_v1t aliases( ( s: /usr/lib/python2.7/site-packages/salt/modules/solrcloud.pyt alias_get_collections� s .@c K@ sx t | t � s t d � � n x/ | D]' } t | t j � s% t d � � q% q% Wt d j d | d d j | � � | � S( u�
Define an alias
Additional parameters (kwargs) may be passed, they will be proxied to http.query
CLI Example:
.. code-block:: bash
salt '*' solrcloud.alias_set my_alias collections=[collection1, colletion2]
uJ Collection parameter must be defined and contain a list of collection nameu Collection name must be a stringuS admin/collections?action=CREATEALIAS&name={alias}&wt=json&collections={collections}t aliast collectionsu , ( R
R& R R R R R R R' ( R2 R; R t
collection( ( s: /usr/lib/python2.7/site-packages/salt/modules/solrcloud.pyt alias_set_collections� s
c K@ s t d j d | � | � d S( u�
Check if a collection exists
Additional parameters (kwargs) may be passed, they will be proxied to http.query
CLI Example:
.. code-block:: bash
salt '*' solrcloud.collection_reload collection_name
u9 admin/collections?action=RELOAD&name={collection}&wt=jsonR< N( R R ( R< R ( ( s: /usr/lib/python2.7/site-packages/salt/modules/solrcloud.pyt collection_reload s c K@ s t d | � d S( u�
List all collections
Additional parameters (kwargs) may be passed, they will be proxied to http.query
CLI Example:
.. code-block:: bash
salt '*' solrcloud.collection_list
u% admin/collections?action=LIST&wt=jsonu collections( R ( R ( ( s: /usr/lib/python2.7/site-packages/salt/modules/solrcloud.pyt collection_list$ s c K@ s1 t | t j � s! t d � � n | t | � k S( u�
Check if a collection exists
Additional parameters (kwargs) may be passed, they will be proxied to http.query
CLI Example:
.. code-block:: bash
salt '*' solrcloud.collection_exists collection_name
u Collection name must be a string( R
R R R R? ( t collection_nameR ( ( s: /usr/lib/python2.7/site-packages/salt/modules/solrcloud.pyt collection_exists6 s c K@ sk t | | � s t d � � n | d k r< d j | � } n d } t d j d | d | d | � | � d S( u�
Create a backup for a collection.
Additional parameters (kwargs) may be passed, they will be proxied to http.query
CLI Example:
.. code-block:: bash
salt '*' solrcloud.core_backup collection_name /mnt/nfs_backup
u Collection doesn't existsu &name={0}u uP {collection}/replication?command=BACKUP&location={location}{backup_name}&wt=jsonR< t backup_namet locationN( RA R t NoneR R ( R@ RC RB R ( ( s: /usr/lib/python2.7/site-packages/salt/modules/solrcloud.pyt collection_backupK s c K@ sm xf t | � D]X } | d k r: d j d | d | � } n d } t d j d | d | d | � | � q
Wd S( u�
Create a backup for all collection present on the server.
Additional parameters (kwargs) may be passed, they will be proxied to http.query
CLI Example:
.. code-block:: bash
salt '*' solrcloud.core_backup /mnt/nfs_backup
u &name={backup}.{collection}t backupR< u uP {collection}/replication?command=BACKUP&location={location}{backup_name}&wt=jsonRB RC N( R? RD R R ( RC RB R R@ ( ( s: /usr/lib/python2.7/site-packages/salt/modules/solrcloud.pyt collection_backup_allh s c K@ sX | d k r i } n t | t � s3 t d � � n t | � } t d | | | � d S( uY
Create a collection,
Additional parameters (kwargs) may be passed, they will be proxied to http.query
CLI Example:
.. code-block:: bash
salt '*' solrcloud.collection_create collection_name
Collection creation options may be passed using the "options" parameter.
Do not include option "name" since it already specified by the mandatory parameter "collection_name"
.. code-block:: bash
salt '*' solrcloud.collection_create collection_name options={"replicationFactor":2, "numShards":3}
Cores options may be passed using the "properties" key in options.
Do not include property "name"
.. code-block:: bash
salt '*' solrcloud.collection_create collection_name options={"replicationFactor":2, "numShards":3, "properties":{"dataDir":"/srv/solr/hugePartitionSollection"}}
u& options parameter must be a dictionaryu- admin/collections?action=CREATE&wt=json&name=N( RD R
R) R R. R ( R@ R* R R+ ( ( s: /usr/lib/python2.7/site-packages/salt/modules/solrcloud.pyt collection_create� s c C@ s+ y t | � t SWn t k
r& t SXd S( u�
Check collections options
CLI Example:
.. code-block:: bash
salt '*' solrcloud.collection_check_options '{"replicationFactor":4}'
N( R. R R t False( R* ( ( s: /usr/lib/python2.7/site-packages/salt/modules/solrcloud.pyt collection_check_options� s
c K@ s� t | � } i | d | d d 6| d | d d d 6t | d | d � d 6t | d | d � d 6| d | d t k d 6} d
| d | k r� | d | d
| d
<n d | d | k r� | d | d
| d <n | S( u�
Get collection options
Additional parameters (kwargs) may be passed, they will be proxied to http.query
CLI Example:
.. code-block:: bash
salt '*' solrcloud.collection_get_options collection_name
u collectionsu
configNameu collection.configNameu routeru nameu router.nameu replicationFactoru maxShardsPerNodeu autoAddReplicasu ruleu snitch( R1 t intR ( R@ R R3 R* ( ( s: /usr/lib/python2.7/site-packages/salt/modules/solrcloud.pyt collection_get_options� s
c K@ se x= t | j � � D]) } | t k r t d | d � � q q Wt | � } t d | | | � d S( uT
Change collection options
Additional parameters (kwargs) may be passed, they will be proxied to http.query
Note that not every parameter can be changed after collection creation
CLI Example:
.. code-block:: bash
salt '*' solrcloud.collection_set_options collection_name options={"replicationFactor":4}
u Option u- can't be modified after collection creation.u= admin/collections?action=MODIFYCOLLECTION&wt=json&collection=N( R& t keysR/ R R. R ( R@ R* R t optionR+ ( ( s: /usr/lib/python2.7/site-packages/salt/modules/solrcloud.pyt collection_set_options� s
(+ t __doc__t
__future__R R R t loggingt salt.exceptionsR t salt.utils.httpt utilsR t salt.extR t getLoggert __name__t logR R R! R" R$ R% R( R/ R R R R. R0 R1 R4 R9 RD R= R> R? RA RE RG RH RJ RL RO ( ( ( s: /usr/lib/python2.7/site-packages/salt/modules/solrcloud.pyt <module> sD ( - (
Mini Shell