def main()

in bot/tools/covdir_gen.py [0:0]

• 64 lines of code
• 13 McCabe index (conditional complexity)

def main():
    # CLI args
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--channel", type=str, default="production", help="Which channel to use"
    )
    parser.add_argument("--nb-tasks", type=int, default=5, help="NB of tasks to create")
    parser.add_argument(
        "--group", type=str, default=slugId(), help="Task group to create/update"
    )
    parser.add_argument(
        "--dry-run",
        action="store_true",
        default=False,
        help="List actions without triggering any new task",
    )
    parser.add_argument(
        "history", type=open, help="JSON payload of /v2/history endpoint"
    )
    args = parser.parse_args()

    # Setup Taskcluster
    taskcluster_config.auth()

    # List existing tags & commits
    print("Group", args.group)
    queue = taskcluster_config.get_service("queue")
    try:
        group = queue.listTaskGroup(args.group)
        commits = [
            task["task"]["payload"]["env"]["REVISION"]
            for task in group["tasks"]
            if task["status"]["state"] not in ("failed", "exception")
        ]
        print(
            "Found {} commits processed in task group {}".format(
                len(commits), args.group
            )
        )
    except Exception as e:
        print("Invalid task group : {}".format(e))
        commits = []

    # Read the history file
    history = json.load(args.history)

    # Load initial dates from our history
    history_dates = {
        item["changeset"]: datetime.fromtimestamp(item["date"]).date()
        for item in history
    }
    dates = [history_dates[commit] for commit in commits if commit in history_dates]

    # Trigger a task for each commit
    nb = 0
    for commit in history:
        date = datetime.fromtimestamp(commit["date"])
        if nb >= args.nb_tasks:
            break
        if commit["changeset"] in commits:
            print(
                f"Skipping commit {commit['changeset']} from {date} : already processed"
            )
            continue

        if date.date() in dates:
            print(f"Skipping commit {commit['changeset']} from {date} : same day")
            continue

        print(f"Triggering commit {commit['changeset']} from {date}")
        if args.dry_run:
            print(">>> No trigger on dry run")
        else:
            out = trigger_task(args.app_channel, args.group, commit)
            print(">>>", out["status"]["taskId"])
        nb += 1
        dates.append(date.date())