Ansible 开发Callback插件之【mail】

callback_plugins/test_mail.py

from __future__ import (absolute_import, division, print_function)
__metaclass__ = type

import os
import smtplib
import json

from ansible.plugins.callback import CallbackBase
from email import encoders
from email.header import Header
from email.mime.text import MIMEText
from email.utils import parseaddr, formataddr

def _format_addr(s):
name, addr = parseaddr(s)
return formataddr(( \
	Header(name, 'utf-8').encode(), \
	addr.encode('utf-8') if isinstance(addr, unicode) else addr))

def mail(subject='Ansible error mail', sender=None, to=None, body=None):

if sender is None:
	sender='<root>'
if to is None:
	to='to@test.com'
if body is None:
	body = subject

from_addr = 'from@test.com'

msg = MIMEText(body, 'plain', 'utf-8')
msg['From'] = _format_addr(u'%s <%s>' % sender,from_addr)
msg['To'] = _format_addr(u'管理员 <%s>' % to)
msg['Subject'] = Header(subject, 'utf-8').encode()

smtp = smtplib.SMTP('smtp.test.com', 25) 
smtp.login('user@test.com','test')
smtp.sendmail(from_addr, [to], msg.as_string())
smtp.quit()

class CallbackModule(CallbackBase):
"""
This Ansible callback plugin mails errors to interested parties.
"""
CALLBACK_VERSION = 2.0
CALLBACK_TYPE = 'notification'
CALLBACK_NAME = 'test_mail'
CALLBACK_NEEDS_WHITELIST = False

def v2_runner_on_failed(self, res, ignore_errors=False):

	host = res._host.get_name()

	if ignore_errors:
		return
	sender = '"Ansible: %s"' % host
	attach = res._task.action
	if 'invocation' in res._result:
		attach = "%s:  %s" % (res._result['invocation']['module_name'], json.dumps(res._result['invocation']['module_args']))

	subject = 'Failed: %s' % attach
	body = 'The following task failed for host ' + host + ':\n\n%s\n\n' % attach

	if 'stdout' in res._result.keys() and res._result['stdout']:
		subject = res._result['stdout'].strip('\r\n').split('\n')[-1]
		body += 'with the following output in standard output:\n\n' + res._result['stdout'] + '\n\n'
	if 'stderr' in res._result.keys() and res._result['stderr']:
		subject = res._result['stderr'].strip('\r\n').split('\n')[-1]
		body += 'with the following output in standard error:\n\n' + res._result['stderr'] + '\n\n'
	if 'msg' in res._result.keys() and res._result['msg']:
		subject = res._result['msg'].strip('\r\n').split('\n')[0]
		body += 'with the following message:\n\n' + res._result['msg'] + '\n\n'
	body += 'A complete dump of the error:\n\n' + self._dump_results(res._result)
	mail(sender=sender, subject=subject, body=body)

def v2_runner_on_unreachable(self, result):

	host = result._host.get_name()
	res = result._result

	sender = '"Ansible: %s" <root>' % host
	if isinstance(res, string_types):
		subject = 'Unreachable: %s' % res.strip('\r\n').split('\n')[-1]
		body = 'An error occurred for host ' + host + ' with the following message:\n\n' + res
	else:
		subject = 'Unreachable: %s' % res['msg'].strip('\r\n').split('\n')[0]
		body = 'An error occurred for host ' + host + ' with the following message:\n\n' + \
			   res['msg'] + '\n\nA complete dump of the error:\n\n' + str(res)
	mail(sender=sender, subject=subject, body=body)

def v2_runner_on_async_failed(self, result):

	host = result._host.get_name()
	res = result._result

	sender = '"Ansible: %s" <root>' % host
	if isinstance(res, string_types):
		subject = 'Async failure: %s' % res.strip('\r\n').split('\n')[-1]
		body = 'An error occurred for host ' + host + ' with the following message:\n\n' + res
	else:
		subject = 'Async failure: %s' % res['msg'].strip('\r\n').split('\n')[0]
		body = 'An error occurred for host ' + host + ' with the following message:\n\n' + \
			   res['msg'] + '\n\nA complete dump of the error:\n\n' + str(res)
	mail(sender=sender, subject=subject, body=body)

# change the default callback
stdout_callback = test_mail
# enable additional callbacks
callback_whitelist = test_mail