__main__.py

#!/usr/bin/env python3

import sys
import shutil
import pathlib
import argparse

from trigger.generate import generate
from trigger.scheduler import scheduler
from trigger.utils import logger, compare

log = logger.setup()


def main():

    # Arguments
    parser = argparse.ArgumentParser(description="Run pipelines arguments")
    parser.add_argument(
        "--dry-run",
        action="store_true",
        help="Testing flag to dry run the script",
    )

    parser.add_argument(
        "--repo1-token",
        metavar="repo1-token",
        default="haha",
        type=str,
        help="Personal token for accessing gitlab.",
    )

    parser.add_argument(
        "--initial-batch",
        metavar="initial-batch",
        default=15,
        type=int,
        help="Initial batch size of pipelines to kick off.",
    )

    parser.add_argument(
        "--max-batch",
        metavar="max-batch",
        default=50,
        type=int,
        help="Maximum batch size of pipelines of created pipelines.",
    )

    parser.add_argument(
        "--repo1-url",
        metavar="repo1-url",
        default="https://repo1.dso.mil/",
        type=str,
        help="Url for Repo1.",
    )

    parser.add_argument(
        "--max-duration",
        metavar="max-duration",
        default=1200,
        type=int,
        help="Max duration for a pipeline to run before working set is scaled down.",
    )

    parser.add_argument(
        "--artifacts-path",
        metavar="artifacts-path",
        default=".artifacts",
        type=str,
        help="Directory to write all artifacts to.",
    )

    args = parser.parse_args()
    # End arguments

    if pathlib.Path(args.artifacts_path).exists():
        try:
            shutil.rmtree(args.artifacts_path)
        except OSError:
            log.exception("Could not remove the {args.artifacts_path}")
            sys.exit(1)

    pathlib.Path(args.artifacts_path).mkdir(parents=True, exist_ok=True)

    projects = generate.list_of_projects(
        repo1_url=args.repo1_url,
        repo1_token=args.repo1_token,
        artifacts_path=args.artifacts_path,
        group="dsop",
        dry_run=args.dry_run,
    )

    s = scheduler.Scheduler(
        repo1_token=args.repo1_token,
        repo1_url=args.repo1_url,
        initial_batch=args.initial_batch,
        max_batch=args.max_batch,
        max_duration=args.max_duration,
    )

    s.schedule(projects=projects, artifacts_path=args.artifacts_path, branch="master")

    #
    # Removing the development rebuilds for now. Please reference
    #   https://repo1.dso.mil/ironbank-tools/trigger/-/issues/18
    #

    compare.compare(filename=args.artifacts_path)
    return 0


if __name__ == "__main__":
    sys.exit(main())