googlefonts · m4rc1e · Dec 8, 2021 · Dec 8, 2021
diff --git a/bin/gftools-gen-push-lists.py b/bin/gftools-gen-push-lists.py
@@ -0,0 +1,104 @@
+#!/usr/bin/env python3
+"""
+Generate the to_production.txt and to_sandbox.txt server files in a local
+google/fonts repository.
+
+to_production.txt file tells the engineers which directories need to be pushed
+to the production server. Likewise, the to_sandbox.txt file is for directories
+to be pushed to the sandbox server.
+
+In order for this script to work, the traffic jam must be kept up to date and
+pull requests must use labels.
+
+Usage:
+gftools gen-push-lists /path/to/google/fonts
+"""
+from github import Github
+from collections import defaultdict
+import os
+import sys
+
+
+def pr_directories(pr):
+    results = set()
+    files = pr.get_files()
+    for f in files:
+        results.add(os.path.dirname(f.filename))
+    return results
+
+
+def write_server_file(data):
+    doc = []
+    for title, directories in data.items():
+        directories = sorted(directories)
+        doc.append("# " + f"{title}")
+        doc.append("\n".join(directories))
+        doc.append("")
+    return "\n".join(doc)
+
+
+def main():
+    if len(sys.argv) != 2:
+        print("Usage: gftools gen-push-lists /path/to/google/fonts")
+        sys.exit()
+
+    to_sandbox = defaultdict(set)
+    to_production = defaultdict(set)
+
+    github = Github(os.environ["GH_TOKEN"])
+    repo = github.get_repo("google/fonts")
+
+    projects = repo.get_projects()
+    traffic_jam = next((p for p in projects if p.name == "Traffic Jam"), None)
+    if not traffic_jam:
+        raise ValueError("Traffic Jam column has been deleted or renamed")
+    columns = traffic_jam.get_columns()
+
+    print("Analysing pull requests in traffic jam. This may take a while!")
+    for col in columns:
+        if col.name not in set(
+            ["Just merged / In transit", "In Sandbox list", "In Production list"]
+        ):
+            continue
+        cards = col.get_cards()
+        for card in cards:
+            content = card.get_content()
+            if not hasattr(content, "labels"):
+                print(f"skipping {card}. No labels!")
+                continue
+
+            labels = set(l.name for l in content.labels)
+            pr = content.as_pull_request()
+            directories = pr_directories(pr)
+            if "-- blocked" in labels or "--- Live" in labels:
+                continue
+            if "II Font Upgrade" in labels or "III VF Replacement" in labels:
+                cat = "Upgrade"
+            elif "II New Font" in labels:
+                cat = "New"
+            elif "II Description/Metadata/OFL" in labels:
+                cat = "Metadata/desc/OFL"
+            elif "II Designer profile" in labels:
+                cat = "Designer profile"
+            else:
+                cat = "Small fix/other"
+            if "--- to sandbox" in labels:
+                to_sandbox[cat] |= directories
+            if "--- to production" in labels:
+                to_production[cat] |= directories
+
+    gf_repo_path = sys.argv[1]
+
+    sb_path = os.path.join(gf_repo_path, "to_sandbox.txt")
+    with open(sb_path, "w") as sb_doc:
+        sb_doc.write(write_server_file(to_sandbox))
+
+    prod_path = os.path.join(gf_repo_path, "to_production.txt")
+    with open(prod_path, "w") as prod_doc:
+        prod_doc.write(write_server_file(to_production))
+
+    print("Done!")
+
+
+if __name__ == "__main__":
+    main()