script for generating branch test data

scripts/gen_branch_testdata.py

@@ -0,0 +1,93 @@
+#!/usr/bin/env /usr/bin/python3
+import subprocess
+import json
+
+try:
+    from unidiff import PatchSet
+except ImportError as e:
+    print(
+        "unidiff is not installed, please install it first with: python3 -m pip install unidiff"
+    )
+    exit(1)
+
+
+try:
+    subprocess.check_output("gh --version", shell=True, text=True)
+except subprocess.CalledProcessError as e:
+    print("gh is not installed, please install it first from https://cli.github.com/")
+    exit(1)
+
+
+def get_last_n_pr_nums(n_prs):
+    list_prs = subprocess.check_output(
+        "gh pr list --state merged | head -n %d | awk '{print $1}'" % n_prs,
+        shell=True,
+        text=True,
+    )
+    return list_prs.splitlines()
+
+
+def process_pr(pr_number):
+    branch_name = subprocess.check_output(
+        "gh pr view %s --json headRefName -q '.headRefName'" % pr_number,
+        shell=True,
+        text=True,
+    ).splitlines()[0]
+    updated_at = subprocess.check_output(
+        "gh pr view %s --json updatedAt -q '.updatedAt'" % pr_number,
+        shell=True,
+        text=True,
+    ).splitlines()[0]
+    title = subprocess.check_output(
+        "gh pr view %s --json title -q '.title'" % pr_number,
+        shell=True,
+        text=True,
+    ).splitlines()[0]
+    diff = subprocess.check_output("gh pr diff %s" % pr_number, shell=True, text=True)
+    patch = PatchSet(diff)
+    files = []
+    for file in patch:
+        hunks = []
+        for hunk in file:
+            hunk_out = {
+                "id": branch_name + ":" + file.path + ":" + str(hunk.target_start),
+                "name": repr(hunk),
+                "diff": str(hunk),
+                "kind": "hunk",
+                "modifiedAt": updated_at,
+                "filePath": file.path,
+            }
+            hunks.append(hunk_out)
+        file_out = {
+            "id": branch_name + ":" + file.path,
+            "path": file.path,
+            "kind": "file",
+            "hunks": hunks,
+        }
+        files.append(file_out)
+    commit = {
+        "id": "Commit:" + branch_name + ":" + pr_number,
+        "description": title,
+        "committedAt": updated_at,
+        "kind": "commit",
+        "files": files,
+    }
+    branch = {
+        "id": branch_name + ":" + pr_number,
+        "name": branch_name,
+        "active": True,
+        "kind": "branch",
+        "commits": [commit],
+    }
+    return branch
+
+
+# prs = get_last_n_pr_nums(4)
+prs = ["426", "427"]  # feel free to paste some some specific PRs
+
+branches = [process_pr(pr) for pr in prs]
+
+print(branches)
+
+with open("scripts/branch_testdata.json", "w") as json_file:
+    json.dump(branches, json_file, indent=4)