ultralytics-actions 0.0.22__tar.gz → 0.0.25__tar.gz

{ultralytics_actions-0.0.22/ultralytics_actions.egg-info → ultralytics_actions-0.0.25}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ultralytics-actions
-Version: 0.0.22
+Version: 0.0.25
 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.22 → ultralytics_actions-0.0.25}/actions/__init__.py

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

{ultralytics_actions-0.0.22 → ultralytics_actions-0.0.25}/actions/summarize_pr.py

@@ -71,6 +71,46 @@ 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 main():
     """Summarize a pull request and update its description with an AI-generated summary."""
     pr_number = PR["number"]
@@ -90,6 +130,11 @@ 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"])
+
 
 if __name__ == "__main__":
     main()

{ultralytics_actions-0.0.22 → ultralytics_actions-0.0.25}/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."""
@@ -57,8 +55,12 @@ def graphql_request(query: str, variables: dict = None) -> dict:
 
 def check_pypi_version(pyproject_toml="pyproject.toml"):
     """Compares local and PyPI package versions to determine if a new version should be published."""
+    import re
+
     import tomllib  # requires Python>=3.11
 
+    version_pattern = re.compile(r"^\d+\.\d+\.\d+$")  # e.g. 0.0.0
+
     with open(pyproject_toml, "rb") as f:
         pyproject = tomllib.load(f)
 
@@ -73,6 +75,9 @@ def check_pypi_version(pyproject_toml="pyproject.toml"):
             local_version = next(line.split("=")[1].strip().strip("'\"") for line in f if line.startswith(attr_name))
 
     print(f"Local Version: {local_version}")
+    if not bool(version_pattern.match(local_version)):
+        print("WARNING: Incorrect local version pattern")
+        return "0.0.0", "0.0.0", False
 
     # Get online version from PyPI
     response = requests.get(f"https://pypi.org/pypi/{package_name}/json")

{ultralytics_actions-0.0.22 → ultralytics_actions-0.0.25/ultralytics_actions.egg-info}/PKG-INFO

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