main.py

import click
import oyaml as yaml
import requests
import urllib3

urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)

base_url = "{}.com/api/v4/projects/{}/repository/tags"


def data_yamls(path):
    with open(path, "r") as f:
        try:
            return yaml.safe_load(f)
        except yaml.YAMLError as exc:
            print(exc)
            return 2


def update_yaml(path, dictionary):
    with open(path, "w") as f:
        dump = yaml.dump(dictionary, default_flow_style=False)
        f.write(dump.replace('\n- ', '\n\n- '))


@click.command()
@click.option("--file", "yaml_file", prompt=True, help="Path to your yaml file")
@click.option("-v", "--verbose", is_flag=True, help="Enables verbose mode")
@click.option("-u", "--update", is_flag=True, help="Update the yamls to latest tags")
def tags(update, verbose, yaml_file):
    data = data_yamls(yaml_file)
    for i, entry in enumerate(data):
        link = entry["src"]
        git_tag = entry.get("version", None)

        base, project = link.split(".com/")
        project = project.replace(".git", "").replace("/", "%2F")
        repo_tags = requests.get(base_url.format(base, project), verify=False).json()

        # check the first tag as they are sorted in descending order
        # need to convert the tag to str to not write it as !!python/unicode
        latest_tag = str(repo_tags[0]["name"])
        if verbose and latest_tag == git_tag:
            print("\033[1;32;40m[INFO] OK for {} ({})\033[0m".format(link, git_tag))
        elif latest_tag != git_tag and git_tag is not None:
            print(
                "\033[1;31;40m[WARN] latest tag for {} is {} but you have {}\033[0m".format(
                    link, latest_tag, git_tag
                )
            )

            if update:
                data[i]["version"] = latest_tag
                if verbose:
                    print("\033[1;34;40m[INFO] updating tag for {} from {} to {}\033[0m".format(link, git_tag, latest_tag))
                update_yaml(yaml_file, data)
        elif git_tag is None:
            print("\033[1;32;40m[INFO] UNKNOWN for {}. Latest is {}\033[0m".format(link, latest_tag))
        else:
            # means latest_tag == git_tag but we don't need to output that
            continue


if __name__ == "__main__":
    tags()