164 lines
4.4 KiB
Python
164 lines
4.4 KiB
Python
#!/usr/bin/python
|
|
|
|
DOCUMENTATION = '''
|
|
|
|
module: pagerduty
|
|
short_description: Create PagerDuty maintenance windows
|
|
description:
|
|
- This module will let you create PagerDuty maintenance windows
|
|
version_added: "1.2"
|
|
author: Justin Johns
|
|
requirements:
|
|
- PagerDuty API access
|
|
options:
|
|
state:
|
|
description:
|
|
- Create a maintenance window or get a list of ongoing windows.
|
|
required: true
|
|
default: null
|
|
choices: [ "running", "started", "ongoing" ]
|
|
aliases: []
|
|
name:
|
|
description:
|
|
- PagerDuty unique subdomain.
|
|
required: true
|
|
default: null
|
|
choices: []
|
|
aliases: []
|
|
user:
|
|
description:
|
|
- PagerDuty user ID.
|
|
required: true
|
|
default: null
|
|
choices: []
|
|
aliases: []
|
|
passwd:
|
|
description:
|
|
- PagerDuty user password.
|
|
required: true
|
|
default: null
|
|
choices: []
|
|
aliases: []
|
|
service:
|
|
description:
|
|
- PagerDuty service ID.
|
|
required: false
|
|
default: null
|
|
choices: []
|
|
aliases: []
|
|
hours:
|
|
description:
|
|
- Length of maintenance window in hours.
|
|
required: false
|
|
default: 1
|
|
choices: []
|
|
aliases: []
|
|
desc:
|
|
description:
|
|
- Short description of maintenance window.
|
|
required: false
|
|
default: Created by Ansible
|
|
choices: []
|
|
aliases: []
|
|
notes:
|
|
- This module does not yet have support to end maintenance windows.
|
|
'''
|
|
|
|
EXAMPLES='''
|
|
# List ongoing maintenance windows.
|
|
- pagerduty: name=companyabc user=example@example.com passwd=password123 state=ongoing
|
|
|
|
# Create a 1 hour maintenance window for service FOO123.
|
|
- pagerduty: name=companyabc
|
|
user=example@example.com
|
|
passwd=password123
|
|
state=running
|
|
service=FOO123
|
|
|
|
# Create a 4 hour maintenance window for service FOO123 with the description "deployment".
|
|
- pagerduty: name=companyabc
|
|
user=example@example.com
|
|
passwd=password123
|
|
state=running
|
|
service=FOO123
|
|
hours=4
|
|
desc=deployment
|
|
'''
|
|
|
|
import json
|
|
import datetime
|
|
import urllib2
|
|
import base64
|
|
|
|
|
|
def ongoing(name, user, passwd):
|
|
|
|
url = "https://" + name + ".pagerduty.com/api/v1/maintenance_windows/ongoing"
|
|
auth = base64.encodestring('%s:%s' % (user, passwd)).replace('\n', '')
|
|
|
|
req = urllib2.Request(url)
|
|
req.add_header("Authorization", "Basic %s" % auth)
|
|
res = urllib2.urlopen(req)
|
|
out = res.read()
|
|
|
|
return False, out
|
|
|
|
|
|
def create(name, user, passwd, service, hours, desc):
|
|
|
|
now = datetime.datetime.utcnow()
|
|
later = now + datetime.timedelta(hours=int(hours))
|
|
start = now.strftime("%Y-%m-%dT%H:%M:%SZ")
|
|
end = later.strftime("%Y-%m-%dT%H:%M:%SZ")
|
|
|
|
url = "https://" + name + ".pagerduty.com/api/v1/maintenance_windows"
|
|
auth = base64.encodestring('%s:%s' % (user, passwd)).replace('\n', '')
|
|
data = json.dumps({'maintenance_window': {'start_time': start, 'end_time': end, 'description': desc, 'service_ids': [service]}})
|
|
|
|
req = urllib2.Request(url, data)
|
|
req.add_header("Authorization", "Basic %s" % auth)
|
|
req.add_header('Content-Type', 'application/json')
|
|
res = urllib2.urlopen(req)
|
|
out = res.read()
|
|
|
|
return False, out
|
|
|
|
|
|
def main():
|
|
|
|
module = AnsibleModule(
|
|
argument_spec=dict(
|
|
state=dict(required=True, choices=['running', 'started', 'ongoing']),
|
|
name=dict(required=True),
|
|
user=dict(required=True),
|
|
passwd=dict(required=True),
|
|
service=dict(required=False),
|
|
hours=dict(default='1', required=False),
|
|
desc=dict(default='Created by Ansible', required=False)
|
|
)
|
|
)
|
|
|
|
state = module.params['state']
|
|
name = module.params['name']
|
|
user = module.params['user']
|
|
passwd = module.params['passwd']
|
|
service = module.params['service']
|
|
hours = module.params['hours']
|
|
desc = module.params['desc']
|
|
|
|
if state == "running" or state == "started":
|
|
if not service:
|
|
module.fail_json(msg="service not specified")
|
|
(rc, out) = create(name, user, passwd, service, hours, desc)
|
|
|
|
if state == "ongoing":
|
|
(rc, out) = ongoing(name, user, passwd)
|
|
|
|
if rc != 0:
|
|
module.fail_json(msg="failed", result=out)
|
|
|
|
module.exit_json(msg="success", result=out)
|
|
|
|
# include magic from lib/ansible/module_common.py
|
|
#<<INCLUDE_ANSIBLE_MODULE_COMMON>>
|
|
main()
|