mirror of https://github.com/zulip/zulip.git
87 lines
2.8 KiB
Python
87 lines
2.8 KiB
Python
|
from typing import Any, Dict, Iterable, Optional, List
|
||
|
|
|
||
|
|
from django.utils.translation import ugettext as _
|
||
|
|
from django.http import HttpRequest, HttpResponse
|
||
|
|
|
||
|
|
from zerver.lib.webhooks.common import check_send_webhook_message
|
||
|
|
from zerver.lib.response import json_success, json_error
|
||
|
|
from zerver.decorator import REQ, has_request_variables, api_key_only_webhook_view
|
||
|
|
from zerver.models import UserProfile
|
||
|
|
|
||
|
|
import ujson
|
||
|
|
import operator
|
||
|
|
|
||
|
|
ANSIBLETOWER_DEFAULT_MESSAGE_TEMPLATE = "{friendly_name}: [#{id} {name}]({url}) {status}\n"
|
||
|
|
|
||
|
|
|
||
|
|
ANSIBLETOWER_JOB_MESSAGE_TEMPLATE = ("{friendly_name}: [#{id} {name}]({url}) {status}\n"
|
||
|
|
"{hosts_final_data}")
|
||
|
|
|
||
|
|
ANSIBLETOWER_JOB_HOST_ROW_TEMPLATE = '* {hostname}: {status}\n'
|
||
|
|
|
||
|
|
@api_key_only_webhook_view('Ansibletower')
|
||
|
|
@has_request_variables
|
||
|
|
def api_ansibletower_webhook(request: HttpRequest, user_profile: UserProfile,
|
||
|
|
payload: Dict[str, Any]=REQ(argument_type='body')) -> HttpResponse:
|
||
|
|
|
||
|
|
body = get_body(payload)
|
||
|
|
subject = payload['name']
|
||
|
|
|
||
|
|
check_send_webhook_message(request, user_profile, subject, body)
|
||
|
|
return json_success()
|
||
|
|
|
||
|
|
def get_body(payload: Dict[str, Any]) -> str:
|
||
|
|
if (payload['friendly_name'] == 'Job'):
|
||
|
|
hosts_list_data = payload['hosts']
|
||
|
|
hosts_data = []
|
||
|
|
for host in payload['hosts']:
|
||
|
|
if (hosts_list_data[host].get('failed') is True):
|
||
|
|
hoststatus = 'Failed'
|
||
|
|
elif (hosts_list_data[host].get('failed') is False):
|
||
|
|
hoststatus = 'Success'
|
||
|
|
hosts_data.append({
|
||
|
|
'hostname': host,
|
||
|
|
'status': hoststatus
|
||
|
|
})
|
||
|
|
|
||
|
|
if (payload['status'] == "successful"):
|
||
|
|
status = 'was successful'
|
||
|
|
else:
|
||
|
|
status = 'failed'
|
||
|
|
|
||
|
|
return ANSIBLETOWER_JOB_MESSAGE_TEMPLATE.format(
|
||
|
|
name=payload['name'],
|
||
|
|
friendly_name=payload['friendly_name'],
|
||
|
|
id=payload['id'],
|
||
|
|
url=payload['url'],
|
||
|
|
status=status,
|
||
|
|
hosts_final_data=get_hosts_content(hosts_data)
|
||
|
|
)
|
||
|
|
|
||
|
|
else:
|
||
|
|
|
||
|
|
if (payload['status'] == "successful"):
|
||
|
|
status = 'was successful'
|
||
|
|
else:
|
||
|
|
status = 'failed'
|
||
|
|
|
||
|
|
data = {
|
||
|
|
"name": payload['name'],
|
||
|
|
"friendly_name": payload['friendly_name'],
|
||
|
|
"id": payload['id'],
|
||
|
|
"url": payload['url'],
|
||
|
|
"status": status
|
||
|
|
}
|
||
|
|
|
||
|
|
return ANSIBLETOWER_DEFAULT_MESSAGE_TEMPLATE.format(**data)
|
||
|
|
|
||
|
|
def get_hosts_content(hosts_data: List[Dict[str, Any]]) -> str:
|
||
|
|
hosts_data = sorted(hosts_data, key=operator.itemgetter('hostname'))
|
||
|
|
hosts_content = ''
|
||
|
|
for host in hosts_data:
|
||
|
|
hosts_content += ANSIBLETOWER_JOB_HOST_ROW_TEMPLATE.format(
|
||
|
|
hostname=host.get('hostname'),
|
||
|
|
status=host.get('status')
|
||
|
|
)
|
||
|
|
return hosts_content
|