Add repo to issues/pr fetched message

gizmo385 · Dec 27, 2024 · ac951cf · ac951cf
1 parent 4cb7e3c
commit ac951cf
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/lazy_github/lib/messages.py b/lazy_github/lib/messages.py
@@ -2,6 +2,7 @@
 
 from textual.message import Message
 
+from lazy_github.lib.github.repositories import RepositorySortKey
 from lazy_github.models.github import (
     Branch,
     FullPullRequest,
@@ -52,7 +53,8 @@ class IssuesAndPullRequestsFetched(Message):
     then send that message to both sections of the UI.
     """
 
-    def __init__(self, issues_and_pull_requests: list[Issue]) -> None:
+    def __init__(self, repo: Repository, issues_and_pull_requests: list[Issue]) -> None:
+        self.repo = repo
         self.issues_and_pull_requests = issues_and_pull_requests
         super().__init__()
 

diff --git a/lazy_github/ui/screens/primary.py b/lazy_github/ui/screens/primary.py
@@ -212,7 +212,7 @@ async def fetch_issues_and_pull_requests(self, repo: Repository) -> None:
         except GithubApiRequestFailed:
             lg.exception("Error fetching issues and PRs from Github API")
         else:
-            issue_and_pr_message = IssuesAndPullRequestsFetched(issues_and_pull_requests)
+            issue_and_pr_message = IssuesAndPullRequestsFetched(repo, issues_and_pull_requests)
             self.pull_requests.post_message(issue_and_pr_message)
             self.issues.post_message(issue_and_pr_message)