mirror of
https://github.com/ansible-collections/community.general.git
synced 2025-07-23 05:10:22 -07:00
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:
parent
6a615f18c5
commit
8748d625b2
1 changed files with 47 additions and 15 deletions
|
@ -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)
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue