Convert the zuul migration script into enrollment script

scripts/enroll-users.py

@@ -0,0 +1,76 @@
+#!/usr/bin/env python3
+
+# Copyright Contributors to the Packit project.
+# SPDX-License-Identifier: MIT
+
+# /// script
+# dependencies = [
+#   "ruamel.yaml",
+#   "requests",
+# ]
+# ///
+
+import argparse
+import copy
+
+import requests
+import ruamel.yaml
+from pathlib import Path
+
+# Constants
+SCRIPTS_DIR = Path(__file__).parent
+ROOT_DIR = SCRIPTS_DIR.parent
+packit_service_file = ROOT_DIR / "secrets/packit/prod/packit-service.yaml.j2"
+SKIP_JINJA_LINES = 32
+DIST_GIT_FORMAT = r"https://src.fedoraproject.org/rpms/{}"
+PAGURE_BZ = "https://src.fedoraproject.org/extras/pagure_bz.json"
+
+parser = argparse.ArgumentParser(
+    description="Provide a comma-separated list of FAS maintainers/groups to bulk add to the packit-service.yaml.j2.",
+)
+parser.add_argument(
+    "maintainers",
+    type=str,
+    nargs="+",
+    help="a comma-separated list of FAS maintainers",
+)
+args = parser.parse_args()
+maintainers = {maintainer.strip() for maintainer in args.maintainers[0].split(",")}
+print(f"Onboarding packages for {maintainers}")
+
+# Using ruamel.yaml to preserve comments and format
+packit_service_yaml = ruamel.yaml.YAML()
+packit_service_yaml.indent(mapping=2, sequence=4, offset=2)
+
+# Get the current packit-service.yaml.j2 file
+with packit_service_file.open("r") as f:
+    jinja_lines = []
+    for _ in range(SKIP_JINJA_LINES):
+        jinja_lines.append(next(f))
+    packit_service = packit_service_yaml.load(f.read())
+
+# Get all active
+response = requests.get(PAGURE_BZ)
+maintainers_data = response.json()
+maintainers_projects = {
+    pkg_name
+    for pkg_name, pkg_maintainers in maintainers_data["rpms"].items()
+    if any(m in maintainers for m in pkg_maintainers)
+}
+
+# Onboard user's projects
+fedora_ci_projects = copy.copy(packit_service["enabled_projects_for_fedora_ci"])
+previous_count = len(fedora_ci_projects)
+for project in maintainers_projects:
+    dist_git_url = DIST_GIT_FORMAT.format(project)
+    if dist_git_url not in fedora_ci_projects:
+        fedora_ci_projects.append(dist_git_url)
+new_count = len(fedora_ci_projects)
+
+# Update the packit-service.yaml.j2 file
+print(f"Number of projects added: {new_count - previous_count}")
+packit_service["enabled_projects_for_fedora_ci"] = sorted(fedora_ci_projects)
+with packit_service_file.open("w") as f:
+    for line in jinja_lines:
+        f.write(line)
+    packit_service_yaml.dump(packit_service, f)