Current File : //proc/self/root/proc/3522530/root/lib/python2.7/site-packages/salt/modules/telemetry.pyo
�
���^c @@ s2 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
e j e � Z
y d d l Z e Z Wn e k
r� e Z n Xd Z d � Z d � Z d d d � Z d � Z d d
� Z d d d d � Z d d � Z d d
� Z d d d � Z d d d � Z d d d d d � Z d S( uf
Connection module for Telemetry
.. versionadded:: 2016.3.0
https://github.com/mongolab/mongolab-telemetry-api-docs/blob/master/alerts.md
:configuration: This module accepts explicit telemetry credentials or
can also read api key credentials from a pillar. More Information available
here__.
.. __: https://github.com/mongolab/mongolab-telemetry-api-docs/blob/master/alerts.md
In the minion's config file:
.. code-block:: yaml
telemetry.telemetry_api_keys:
- abc123 # Key 1
- efg321 # Backup Key 1
telemetry_api_base_url: https://telemetry-api.mongolab.com/v0
:depends: requests
i ( t absolute_importt unicode_literalst print_functionN( t sixu telemetryc C@ s t s
t St S( N( t HAS_REQUESTSt Falset __virtualname__( ( ( s: /usr/lib/python2.7/site-packages/salt/modules/telemetry.pyt __virtual__2 s c C@ s t d | � } | j d � S( Nu
config.optionu telemetry_api_base_url( t __salt__t get( t profilet config( ( s: /usr/lib/python2.7/site-packages/salt/modules/telemetry.pyt _get_telemetry_base9 s c C@ s� | d k r' | d k r' t d � � n | r� t | t j � rR t d | � } n t | t � rj | } n | r� | j d � d } q� t d � � n i | d 6d d 6S( Nu Missing api_key and profileu
config.optionu telemetry_api_keysi u Missing api_keyu Telemetry-API-Keyu application/jsonu content-type( t Nonet Exceptiont
isinstanceR t string_typesR t dictR ( t api_keyR
t _profile( ( s: /usr/lib/python2.7/site-packages/salt/modules/telemetry.pyt _auth> s c C@ sL d j | � } | t k r< t | } | | | <| t | <n t j | g � S( Nu telemetry.{0}.alerts( t formatt __context__R ( t
deployment_idt metric_namet alertt keyt alerts( ( s: /usr/lib/python2.7/site-packages/salt/modules/telemetry.pyt
_update_cacheP s
c C@ s� d } t d | � } | t k r� t | � d } t j | d | �} | j d k r� i } x? t | j � � D]+ \ } } | j d d � | | j d � <qn W| t | <q� n t | j | t � S( u�
Given an email address, checks the local
cache if corresponding email address: channel_id
mapping exists
email
Email escalation policy
profile
A dict of telemetry config information.
u telemetry.channelsR
u5 /notification-channels?_type=EmailNotificationChannelt headersi� u _idu falseu email( R R R t requestsR t status_codet enumeratet jsonR ( t emailR
R t autht get_urlt responset cache_resultt indexR ( ( s: /usr/lib/python2.7/site-packages/salt/modules/telemetry.pyt _retrieve_channel_id[ s #c C@ sQ t d | � } t } d j | � } | t k r� y2 t | � d j | � } t j | d | �} Wn0 t j j k
r� } t j
t j | � � t SXi }
| j
d k r� x@ | j � D]/ } | |
| j d i � j d � <|
t | <q� Wq� n t j | � sg St | j � } | r*t | j | � Sg | D] } d | k r1| d ^ q1S( u�
Get all alert definitions associated with a given deployment or if metric_name
is specified, obtain the specific alert config
Returns dictionary or list of dictionaries.
CLI Example:
salt myminion telemetry.get_alert_config rs-ds033197 currentConnections profile=telemetry
salt myminion telemetry.get_alert_config rs-ds033197 profile=telemetry
R
u telemetry.{0}.alertsu /alerts?deployment={0}R i� u conditionu metricu _id( R R R R R R R t
exceptionst RequestExceptiont logt errorR t text_typeR R! t values( R R R R
R# R R R$ R% t et http_resultR ( ( s: /usr/lib/python2.7/site-packages/salt/modules/telemetry.pyt get_alert_configw s*
c C@ s t d | � } t | � } | s t j d | � t | � d } i d d 6| | j d � d d 6| d 6} t j | d
t j j
j | � d | �} | j d k r� t j d
| j
d � | � | j
� j
d � } | t d | <q t d j | � � � n | S( u|
Given an email address, creates a notification-channels
if one is not found and also returns the corresponding
notification channel id.
notify_channel
Email escalation policy
profile
A dict of telemetry config information.
CLI Example:
salt myminion telemetry.get_notification_channel_id userx@company.com profile=telemetry
R
u$ %s channel does not exist, creating.u /notification-channelsu EmailNotificationChannelu _typeu @u EscalationPolicyu nameu emailt dataR i� uI Successfully created EscalationPolicy %s with EmailNotificationChannel %su _idu telemetry.channelsu* Failed to created notification channel {0}( R R( R+ t infoR t findR t postt saltt utilsR! t dumpsR R R R R ( t notify_channelR
R# t notification_channel_idt post_urlR2 R% ( ( s: /usr/lib/python2.7/site-packages/salt/modules/telemetry.pyt get_notification_channel_id� s"
* c C@ s� t d | � } y, t j t | � d j | � d | �} Wn0 t j j k
rm } t j t j
| � � t SX| j d k r� | j
� } t | � d k r� | Sd j | � Si | j d 6t j j
j | j � j d d � d
6Sd S( u�
get all the alarms set up against the current deployment
Returns dictionary of alarm information
CLI Example:
salt myminion telemetry.get_alarms rs-ds033197 profile=telemetry
R
u /alerts?deployment={0}R i� i u% No alarms defined for deployment: {0}u err_codeu erru u err_msgN( R R R R R R) R* R+ R, R R- R R R! t lenR6 R7 t loadst text( R R
R# R% R/ t alarms( ( s: /usr/lib/python2.7/site-packages/salt/modules/telemetry.pyt
get_alarms� s ,
c C@ s� t | | � } t | � d } d j | � } i | d 6| j d � d 6t | j d � � j � d 6i | d 6| j d � d 6| j d � d 6d
6} y. t j | d t j j
j | � d | �} Wn/ t j j
k
r� }
t j t j |
� � n X| j d
k r[| j d k r[t j d | | � t j d | | | j
� � t | | | j
� � n% t j d | | t j j
j | � � | j d
k o�| j d k | j
� f S( u�
create an telemetry alarms.
data is a dict of alert configuration data.
Returns (bool success, str message) tuple.
CLI Example:
salt myminion telemetry.create_alarm rs-ds033197 {} profile=telemetry
u /alertsu telemetry.{0}.alertsu
deploymentu filteru escalate_tou notificationChannelu metricu maxu minu conditionR2 R i� i, u- Created alarm on metric: %s in deployment: %su1 Updating cache for metric %s in deployment %s: %suB Failed to create alarm on metric: %s in deployment %s: payload: %s( R R R R R< t splitR R5 R6 R7 R! R8 R) R* R+ R, R R- R R3 t debugR ( R R R2 R R
R# t request_uriR t post_bodyR% R/ ( ( s: /usr/lib/python2.7/site-packages/salt/modules/telemetry.pyt create_alarm� s0 . c C@ s� t | | � } t | | | | � } | s@ t d j | | � f St | � d | d } i | d 6| j d � d 6t | j d � � j � d 6i | d 6| j d � d 6| j d
� d
6d 6} y. t j | d t
j j j
| � d
| �} Wn9 t j j k
r(}
t j d |
� t t j |
� f SX| j d k r�| j d k r�t j d | | | j � � t | | | j � � t j d | | � t | j � f St j d � j t
j j j | � t
j j j | � t
j j j
| � � } t j | � t | f S( u�
update an telemetry alarms. data is a dict of alert configuration data.
Returns (bool success, str message) tuple.
CLI Example:
salt myminion telemetry.update_alarm rs-ds033197 {} profile=telemetry
u6 No entity found matching deployment {0} and alarms {1}u /alerts/u _idu
deploymentu filteru escalate_tou notificationChannelu metricu maxu minu conditionR2 R u Update failed: %si� i, u1 Updating cache for metric %s in deployment %s: %su- Updated alarm on metric: %s in deployment: %suE Failed to create alarm on metric: {0} in deployment: {1} payload: {2}( R R1 R R R R R< RB R t putR6 R7 R! R8 R) R* R+ R, R R- R RC R R3 t Truet stringutilst
to_unicode( R R R2 R R
R# R RD RE R% R/ t err_msg( ( s: /usr/lib/python2.7/site-packages/salt/modules/telemetry.pyt update_alarm" s> .
c C@ st t d | � } | d
k r6 t | d | d | �} n | g } t | � d k rd t d j | � f Sg } x� | D]� } t | � d j | � } yE t j | d | �}
| r� t j
d | | � t | | d
� n Wn) t j j
k
r } t j d | � n X|
j d k rq | j | � qq qq Wt | � d k rUt d
j d j | � | � f St d j d j | � | � f S( u delete an alert specified by alert_id or if not specified blows away all the alerts
in the current deployment.
Returns (bool success, str message) tuple.
CLI Example:
salt myminion telemetry.delete_alarms rs-ds033197 profile=telemetry
R
R i u4 failed to find alert associated with deployment: {0}u /alerts/{0}R u( updating cache and delete %s key from %su Delete failed: %si� u. Failed to delete {0} alarms in deployment: {1}u , u2 Successfully deleted {0} alerts in deployment: {1}N( R R
R1 R= R R R R t deleteR+ RC R R) R* R, R t appendt joinRH ( R t alert_idR R R
R# t alert_idst failed_to_deletet idt
delete_urlR% R/ ( ( s: /usr/lib/python2.7/site-packages/salt/modules/telemetry.pyt
delete_alarmsZ s,
( t __doc__t
__future__R R R t loggingt salt.utils.jsonR6 t salt.utils.stringutilst salt.extR t getLoggert __name__R+ R RH R t ImportErrorR R R R R
R R R( R1 R< RA RF RL RU ( ( ( s: /usr/lib/python2.7/site-packages/salt/modules/telemetry.pyt <module> s.
+ , 58
Mini Shell