Function main

ci/tools/gh/limit_agent_pr_workflows.py:178–212

(argv: list[str] | None = None)

def main(argv: list[str] | None = None) -> int:
    args = parse_args(sys.argv[1:] if argv is None else argv)
    allowlist = DEFAULT_PR_ALLOWLIST | set(args.allow_pr)

    if args.print_allowlist:
        for filename in sorted(allowlist):
            print(filename)
        return 0

    workflows_dir = args.workflows_dir
    if not workflows_dir.is_dir():
        print(f"Workflow directory not found: {workflows_dir}", file=sys.stderr)
        return 2

    results = [
        remove_pull_request_if_needed(path, allow_pr=path.name in allowlist)
        for path in workflow_files(workflows_dir)
    ]
    changed = [result for result in results if result.changed]

    for result in changed:
        print(result.path.as_posix())
        for change in result.changes:
            print(f"  - {change}")

    if args.apply:
        for result in changed:
            result.path.write_text(result.updated_text, encoding="utf-8")
        print(f"Updated {len(changed)} workflow file(s).")
    else:
        print(f"Would update {len(changed)} workflow file(s). Pass --apply to rewrite.")

    if args.check and changed:
        return 1
    return 0


if __name__ == "__main__":