generate_codeowners.py

#!/usr/bin/env python

import sys
from collections import defaultdict
from pathlib import Path
from typing import Dict

import yaml
from yamale import YamaleError, make_data, make_schema, validate
from yamale.validators import DefaultValidators

OWNERS = Path("owners.yaml")
OWNERS_SCHEMA = Path("owners_schema.yaml")


def check_owners_file():
    for f in sys.argv[1:]:
        if f == str(OWNERS):
            return False
    return True


def validate_schema():
    validators = DefaultValidators.copy()
    schema = make_schema(OWNERS_SCHEMA, validators=validators)
    config_data = make_data(OWNERS)
    try:
        validate(schema=schema, data=config_data, strict=True)
    except YamaleError as e:
        exit(
            f"Error validating config from {str(OWNERS)}. Errors: {','.join([str(result) for result in e.results])}"
        )


def main():
    if check_owners_file():
        return

    validate_schema()
    with OWNERS.open("r") as ows:
        try:
            owners_file = yaml.safe_load(ows)
        except yaml.YAMLError as exc:
            exit(str(exc))
        codeowners_root = ""
        codeowners = ""
        for team in owners_file["teams"]:
            group_name = team.get("group_name", "root")
            joined_owners = " ".join([f"@{o}" for o in team.get("members", [])])
            paths = ""
            for path in team.get("owned_code_paths", []):
                paths += f"{path} {joined_owners}\n"

            if group_name != "root":
                codeowners += f"\n[{group_name}]\n"
                codeowners += paths
            else:
                codeowners_root += paths

        with Path("CODEOWNERS").open("w") as owners:
            owners.write(codeowners_root + "\n")
            owners.write(codeowners)


if __name__ == "__main__":
    main()