github-overlord 0.6.4__tar.gz → 0.6.5__tar.gz

{github_overlord-0.6.4 → github_overlord-0.6.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: github-overlord
-Version: 0.6.4
+Version: 0.6.5
 Summary: Automatically merge Dependabot PRs in public repositories
 Home-page: https://github.com/iloveitaly/github-overlord
 Keywords: github,dependabot,automation

{github_overlord-0.6.4 → github_overlord-0.6.5}/github_overlord/__init__.py

@@ -10,7 +10,11 @@ from github.GithubObject import NotSet
 from github.Notification import Notification
 from github.PullRequest import PullRequest
 
-from github_overlord.stale_commenter import check_for_stale_comments, is_stale_comment
+from github_overlord.stale_commenter import (
+    check_for_stale_comments,
+    inspect_repo_for_stale_prs,
+    is_stale_comment,
+)
 from github_overlord.utils import log
 
 AUTOMATIC_MERGE_MESSAGE = "Automatically merged with GitHub overlord"
@@ -156,11 +160,14 @@ def cli():
     pass
 
 
-def extract_repo_reference_from_github_url(url: str):
+def extract_repo_reference_from_github_url(url: str | None):
     """
     Extract the owner and repo from a GitHub URL
     """
 
+    if url is None:
+        return None
+
     # convert 'https://github.com/iloveitaly/todoist-digest/pulls' to 'iloveitaly/todoist-digest'
     if "github.com" in url:
         match = re.search(r"github\.com/([^/]+)/([^/]+)", url)
@@ -185,8 +192,7 @@ def dependabot(token, dry_run, repo):
     Automatically merge dependabot PRs in public repos that have passed CI checks
     """
 
-    if repo:
-        repo = extract_repo_reference_from_github_url(repo)
+    repo = extract_repo_reference_from_github_url(repo)
 
     main(token, dry_run, repo)
 
@@ -201,14 +207,25 @@ def dependabot(token, dry_run, repo):
 @click.option("--dry-run", is_flag=True, help="Run script without merging PRs")
 @click.option("--repo", help="Only process a single repository")
 def keep_alive_prs(token, dry_run, repo):
-    repo = extract_repo_reference_from_github_url(repo)
+    assert token, "GitHub token is required"
+    # TODO should assert on openai setup
 
     github = Github(token)
-    login = github.get_user().login
+    user = github.get_user()
+    login = user.login
+
+    repo = extract_repo_reference_from_github_url(repo)
+
+    if repo:
+        inspect_repo_for_stale_prs(dry_run, login, github.get_repo(repo))
+        return
+
+    # TODO this isn't perfect because you may be a contributor :/
+    user.get_repos(type="public") | fp.filter(
+        lambda repo: repo.owner.login != login
+    ) | fp.map(fp.partial(inspect_repo_for_stale_prs, dry_run, login)) | fp.to_list()
 
-    github.get_repo(repo).get_pulls(state="open") | fp.filter(
-        lambda pr: pr.user.login == login
-    ) | fp.map(fp.partial(check_for_stale_comments, dry_run)) | fp.to_list()
+    log.info("stale PR check complete")
 
 
 def notifications(token, dry_run):

{github_overlord-0.6.4 → github_overlord-0.6.5}/github_overlord/stale_commenter.py

@@ -1,12 +1,26 @@
 import json
 
+import funcy_pipe as fp
+from github import Github
 from github.IssueComment import IssueComment
 from github.PullRequest import PullRequest
+from github.Repository import Repository
 from openai import OpenAI
 
 from github_overlord.utils import log
 
 
+def inspect_repo_for_stale_prs(dry_run: bool, login: str, repo: Repository):
+    log.debug("inspecting repo for stale PRs", repo=repo.full_name)
+
+    return (
+        repo.get_pulls(state="open")
+        | fp.filter(lambda pr: pr.user.login == login)
+        | fp.map(fp.partial(check_for_stale_comments, dry_run))
+        | fp.to_list()
+    )
+
+
 def check_for_stale_comments(dry_run: bool, pr: PullRequest):
     """
     Look at PRs which you have written:

{github_overlord-0.6.4 → github_overlord-0.6.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "github-overlord"
-version = "0.6.4"
+version = "0.6.5"
 description = "Automatically merge Dependabot PRs in public repositories"
 authors = ["Michael Bianco <mike@mikebian.co>"]
 readme = "README.md"