ultralytics-actions 0.0.24__tar.gz → 0.0.26__tar.gz

{ultralytics_actions-0.0.24/ultralytics_actions.egg-info → ultralytics_actions-0.0.26}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ultralytics-actions
-Version: 0.0.24
+Version: 0.0.26
 Summary: Ultralytics Actions for GitHub automation and PR management.
 Author-email: Glenn Jocher <glenn.jocher@ultralytics.com>
 Maintainer-email: Ultralytics <hello@ultralytics.com>

{ultralytics_actions-0.0.24 → ultralytics_actions-0.0.26}/actions/__init__.py

@@ -22,4 +22,4 @@
 #     ├── test_summarize_pr.py
 #     └── ...
 
-__version__ = "0.0.24"
+__version__ = "0.0.26"

{ultralytics_actions-0.0.24 → ultralytics_actions-0.0.26}/actions/summarize_pr.py

@@ -13,7 +13,7 @@ from .utils import (
     get_pr_diff,
 )
 
-# Action settings
+# Constants
 SUMMARY_START = (
     "## 🛠️ PR Summary\n\n<sub>Made with ❤️ by [Ultralytics Actions](https://github.com/ultralytics/actions)<sub>\n\n"
 )
@@ -71,12 +71,60 @@ def update_pr_description(repo_name, pr_number, new_summary, max_retries=2):
     return update_response.status_code
 
 
+def label_fixed_issues(pr_number):
+    """Labels issues that are closed by this PR when it's merged."""
+    # GraphQL query to get closing issues
+    query = """
+query($owner: String!, $repo: String!, $pr_number: Int!) {
+    repository(owner: $owner, name: $repo) {
+        pullRequest(number: $pr_number) {
+            closingIssuesReferences(first: 50) {
+                nodes {
+                    number
+                }
+            }
+        }
+    }
+}
+"""
+
+    owner, repo = GITHUB_REPOSITORY.split("/")
+    variables = {"owner": owner, "repo": repo, "pr_number": pr_number}
+    graphql_url = "https://api.github.com/graphql"
+    response = requests.post(graphql_url, json={"query": query, "variables": variables}, headers=GITHUB_HEADERS)
+    if response.status_code != 200:
+        print(f"Failed to fetch linked issues. Status code: {response.status_code}")
+        return
+
+    try:
+        issues = response.json()["data"]["repository"]["pullRequest"]["closingIssuesReferences"]["nodes"]
+        for issue in issues:
+            issue_number = issue["number"]
+            label_url = f"{GITHUB_API_URL}/repos/{GITHUB_REPOSITORY}/issues/{issue_number}/labels"
+            label_response = requests.post(label_url, json={"labels": ["fixed"]}, headers=GITHUB_HEADERS)
+            if label_response.status_code == 200:
+                print(f"Added 'fixed' label to issue #{issue_number}")
+            else:
+                print(f"Failed to add label to issue #{issue_number}. Status: {label_response.status_code}")
+    except KeyError as e:
+        print(f"Error parsing GraphQL response: {e}")
+        return
+
+
+def remove_todos_on_merge(pr_number):
+    """Removes specified labels from PR."""
+    for label in ["TODO"]:  # Can be extended with more labels in the future
+        requests.delete(
+            f"{GITHUB_API_URL}/repos/{GITHUB_REPOSITORY}/issues/{pr_number}/labels/{label}", headers=GITHUB_HEADERS
+        )
+
+
 def main():
     """Summarize a pull request and update its description with an AI-generated summary."""
     pr_number = PR["number"]
 
     print(f"Retrieving diff for PR {pr_number}")
-    diff = get_pr_diff(PR["number"])
+    diff = get_pr_diff(pr_number)
 
     # Generate PR summary
     print("Generating PR summary...")
@@ -90,6 +138,13 @@ def main():
     else:
         print(f"Failed to update PR description. Status code: {status_code}")
 
+    # Update linked issues
+    if PR.get("merged"):
+        print("PR is merged, labeling fixed issues...")
+        label_fixed_issues(pr_number)
+        print("Removing TODO label from PR...")
+        remove_todos_on_merge(pr_number)
+
 
 if __name__ == "__main__":
     main()

{ultralytics_actions-0.0.24 → ultralytics_actions-0.0.26}/actions/utils/github_utils.py

@@ -23,8 +23,6 @@ if GITHUB_EVENT_PATH:
 PR = EVENT_DATA.get("pull_request", {})
 DISCUSSION = EVENT_DATA.get("discussion", {})
 
-INPUTS = {k[6:].lower(): v for k, v in os.environ.items() if k.startswith("INPUT_")}  # actions inputs dictionary
-
 
 def get_pr_diff(pr_number: int) -> str:
     """Retrieves the diff content for a specified pull request in a GitHub repository."""

{ultralytics_actions-0.0.24 → ultralytics_actions-0.0.26/ultralytics_actions.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ultralytics-actions
-Version: 0.0.24
+Version: 0.0.26
 Summary: Ultralytics Actions for GitHub automation and PR management.
 Author-email: Glenn Jocher <glenn.jocher@ultralytics.com>
 Maintainer-email: Ultralytics <hello@ultralytics.com>