Current File : //usr/lib/python2.7/site-packages/salt/modules/pagerduty_util.pyc
�
���^c @@ s4 d Z d d l m Z m Z m Z d d l Z d d l Z d � Z d d d d � Z
d d d d � Z d d d d � Z d d d d � Z
d d d d
� Z d d d d d d d d d d e d
� Z d � Z d d d d � Z d d d d d � Z d d d d � Z d d d d d � Z d d d d � Z d S( u�
Module for manageing PagerDuty resource
:configuration: This module can be used by specifying the name of a
configuration profile in the minion config, minion pillar, or master
config. The default configuration profile name is 'pagerduty.'
For example:
.. code-block:: yaml
pagerduty:
pagerduty.api_key: F3Rbyjbve43rfFWf2214
pagerduty.subdomain: mysubdomain
For PagerDuty API details, see https://developer.pagerduty.com/documentation/rest
i ( t absolute_importt print_functiont unicode_literalsNc C@ s t S( u>
No dependencies outside of what Salt itself requires
( t True( ( ( s? /usr/lib/python2.7/site-packages/salt/modules/pagerduty_util.pyt __virtual__ s u pagerdutyc C@ s t d d d | d | d | �S( uk
List users belonging to this account
CLI Example:
salt myminion pagerduty.get_users
u usersu idt profilet subdomaint api_key( t _list_items( R R R ( ( s? /usr/lib/python2.7/site-packages/salt/modules/pagerduty_util.pyt get_users" s c C@ s t d d d | d | d | �S( uq
List services belonging to this account
CLI Example:
salt myminion pagerduty.get_services
u servicesu idR R R ( R ( R R R ( ( s? /usr/lib/python2.7/site-packages/salt/modules/pagerduty_util.pyt get_services4 s c C@ s t d d d | d | d | �S( us
List schedules belonging to this account
CLI Example:
salt myminion pagerduty.get_schedules
u schedulesu idR R R ( R ( R R R ( ( s? /usr/lib/python2.7/site-packages/salt/modules/pagerduty_util.pyt
get_schedulesF s c C@ s t d d d | d | d | �S( u�
List escalation_policies belonging to this account
CLI Example:
salt myminion pagerduty.get_escalation_policies
u escalation_policiesu idR R R ( R ( R R R ( ( s? /usr/lib/python2.7/site-packages/salt/modules/pagerduty_util.pyt get_escalation_policiesX s c C@ sN t d | d | d | d | � } i } x | | D] } | | | | <q2 W| S( u]
List items belonging to an API call.
This method should be in utils.pagerduty.
R R R t action( t _query( R
t keyR R R t itemst rett item( ( s? /usr/lib/python2.7/site-packages/salt/modules/pagerduty_util.pyR j s u GETu api/v1c C@ s$ | r t d | � } n i | d 6| d 6} | d k rU d j | d | | � } n | d k rj i } n | d k r i } n i d j | d � d 6} | d k r� d | d <n t j | | d
| d | d t j j j | � d
|
�}
|
j d k s |
j d k r
d S|
j � } d | k r d | k r d | k r | d } | d } | d } x� | | | k r | | } d } | | d <| | d <t j | | d
| d | d | d
|
�j � } | d } | d } x@ | j
� D]2 \ } } t | t � r�| | c | | 7<q�q�Wq^Wn | S( uR
Query the PagerDuty API.
This method should be in utils.pagerduty.
u
config.optionu pagerduty.api_keyu pagerduty.subdomainu! https://{0}.pagerduty.com/{1}/{2}u Token token={0}u
Authorizationu GETu application/jsonu Content-typet headerst paramst datat verifyu u totalu offsetu limitid N(
t __salt__t Nonet formatt requestst requestt saltt utilst jsont dumpst textR t
isinstancet list( t methodR t urlt pathR
R t serviceR R R t
verify_sslt credsR t resultt result_jsont offsett limitt totalt next_page_resultst kt v( ( s? /usr/lib/python2.7/site-packages/salt/modules/pagerduty_util.pyR | sb
$
"c C@ s0 d | k r | d Sd | k r, | d d Sd S( uf
helper method to find the resource id, since PD API doesn't always return it in the same way
u idu scheduleN( R ( t resource( ( s? /usr/lib/python2.7/site-packages/salt/modules/pagerduty_util.pyt _get_resource_id� s
c
C@ s� d t k r i t d <n | t d k rx | d k rB | d } n | } t d | d | d | d | � | t d | <n x| t d | D]l } xc | D][ } | | | k r� | d k r� t d d j | | d
� d | d | d | � } | S| Sq� Wq� Wd S( um
Get any single pagerduty resource by key.
We allow flexible lookup by any of a list of identifier_fields.
So, for example, you can look up users by email address or name by calling:
get_resource('users', key, ['name', 'email'], ...)
This method is mainly used to translate state sls into pagerduty id's for dependent objects.
For example, a pagerduty escalation policy contains one or more schedules, which must be passed
by their pagerduty id. We look up the schedules by name (using this method), and then translate
the names into id's.
This method is implemented by getting all objects of the resource type (cached into __context__),
then brute force searching through the list and trying to match any of the identifier_fields.
The __context__ cache is purged after any create, update or delete to the resource.
u pagerduty_util.resource_cacheu servicesu ?include[]=escalation_policyR
R R R u schedulesu {0}/{1}u idN( t __context__R R R (
t
resource_nameR t identifier_fieldsR R R R
R1 t fieldt full_resource_info( ( s? /usr/lib/python2.7/site-packages/salt/modules/pagerduty_util.pyt get_resource� s*
c
C@ s� d } xL | D]D } | | k r
t | | | | | | | � } | d k rQ PqQ q
q
W| d k r� t d ro d St d | =t d d d | d | d | d | d
| � Si } | r� | | | � } ni xf | j � D]X \ }
} |
j d � r� q� n | j |
d � } | d k r� | | k r� | | |
<q� q� Wt | � d k r�t d rLd
St d | =t | � }
t d d d d j
| |
� d | d | d | d
| � St Sd S( u/
create or update any pagerduty resource
Helper method for present().
Determining if two resources are the same is different for different PD resource, so this method accepts a diff function.
The diff function will be invoked as diff(state_information, object_returned_from_pagerduty), and
should return a dict of data to pass to the PagerDuty update API method, or None if no update
is to be performed. If no diff method is provided, the default behavor is to scan the keys in the state_information,
comparing the matching values in the object_returned_from_pagerduty, and update any values that differ.
examples:
create_or_update_resource("user", ["id","name","email"])
create_or_update_resource("escalation_policies", ["id","name"], diff=my_diff_function)
u testu would createu pagerduty_util.resource_cacheR# u POSTR
R R R R u _i u would updateu PUTu {0}/{1}N( R R8 t __opts__R3 R R t
startswitht gett lenR2 R R ( R4 R5 R t diffR R R R1 R6 t data_to_updateR/ R0 t resource_valuet resource_id( ( s? /usr/lib/python2.7/site-packages/salt/modules/pagerduty_util.pyt create_or_update_resource� s8
+
$c C@ s t | | | | | | � } | rw t d r/ d St d | =t | � } t d d d d j | | � d | d | d
| � St Sd S( u�
delete any pagerduty resource
Helper method for absent()
example:
delete_resource("users", key, ["id","name","email"]) # delete by id or name or email
u testu would deleteu pagerduty_util.resource_cacheR# u DELETER
u {0}/{1}R R R N( R8 R9 R3 R2 R R R ( R4 R R5 R R R R1 R@ ( ( s? /usr/lib/python2.7/site-packages/salt/modules/pagerduty_util.pyt delete_resource8 s
1c K@ s� i | d d 6i d 6d d 6d d 6} t | | | d | d | d | d | �} | t k r_ ng | d k rx t | d <nN t d
r� | | d <n7 d | k r� t | d <| | d <n t | d <| | d <| S(
uI
Generic resource.present state method. Pagerduty state modules should be a thin wrapper over this method,
with a custom diff function.
This method calls create_or_update_resource() and formats the result as a salt state return value.
example:
resource_present("users", ["id","name","email"])
u nameu changesu resultu u commentR= R R R u testu errorN( R RA R R9 t False( R1 R5 R= R R R t kwargsR R) ( ( s? /usr/lib/python2.7/site-packages/salt/modules/pagerduty_util.pyt resource_presentN s.
c
K@ s� i | d d 6i d 6d d 6d d 6} x� | j � D]� \ } } | | k rQ q3 n t | | | d | d | d | �} | d k r� t | d <d j | � | d <| S| t k r� q3 q3 t d
r� | | d <| Sd | k r3 t | d <| | d <| Sq3 W| S(
u=
Generic resource.absent state method. Pagerduty state modules should be a thin wrapper over this method,
with a custom diff function.
This method calls delete_resource() and formats the result as a salt state return value.
example:
resource_absent("users", ["id","name","email"])
u nameu changesu resultu u commentR R R u {0} deletedu testu errorN( R R RB R R R9 RC (
R1 R5 R R R RD R R/ R0 R) ( ( s? /usr/lib/python2.7/site-packages/salt/modules/pagerduty_util.pyt resource_absents s6
( t __doc__t
__future__R R R R t salt.utils.jsonR R R R R
R R R R R R2 R8 RA RB RE RF ( ( ( s? /usr/lib/python2.7/site-packages/salt/modules/pagerduty_util.pyt <module> s$ I -9%
Mini Shell