tools/plugins.py

import re
import requests

from pygerrit2 import GerritRestAPI, HTTPBasicAuthFromNetrc

url = "https://gerrit-review.googlesource.com"
auth = HTTPBasicAuthFromNetrc(url=url)
api = GerritRestAPI(url=url, auth=auth)
plugins = api.get("/projects/?p=plugins%2f&d")

header = "|Name|State|Changes (last 3 months)|Description"
dashes = "|----|-----|-------|-----------"
spacer = "|    |     |       |           "

branches = ["master"] + ["stable-%s" % version for version in ["3.1", "3.0", "2.16"]]

checkMark = "✔"
unicodeSquare = "⃞"
greenCheckMark = "✅"
lock = "🔒"
redCross = "❌"


def getBranches(pluginId):
    branchList = api.get("/projects/%s/branches/" % pluginId)
    pluginBranches = []
    for b in branchList:
        if b["ref"].startswith("refs/heads/"):
            ref = b["ref"]
            pluginBranches += [ref[len("refs/heads/") :]]
    return pluginBranches


def getBranchResults(pluginId, pluginName, builds):
    pluginBranches = getBranches(pluginId)
    results = ""
    for branch in branches:
        string = r"^plugin-%s[\w|-]*-%s$" % (pluginName, branch)
        pattern = re.compile(string)
        result = unicodeSquare

        for job in builds["jobs"]:
            if pattern.match(job["name"]):
                result = job["lastBuild"]["result"]
                result = greenCheckMark if result == "SUCCESS" else redCross
                break

        if branch in pluginBranches:
            result = checkMark + "|" + result
        else:
            result = "%s|%s" % (unicodeSquare, result)
        results = result if not results else "%s|%s" % (results, result)
    return results


def getRecentChangesCount(pluginName):
    changes = api.get("/changes/?q=project:%s+-age:3months" % pluginName)
    return len(changes)


page = "plugins"
with open("tools/template.md", "r") as template:
    with open("pages/site/plugins/%s.md" % page, "w") as output:
        for line in template:
            output.write(line.replace("template", page))

        output.write("\n")
        for b in branches:
            output.write(
                "[%s]: https://gerrit-ci.gerritforge.com/view/Plugins-%s/\n" % (b, b)
            )

        for branch in branches:
            header += "|Branch|CI"
            dashes += "|-----:|--"
            spacer += "|[%s]|" % branch
        output.write("\n%s|\n%s|\n%s|\n" % (header, dashes, spacer))

        url = (
            "https://gerrit-ci.gerritforge.com/api/json?pretty=true&tree=jobs"
            + "[name,lastBuild[result]]"
        )
        builds = requests.get(url).json()
        links = "\n"
        for p in plugins:
            name = p[len("plugins/") :]
            plugin = plugins[p]

            if plugin["state"] == "ACTIVE":
                state = greenCheckMark
                changes = getRecentChangesCount(p)
                availableBranches = getBranchResults(plugin["id"], name, builds)
            else:
                state = lock
                changes = 0
                availableBranches = "|".join(
                    ["%s|%s" % (unicodeSquare, unicodeSquare) for b in branches]
                )

            if "description" in plugin:
                description = plugin["description"].split("\n")[0].rstrip(r"\.")
            else:
                description = unicodeSquare

            line = "|[%s]|%s|%d|%s|%s|\n" % (
                name,
                state,
                changes,
                description,
                availableBranches,
            )
            output.write(line)
            links += "[%s]: https://gerrit.googlesource.com/plugins/%s\n" % (name, name)

        output.write(links)