update_backport_labels.py

# WARNING: DO NOT EDIT!
#
# This file was generated by plugin_template, and is managed by it. Please use
# './plugin-template --github pulp_rust' to update this file.
#
# For more info visit https://github.com/pulp/plugin_template

import os
import random

import requests
import yaml


def random_color():
    """Generates a random 24-bit number in hex"""
    color = random.randrange(0, 2**24)
    return format(color, "06x")


session = requests.Session()
token = os.getenv("GITHUB_TOKEN")

headers = {
    "Authorization": f"token {token}",
    "Accept": "application/vnd.github+json",
    "X-GitHub-Api-Version": "2022-11-28",
}
session.headers.update(headers)

# get all labels from the repository's current state
response = session.get("https://api.github.com/repos/pulp/pulp_rust/labels", headers=headers)
assert response.status_code == 200
old_labels = set([x["name"] for x in response.json() if x["name"].startswith("backport-")])

# get list of branches from template_config.yml
with open("./template_config.yml", "r") as f:
    plugin_template = yaml.safe_load(f)
branches = set(plugin_template["supported_release_branches"])
latest_release_branch = plugin_template["latest_release_branch"]
if latest_release_branch is not None:
    branches.add(latest_release_branch)
new_labels = {"backport-" + x for x in branches}

# delete old labels that are not in new labels
for label in old_labels.difference(new_labels):
    response = session.delete(
        f"https://api.github.com/repos/pulp/pulp_rust/labels/{label}", headers=headers
    )
    assert response.status_code == 204

# create new labels that are not in old labels
for label in new_labels.difference(old_labels):
    color = random_color()
    response = session.post(
        "https://api.github.com/repos/pulp/pulp_rust/labels",
        headers=headers,
        json={"name": label, "color": color},
    )
    assert response.status_code == 201