Current File : //usr/lib/python2.7/site-packages/salt/engines/hipchat.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 y d d l Z Wn e k
ro d Z n Xd d l Z d d l
Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z e j e � Z d Z d Z d Z d Z d � Z d
d d d � Z d
d d d
� Z! d
d d d � Z" d d d d e# d d d d d d d d d �
Z$ d S( uj
An engine that reads messages from Hipchat and sends them to the Salt
event bus. Alternatively Salt commands can be sent to the Salt master
via Hipchat by setting the control parameter to ``True`` and using command
prefaced with a ``!``. Only token key is required, but room and control
keys make the engine interactive.
.. versionadded: 2016.11.0
:depends: hypchat
:configuration: Example configuration
.. code-block:: yaml
engines:
- hipchat:
api_url: http://api.hipchat.myteam.com
token: 'XXXXXX'
room: 'salt'
control: True
valid_users:
- SomeUser
valid_commands:
- test.ping
- cmd.run
- list_jobs
- list_commands
aliases:
list_jobs:
cmd: jobs.list_jobs
list_commands:
cmd: pillar.get salt:engines:hipchat:valid_commands target=saltmaster
max_rooms: 0
wait_time: 1
i ( t absolute_importt print_functiont unicode_literalsN( t sixu https://api.hipchat.comi i� u hipchatc C@ s t d k r t St d f S( Nu hypchat is not installed( t hypchatt Nonet __virtualname__t False( ( ( s8 /usr/lib/python2.7/site-packages/salt/engines/hipchat.pyt __virtual__F s u c C@ s) t j j | � s* t d j | � � � n t | � d k rK t d � � n d j | | � } i d d 6} d | | d <t j j j i | d 6� } t j j
j | d
� �R } t d � j | t j j
t j j j | � � t j j j | j � � � }
Wd QXt j j j | d
d d | d |
�d S( u�
Send file to a HipChat room via API version 2
Parameters
----------
token : str
HipChat API version 2 compatible token - must be token for active user
room: str
Name or API ID of the room to notify
filepath: str
Full path of file to be sent
message: str, optional
Message to send to room
api_url: str, optional
Hipchat API URL to use, defaults to http://api.hipchat.com
u File '{0}' does not existi� u Message too longu {0}/v2/room/{1}/share/fileu* multipart/related; boundary=boundary123456u Content-typeu Bearer u
Authorizationu messageu rbu� --boundary123456
Content-Type: application/json; charset=UTF-8
Content-Disposition: attachment; name="metadata"
{0}
--boundary123456
Content-Disposition: attachment; name="file"; filename="{1}"
{2}
--boundary123456--Nt methodu POSTt header_dictt data( t ost patht isfilet
ValueErrort formatt lent saltt utilst jsont dumpst filest fopent strt basenamet stringutilst to_strt readt httpt query( t tokent roomt filepatht messaget outputtert api_urlt urlt headerst msgt rfht payload( ( s8 /usr/lib/python2.7/site-packages/salt/engines/hipchat.pyt
_publish_fileK s
$u nestedc
C@ s� d j | | � } i d d 6} d | | d <t j j j | d d | d t d t d
| d t �d | d <t j j | | d t �} t j j j | d d | d t d t d
| d t �d
S( u/
Publishes the HTML-formatted message.
u {0}/v2/room/{1}/notificationu
text/plainu Content-typeu Bearer u
Authorizationu POSTR t decodet statusR
t optsu text/htmlN( R R R R R t Truet __opts__t outputt html_format( R R R R" R# R$ R% R& ( ( s8 /usr/lib/python2.7/site-packages/salt/engines/hipchat.pyt _publish_html_message~ s,
c
C@ s� d j | | � } i d d 6} d | | d <t j j j | d d | d t d t d
| d t �d } | t j j | | d t �7} t j j j | d d | d t d t d
| d t �d
S( u.
Publishes the output format as code.
u {0}/v2/room/{1}/notificationu
text/plainu Content-typeu Bearer u
Authorizationu POSTR R+ R, R
R- u /code N( R R R R R R. R/ R0 t
string_format( R R R R" R# R$ R% R& ( ( s8 /usr/lib/python2.7/site-packages/salt/engines/hipchat.pyt _publish_code_message� s,
u saltu !u salt/engines/hipchat/incomingu filec * @ s d } t j d � d k r= t j j j t t d � j � n d � � f d � � � � f d � } | pm | } | s� t d � � n t t j
j t � j � } | s� t
} n t j | d | �} | s� t d � � n t j d � i } |
d k r t }
|
| d
<n |
d k r$|
| d
<n | j | � d } x' | D] } | d
| k r>| } q>q>W| s�t j d | � t j d � t d j | � � � n | j d d � d d d } xVt ry | j d | � d } Wn' t j j k
rt j d � q�n X| d d } x�| | d | � D]�\ } } | sLt j d � d S| r{| | k r{| j d j | � � d Sn g } i } t j j j | � } | d } | rt | t � r| | j � k r| | j d � } t j j j | � } | d } n t! | � d k rjxS | d D]D } d | k rV| j" d d � \ } } | | | <q | j# | � q Wn d | k rd } n | d } | d =d | k r�d }! n | d }! | d =d | k r�| d }
| d =n d! | k r�| d! } | d! =n | r%| | k r%| j d" j | � � d Sn i }" | | k rgt j
j$ t � }# |# j% | d# | d$ | �}" n? t j& j' � }$ |$ j% d% j | � | | | d& d% j |! � �}" | r�d' d' j( | � n d( }% | r�d' d' j( d) � t) j* | � D� � n d( }& d* j | | |% |& | � }' | d+ k rCt+ | | |" d, |' d- |
d. | �q)| d/ k rtt, | | |" d, |' d- |
d. | �q)t j j- j. � }( t j j- j/ |( d0 � �) }) t j j0 j1 |" |) d1 t d2 d3 �Wd QXt2 | | |( d, |' d. | �t j j- j3 |( � q)Wt j | p t4 � q�Wd S(4 u�
Listen to Hipchat messages and forward them to Salt.
token
The HipChat API key. It requires a key for global usgae,
assigned per user, rather than room.
room
The HipChat room name.
aliases
Define custom aliases.
valid_users
Restrict access only to certain users.
valid_commands
Restrict the execution to a limited set of commands.
control
Send commands to the master.
trigger: ``!``
Special character that triggers the execution of salt commands.
tag: ``salt/engines/hipchat/incoming``
The event tag on the Salt bus.
api_url: ``https://api.hipchat.com``
The URL to the HipChat API.
.. versionadded:: 2017.7.0
max_rooms: ``1000``
Maximum number of rooms allowed to fetch. If set to 0,
it is able to retrieve the entire list of rooms.
wait_time: ``5``
Maximum wait time, in seconds.
output_type: ``file``
The type of the output. Choose bewteen:
- ``file``: save the output into a temporary file and upload
- ``html``: send the output as HTML
- ``code``: send the output as code
This can be overridden when executing a command, using the ``--out-type`` argument.
.. versionadded:: 2017.7.0
outputter: ``nested``
The format to display the data, using the outputters available on the CLI.
This argument can also be overridden when executing a command, using the ``--out`` option.
.. versionadded:: 2017.7.0
HipChat Example:
.. code-block:: text
! test.ping
! test.ping target=minion1
! test.ping --out=nested
! test.ping --out-type=code --out=table
u __roleu masteru sock_dirc @ s+ � r � | | � n t d | | � d S( u(
fire event to salt bus
u
event.sendN( t __salt__( t tagR' ( t fire_master( s8 /usr/lib/python2.7/site-packages/salt/engines/hipchat.pyt fire s c 3@ si xb | D]Z } | d } | j | � r � � | � | j | d � j � } | d d | f Vq q Wd S( u yield partner message u messageu u fromu mention_nameN( t
startswitht replacet strip( t all_messagest triggerR" t message_textt text( R8 R6 ( s8 /usr/lib/python2.7/site-packages/salt/engines/hipchat.pyt _eval_bot_mentions s
u Hipchat token not foundt endpointu Unable to connect to hipchatu Connected to Hipchatu max_resultsi u itemsu nameu Unable to connect to room %si u Unable to connect to room {0}t
maxResultsi u idt
not_beforei i����u! Engine not configured for controlNu' {0} not authorized to run Salt commandsu cmdu =u targetu *u tgt_typeu globu --outu
--out-typeu Using {0} is not allowed.t argt kwargu {0}t tgt_typeu u c s@ s' | ] \ } } d j | | � Vq d S( u {0}={1}N( R ( t .0t keyt value( ( s8 /usr/lib/python2.7/site-packages/salt/engines/hipchat.pys <genexpr>� s u" @{0} Results for: {1}{2}{3} on {4}u htmlR" R# R$ u codeu w+t sort_keyst indenti (5 R R/ t getR R t eventt get_master_eventt
fire_eventt UserWarningt sortedt runnert Runnert functionst _DEFAULT_API_URLR t HypChatt logt debugt _DEFAULT_MAX_ROOMSt roomst timet sleepR t latestR. t requestst HttpServiceUnavailableR" t argst shlex_splitt
isinstancet dictt keysR t splitt appendt RunnerClientt cmdt clientt LocalClientt joinR t iteritemsR2 R4 R t mkstempR R t dumpR* t safe_rmt _DEFAULT_SLEEP(* R R t aliasest valid_userst valid_commandst controlR= R6 t api_keyR$ t max_roomst wait_timet output_typeR# t target_roomR@ t runner_functionst hipct rooms_kwargst all_roomst a_roomt after_message_idt new_messagest partnerR? R` t kwargst cmdlineRh t itemRH RI t targetRF t retRR t localt nice_argst nice_kwargst message_stringt tmp_path_fnt fp_( ( R8 R7 R6 s8 /usr/lib/python2.7/site-packages/salt/engines/hipchat.pyt start� s� P
'
0 2 % %( (% t __doc__t
__future__R R R t loggingR[ R R t ImportErrorR t salt.utils.argsR t salt.utils.eventt salt.utils.filest salt.utils.httpt salt.utils.jsont salt.utils.stringutilst salt.runnert salt.clientt salt.loadert salt.outputt salt.extR t getLoggert __name__RW RU Rp RY R R R* R2 R4 R R� ( ( ( s8 /usr/lib/python2.7/site-packages/salt/engines/hipchat.pyt <module>$ sR
3
Mini Shell