def test__emailActionAgent():
rssClient = ResourceStatusClient()
# clean up
res = rssClient.delete('ResourceStatusCache')
assert res['OK'] is True
res = action.run()
assert res['OK'] is True
res = agent.execute()
assert res['OK'] is True
class EmailAgent(AgentModule):
def __init__(self, *args, **kwargs):
AgentModule.__init__(self, *args, **kwargs)
self.diracAdmin = None
self.default_value = None
self.rsClient = ResourceStatusClient()
def initialize(self, *args, **kwargs):
"""EmailAgent initialization"""
self.diracAdmin = DiracAdmin()
return S_OK()
@staticmethod
def _groupBySiteName(result):
"""
Group results by SiteName
"""
siteNameCol = result["Columns"].index("SiteName")
resultValue = result["Value"]
siteNameDict = {}
for row in resultValue:
if row[siteNameCol] not in siteNameDict:
siteNameDict[row[siteNameCol]] = [row]
else:
siteNameDict[row[siteNameCol]].append(row)
return siteNameDict
def execute(self):
result = self.rsClient.select("ResourceStatusCache")
if not result["OK"]:
return S_ERROR()
columnNames = result["Columns"]
result = self._groupBySiteName(result)
for site, records in result.items():
email = ""
html_body = ""
html_elements = ""
if gConfig.getValue("/DIRAC/Setup"):
setup = "(" + gConfig.getValue("/DIRAC/Setup") + ")\n\n"
else:
setup = ""
html_header = """\
<!DOCTYPE html>
<html>
<head>
<meta charset='UTF-8'>
<style>
table{{color:#333;font-family:Helvetica,Arial,sans-serif;min-width:700px;border-collapse:collapse;border-spacing:0}}
td,th{{border:1px solid transparent;height:30px;transition:all .3s}}th{{background:#DFDFDF;font-weight:700}}
td{{background:#FAFAFA;text-align:center}}.setup{{font-size:150%;color:grey}}.Banned{{color:red}}.Error{{color:#8b0000}}
.Degraded{{color:gray}}.Probing{{color:#00f}}.Active{{color:green}}tr:nth-child(even)
td{{background:#F1F1F1}}tr:nth-child(odd)
td{{background:#FEFEFE}}tr td:hover{{background:#666;color:#FFF}}
</style>
</head>
<body>
<p class="setup">{setup}</p>
""".format(
setup=setup
)
for row in records:
statusType = row[columnNames.index("StatusType")]
resourceName = row[columnNames.index("ResourceName")]
status = row[columnNames.index("Status")]
time = row[columnNames.index("Time")]
previousStatus = row[columnNames.index("PreviousStatus")]
html_elements += (
"<tr>"
+ "<td>"
+ statusType
+ "</td>"
+ "<td>"
+ resourceName
+ "</td>"
+ "<td class='"
+ status
+ "'>"
+ status
+ "</td>"
+ "<td>"
+ str(time)
+ "</td>"
+ "<td class='"
+ previousStatus
+ "'>"
+ previousStatus
+ "</td>"
+ "</tr>"
)
html_body = """\
<table>
<tr>
<th>Status Type</th>
<th>Resource Name</th>
<th>Status</th>
<th>Time</th>
<th>Previous Status</th>
</tr>
{html_elements}
</table>
</body>
</html>
""".format(
html_elements=html_elements
)
email = html_header + html_body
subject = "RSS actions taken for " + str(site) + "\n"
self._sendMail(subject, email, html=True)
self.rsClient.delete("ResourceStatusCache")
return S_OK()
def _sendMail(self, subject, body, html=False):
userEmails = self._getUserEmails()
if not userEmails["OK"]:
return userEmails
# User email address used to send the emails from.
fromAddress = RssConfiguration.RssConfiguration().getConfigFromAddress()
for user in userEmails["Value"]:
# FIXME: should not I get the info from the RSS User cache ?
resEmail = self.diracAdmin.sendMail(user, subject, body, fromAddress=fromAddress, html=html)
if not resEmail["OK"]:
return S_ERROR('Cannot send email to user "%s"' % user)
return S_OK()
def _getUserEmails(self):
configResult = RssConfiguration.getnotificationGroups()
if not configResult["OK"]:
return configResult
try:
notificationGroups = configResult["Value"]["notificationGroups"]
except KeyError:
return S_ERROR("%s/notificationGroups not found")
notifications = RssConfiguration.getNotifications()
if not notifications["OK"]:
return notifications
notifications = notifications["Value"]
userEmails = []
for notificationGroupName in notificationGroups:
try:
userEmails.extend(notifications[notificationGroupName]["users"])
except KeyError:
self.log.error("%s not present" % notificationGroupName)
return S_OK(userEmails)