From 525c832d2bd940f005070a52a511c628beb9dc69 Mon Sep 17 00:00:00 2001
From: Marc Foley <m.foley.88@gmail.com>
Date: Thu, 28 Mar 2024 13:15:28 +0000
Subject: [PATCH] packager: add labels to prs

---
 Lib/gftools/gfgithub.py |  6 ++++++
 Lib/gftools/packager.py | 15 ++++++++++++++-
 2 files changed, 20 insertions(+), 1 deletion(-)

diff --git a/Lib/gftools/gfgithub.py b/Lib/gftools/gfgithub.py
index 38d36f2e..afeaf179 100644
--- a/Lib/gftools/gfgithub.py
+++ b/Lib/gftools/gfgithub.py
@@ -137,3 +137,9 @@ def pr_files(self, pr_number: int, sleep=4):
             # shouldn't hit any issues.
             time.sleep(sleep)
         return res
+
+    def add_labels(self, issue_number: int, labels: typing.List[str]):
+        return self._post(
+            self.rest_url(f"issues/{issue_number}/labels"),
+            {"labels": labels},
+        )
diff --git a/Lib/gftools/packager.py b/Lib/gftools/packager.py
index 1cae84aa..c6e3493d 100644
--- a/Lib/gftools/packager.py
+++ b/Lib/gftools/packager.py
@@ -23,7 +23,12 @@
 from gftools.scripts.add_font import main as add_font
 from gftools.tags import GFTags
 from gftools.util import google_fonts as fonts
-from gftools.utils import download_file, is_google_fonts_repo, format_html
+from gftools.utils import (
+    download_file,
+    is_google_fonts_repo,
+    format_html,
+    Google_Fonts_has_family,
+)
 
 log = logging.getLogger("gftools.packager")
 LOG_FORMAT = "%(message)s"
@@ -422,6 +427,7 @@ def pr_family(
     family_path: Path,
     title: str,
     body: str,
+    family_name: str,
     base_repo: str = "google",
     head_repo: str = "google",
 ):
@@ -436,6 +442,12 @@ def pr_family(
     else:
         resp = google_fonts.create_issue_comment(open_prs[0]["number"], "Updated")
         log.info(f"Updated PR '{resp['html_url']}'")
+
+    # add labels to pr
+    if Google_Fonts_has_family(family_name):
+        google_fonts.create_issue_comment(open_prs[0]["number"], ["I Font Upgrade"])
+    else:
+        google_fonts.create_issue_comment(open_prs[0]["number"], ["I New Font"])
     return True
 
 
@@ -551,6 +563,7 @@ def make_package(
                 family_path,
                 title,
                 msg,
+                metadata.name,
                 base_repo,
                 head_repo,
             )