From 5454f29f4cead4c2a17f769a3fce9a232e3cf2a8 Mon Sep 17 00:00:00 2001
From: jack1142 <6032823+jack1142@users.noreply.github.com>
Date: Sun, 9 Jan 2022 03:58:55 +0100
Subject: [PATCH] Add `CherryPicker.pr_number` attribute

---
 cherry_picker/cherry_picker.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/cherry_picker/cherry_picker.py b/cherry_picker/cherry_picker.py
index 1233cbd..245e299 100755
--- a/cherry_picker/cherry_picker.py
+++ b/cherry_picker/cherry_picker.py
@@ -128,6 +128,10 @@ def __init__(
         self.auto_pr = auto_pr
         self.prefix_commit = prefix_commit
 
+        # This is set to the PR number when cherry-picker successfully
+        # creates a PR through API.
+        self.pr_number = None
+
     def set_paused_state(self):
         """Save paused progress state into Git config."""
         if self.chosen_config_path is not None:
@@ -331,7 +335,9 @@ def create_gh_pr(self, base_branch, head_branch, *, commit_message, gh_auth):
         url = CREATE_PR_URL_TEMPLATE.format(config=self.config)
         response = requests.post(url, headers=request_headers, json=data)
         if response.status_code == requests.codes.created:
-            click.echo(f"Backport PR created at {response.json()['html_url']}")
+            response_data = response.json()
+            click.echo(f"Backport PR created at {response_data['html_url']}")
+            self.pr_number = response_data['number']
         else:
             click.echo(response.status_code)
             click.echo(response.text)