Added support for setting up maintenance windows of under an hour.

Added support for using tokens instead of usernames and passwords.
This commit is contained in:
Andrew Newdigate 2014-03-05 22:22:56 +00:00
commit 8748d625b2

View file

@ -39,6 +39,13 @@ options:
default: null default: null
choices: [] choices: []
aliases: [] aliases: []
requester_id:
description:
- ID of user making the request. Only needed when using a token and creating a maintenance_window.
required: true
default: null
choices: []
aliases: []
service: service:
description: description:
- PagerDuty service ID. - PagerDuty service ID.
@ -53,6 +60,13 @@ options:
default: 1 default: 1
choices: [] choices: []
aliases: [] aliases: []
minutes:
description:
- Maintenance window in minutes (this is added to the hours).
required: false
default: 0
choices: []
aliases: []
desc: desc:
description: description:
- Short description of maintenance window. - Short description of maintenance window.
@ -90,34 +104,42 @@ import datetime
import urllib2 import urllib2
import base64 import base64
def create_req(url, data, name, user, passwd, token):
req = urllib2.Request(url, data)
if token:
req.add_header("Authorization", "Token token=%s" % token)
else:
auth = base64.encodestring('%s:%s' % (user, passwd)).replace('\n', '')
req.add_header("Authorization", "Basic %s" % auth)
def ongoing(name, user, passwd): return req
def ongoing(name, user, passwd, token):
url = "https://" + name + ".pagerduty.com/api/v1/maintenance_windows/ongoing" url = "https://" + name + ".pagerduty.com/api/v1/maintenance_windows/ongoing"
auth = base64.encodestring('%s:%s' % (user, passwd)).replace('\n', '') req = create_req(url, None, name, user, passwd, token)
req = urllib2.Request(url)
req.add_header("Authorization", "Basic %s" % auth)
res = urllib2.urlopen(req) res = urllib2.urlopen(req)
out = res.read() out = res.read()
return False, out return False, out
def create(name, user, passwd, service, hours, desc): def create(name, user, passwd, token, requester_id, service, hours, minutes, desc):
now = datetime.datetime.utcnow() now = datetime.datetime.utcnow()
later = now + datetime.timedelta(hours=int(hours)) later = now + datetime.timedelta(hours=int(hours), minutes=int(minutes))
start = now.strftime("%Y-%m-%dT%H:%M:%SZ") start = now.strftime("%Y-%m-%dT%H:%M:%SZ")
end = later.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" url = "https://" + name + ".pagerduty.com/api/v1/maintenance_windows"
auth = base64.encodestring('%s:%s' % (user, passwd)).replace('\n', '') request_data = {'maintenance_window': {'start_time': start, 'end_time': end, 'description': desc, 'service_ids': [service]}}
data = json.dumps({'maintenance_window': {'start_time': start, 'end_time': end, 'description': desc, 'service_ids': [service]}}) if requester_id:
request_data['requester_id'] = requester_id
req = urllib2.Request(url, data) data = json.dumps(request_data)
req.add_header("Authorization", "Basic %s" % auth)
req = create_req(url, data, name, user, passwd, token)
req.add_header('Content-Type', 'application/json') req.add_header('Content-Type', 'application/json')
res = urllib2.urlopen(req) res = urllib2.urlopen(req)
out = res.read() out = res.read()
@ -130,10 +152,13 @@ def main():
argument_spec=dict( argument_spec=dict(
state=dict(required=True, choices=['running', 'started', 'ongoing']), state=dict(required=True, choices=['running', 'started', 'ongoing']),
name=dict(required=True), name=dict(required=True),
user=dict(required=True), user=dict(required=False),
passwd=dict(required=True), passwd=dict(required=False),
token=dict(required=False),
service=dict(required=False), service=dict(required=False),
requester_id=dict(required=False),
hours=dict(default='1', required=False), hours=dict(default='1', required=False),
minutes=dict(default='0', required=False),
desc=dict(default='Created by Ansible', required=False) desc=dict(default='Created by Ansible', required=False)
) )
) )
@ -142,17 +167,24 @@ def main():
name = module.params['name'] name = module.params['name']
user = module.params['user'] user = module.params['user']
passwd = module.params['passwd'] passwd = module.params['passwd']
token = module.params['token']
service = module.params['service'] service = module.params['service']
hours = module.params['hours'] hours = module.params['hours']
minutes = module.params['minutes']
token = module.params['token']
desc = module.params['desc'] desc = module.params['desc']
requester_id = module.params['requester_id']
if not token and not (user or passwd):
module.fail_json(msg="neither user and passwd nor token specified")
if state == "running" or state == "started": if state == "running" or state == "started":
if not service: if not service:
module.fail_json(msg="service not specified") module.fail_json(msg="service not specified")
(rc, out) = create(name, user, passwd, service, hours, desc) (rc, out) = create(name, user, passwd, token, requester_id, service, hours, minutes, desc)
if state == "ongoing": if state == "ongoing":
(rc, out) = ongoing(name, user, passwd) (rc, out) = ongoing(name, user, passwd, token)
if rc != 0: if rc != 0:
module.fail_json(msg="failed", result=out) module.fail_json(msg="failed", result=out)