Current File : //usr/lib/python2.7/site-packages/salt/runners/vistara.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 e j e
� Z d � Z d � Z
e d � Z d � Z d � Z d � Z d S(
u
Vistara Runner
Runner to interact with the Vistara (http://www.vistarait.com/) REST API
:codeauthor: Brad Thurber <brad.thurber@gmail.com>
To use this runner, the Vistara client_id and Vistara oauth2 client_key
and client_secret must be set in the master config.
For example ``/etc/salt/master.d/_vistara.conf``:
.. code-block:: yaml
vistara:
client_id: client_012345
client_key: N0tReallyaR3alKeyButShouldB12345
client_secret: ThisI5AreallyLongsecretKeyIwonderwhyTheyMakethemSoBigTheseDays00
i ( t absolute_importt print_functiont unicode_literalsNc C@ s� d t k r t d n d } | r{ | j d d � } | j d d � } | j d d � } | sm | sm | rw t d f St St d f S( u@
Check to see if master config has the necessary config
u vistarau client_idu
client_keyu
client_secretub vistara client_id or client_key or client_secret has not been specified in the Salt master config.ud vistara config has not been specificed in the Salt master config. See documentation for this runner.N( t __opts__t Nonet gett Falset True( t vistara_configt client_idt
client_keyt
client_secret( ( s8 /usr/lib/python2.7/site-packages/salt/runners/vistara.pyt __virtual__$ s
c C@ s1 i t d d d 6t d d d 6t d d d 6S( uF
Return the Vistara configuration read from the master config
u vistarau client_idu
client_keyu
client_secret( R ( ( ( s8 /usr/lib/python2.7/site-packages/salt/runners/vistara.pyt _get_vistara_configuration8 s c C@ s� t � } | s t St | d | d � } | s4 d Sd j | � } t | | d | � } | sc d St | � } | r� | d k r� d j | � Sg } xR | D]J } | d } t j | � t | | d | � }
|
s� t S| j |
� q� W| S(
u�
Deletes a device from Vistara based on DNS name or partial name. By default,
delete_device will only perform the delete if a single host is returned. Set
safety_on=False to delete all matches (up to default API search page size)
CLI Example:
.. code-block:: bash
salt-run vistara.delete_device 'hostname-101.mycompany.com'
salt-run vistara.delete_device 'hostname-101'
salt-run vistara.delete_device 'hostname-1' safety_on=False
u
client_keyu
client_secretu" Vistara access token not availableu dnsName:{0}u client_idu No devices foundi uK Expected to delete 1 device and found {0}. Set safety_on=False to override.u id(
R
R t _get_oath2_access_tokent formatt _search_devicest lent logt debugt _delete_resourcet append( t namet safety_ont configt access_tokent query_stringt devicest device_countt delete_responsest devicet device_idt delete_response( ( s8 /usr/lib/python2.7/site-packages/salt/runners/vistara.pyt
delete_deviceC s.
c C@ s� d j | � } i | d 6d d 6d d 6} i | d 6} d } d j | � } t j j j d | d
| d | d | d
t � } | j d d � } | s� t St j j j
| d � }
|
d } | S( Nu
Bearer {0}u
Authorizationu application/jsonu Content-Typeu Acceptu queryStringu GETu8 https://api.vistara.io/api/v2/tenants/{0}/devices/searcht urlt methodt header_dictt paramst optsu bodyu results( R t saltt utilst httpt queryR R R R t jsont loads( R R R t
authstringt headersR% R# R" t respt respbodyt respbodydictt
deviceresults( ( s8 /usr/lib/python2.7/site-packages/salt/runners/vistara.pyR u s*
c
C@ s� d j | � } i | d 6d d 6d d 6} d } d j | | � } t j j j d | d | d
| d t � } | j d d � } | s� t St j j j
| d � } | S(
Nu
Bearer {0}u
Authorizationu application/jsonu Content-Typeu Acceptu DELETEuC https://api.vistara.io/api/v2/tenants/{0}/rtype/DEVICE/resource/{1}R" R# R$ R& u body( R R' R( R) R* R R R R R+ R, (
R R R R- R. R# R" R/ R0 R1 ( ( s8 /usr/lib/python2.7/site-packages/salt/runners/vistara.pyR � s"
c C@ s� | r | r t j d � t Sd } d } i d d 6d d 6} i d d 6| d
6| d 6} t j j j d | d
| d | d | d t � } | j d d � } | s� t St j j
j | � d } | S( u8
Query the vistara API and get an access_token
uQ client_key and client_secret have not been specified and are required parameters.u POSTu' https://api.vistara.io/auth/oauth/tokenu! application/x-www-form-urlencodedu Content-Typeu application/jsonu Acceptu client_credentialsu
grant_typeu client_idu
client_secretR" R# R$ R% R& u bodyu access_tokenN( R t errorR R' R( R) R* R R R R+ R, ( R
R R# R" R. R% R/ R0 R ( ( s8 /usr/lib/python2.7/site-packages/salt/runners/vistara.pyR � s0
( t __doc__t
__future__R R R t loggingt salt.outputR' t salt.utils.httpt getLoggert __name__R R R
R R! R R R ( ( ( s8 /usr/lib/python2.7/site-packages/salt/runners/vistara.pyt <module> s 2 #
Mini Shell