コード例 #1
0
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
コード例 #2
0
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)