PatchTeams.py

#!/usr/bin/env python3

# PatchTeams.py v1.0
# Tony Williams 25/07/2019
#

"""See docstring for PatchTeams class"""

import json
import plistlib
import os.path as path
import datetime
import logging
import logging.handlers
import sys
import requests

__all__ = ["PatchTeams"]

# logging requirements
LOGFILE = "PatchTeams.log"
LOGLEVEL = logging.DEBUG


class PatchTeams:
    """When given the location of an output plist from Autopkg parses it
    and sends the details on packages productionized to Jamf Pro to Teams
    """

    description = __doc__

    def __init__(self):

        # extremely dumb command line processing
        try:
            self.plist = sys.argv[1]
        except IndexError:
            self.plist = "autopkg.plist"

        # URL of Teams webhook
        self.url = "https://outlook.office.com/webhook/"
        # token
        self.url += "76ea46bf-3dda-41f0-831d-b0dc655e4f97@43f93f8a-55a8-4263-bd84"
        self.url += "-e03688a2ab2d/IncomingWebhook/0ac15911fcfa42deb"
        self.url += "1d07f0672950542/63a48cfb-c3ef-4ee9-be63-fafbe4177f30"

        # set up logging
        now = datetime.datetime.now().strftime("%d/%m/%Y %H:%M")
        frmt = "%(levelname)s {} %(message)s".format(now)
        # set up logging
        logging.basicConfig(filename=LOGFILE, level=LOGLEVEL, format=frmt)
        self.logger = logging.getLogger("")
        # set logging formatting
        # ch = logging.StreamHandler()
        ch = logging.handlers.TimedRotatingFileHandler(
            LOGFILE, when="D", interval=7, backupCount=4
        )
        ch.setFormatter(logging.Formatter(frmt))
        self.logger.addHandler(ch)
        self.logger.setLevel(LOGLEVEL)

        # JSON for the message to Teams
        # "sections" will be replaced by our work
        self.template = """
        {
            "@context": "https://schema.org/extensions",
            "@type": "MessageCard",
            "themeColor": "0072C6",
            "title": "Patch Manager",
            "text": "Patches to testing",
            "sections": [
            ]
        }
        """

        # JSON for a section of a message
        # we will have a section for each package uploaded
        # in this Autopkg run
        # (It looks ugly to keep pylint happy)
        self.section = """
        {"startGoup": "true", "patch_id": "**AppName**", "text": "version"
        }
        """

        # JSON template for the error message card.
        self.err_template = """
        {
            "@context": "https://schema.org/extensions",
            "@type": "MessageCard",
            "themeColor": "0072C6",
            "title": "Autopkg",
            "text": "Patch management errors",
            "sections": [
            ]
        }
        """

        # JSON template for a single error on error card.
        self.err_section = """
        {
            "text": "A long message",
            "startGoup": "true",
            "title": "**Firefox.pkg**"
        }
        """

        # JSON template for the error message card.
        self.none_template = """
        {
            "@context": "https://schema.org/extensions",
            "@type": "MessageCard",
            "themeColor": "0072C6",
            "title": "PatchManager",
            "text": "**Empty Run**"
        }
        """

    def PatchTeams(self):
        """Do the patches changed!"""
        self.logger.info("Starting Run")
        sections = []
        empty = False
        jsr = "patch_manager_summary_result"
        try:
            fp = open(self.plist, "rb")
            pl = plistlib.load(fp)
        except IOError:
            self.logger.error("Failed to load %s", self.plist)
            sys.exit()
        item = 0
        if jsr not in pl["summary_results"]:
            self.logger.debug("No Patch results")
            empty = True
        else:
            pkgs = pl["summary_results"][jsr]["data_rows"]
            for p in pkgs:
                sections.append(json.loads(self.section))
                name = p["title"]
                version = p["version"]
                self.logger.debug("Version: %s Name: %s", version, name)
                sections[item]["title"] = "**%s**" % name
                sections[item]["text"] = version
                item = item + 1
            j = json.loads(self.template)
            j["sections"] = sections
            d = json.dumps(j)
            requests.post(self.url, data=d)
        # do the error messages
        fails = pl["failures"]
        if len(fails) == 0:  # no failures
            if empty:
                requests.post(self.url, self.none_template)
            sys.exit()
        sections = []
        item = 0
        for f in fails:
            sections.append(json.loads(self.err_section))
            sections[item]["title"] = "**%s**" % f["recipe"]
            sections[item]["text"] = f["message"].replace("\n", " ")
            item = item + 1
        j = json.loads(self.err_template)
        j["sections"] = sections
        d = json.dumps(j)
        requests.post(self.url, d)


if __name__ == "__main__":
    PatchTeams = PatchTeams()
    PatchTeams.PatchTeams()