ultralytics-actions 0.0.69__py3-none-any.whl → 0.0.70__py3-none-any.whl

actions/__init__.py

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

actions/dispatch_actions.py

@@ -0,0 +1,127 @@
+# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
+
+import time
+from datetime import datetime
+from typing import Dict, List
+
+import requests
+
+from .utils import GITHUB_API_URL, Action, remove_html_comments
+
+# Configuration
+RUN_CI_KEYWORD = "@ultralytics/run-ci"  # and then to merge "@ultralytics/run-ci-and-merge"
+WORKFLOW_FILES = ["format.yml", "ci.yml", "docker.yml"]
+
+
+def get_pr_branch(event) -> str:
+    """Gets the PR branch name."""
+    pr_number = event.event_data["issue"]["number"]
+    pr_data = event.get_repo_data(f"pulls/{pr_number}")
+    return pr_data.get("head", {}).get("ref", "main")
+
+
+def trigger_and_get_workflow_info(event, branch: str) -> List[Dict]:
+    """Triggers workflows and returns their information."""
+    repo = event.repository
+    results = []
+
+    # Trigger all workflows
+    for file in WORKFLOW_FILES:
+        requests.post(
+            f"{GITHUB_API_URL}/repos/{repo}/actions/workflows/{file}/dispatches",
+            json={"ref": branch},
+            headers=event.headers,
+        )
+
+    # Wait for workflows to be created
+    time.sleep(10)
+
+    # Collect information about all workflows
+    for file in WORKFLOW_FILES:
+        # Get workflow name
+        response = requests.get(f"{GITHUB_API_URL}/repos/{repo}/actions/workflows/{file}", headers=event.headers)
+        name = file.replace(".yml", "").title()
+        if response.status_code == 200:
+            name = response.json().get("name", name)
+
+        # Get run information
+        run_url = f"https://github.com/{repo}/actions/workflows/{file}"
+        run_number = None
+
+        runs_response = requests.get(
+            f"{GITHUB_API_URL}/repos/{repo}/actions/workflows/{file}/runs?branch={branch}&event=workflow_dispatch&per_page=1",
+            headers=event.headers,
+        )
+
+        if runs_response.status_code == 200:
+            runs = runs_response.json().get("workflow_runs", [])
+            if runs:
+                run_url = runs[0].get("html_url", run_url)
+                run_number = runs[0].get("run_number")
+
+        results.append({"name": name, "file": file, "url": run_url, "run_number": run_number})
+
+    return results
+
+
+def update_comment(event, comment_body: str, triggered_actions: List[Dict], branch: str) -> bool:
+    """Updates the comment with workflow information."""
+    if not triggered_actions:
+        return False
+
+    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S UTC")
+    summary = (
+        f"\n\n## ⚡ Actions Trigger\n\n"
+        f"<sub>Made with ❤️ by [Ultralytics Actions](https://www.ultralytics.com/actions)<sub>\n\n"
+        f"GitHub Actions below triggered via workflow dispatch on this "
+        f"PR branch `{branch}` at {timestamp} with `@ultralytics/dispatch-actions command`:\n\n"
+    )
+
+    for action in triggered_actions:
+        run_info = f" run {action['run_number']}" if action["run_number"] else ""
+        summary += f"* ✅ [{action['name']}]({action['url']}): `{action['file']}`{run_info}\n"
+
+    new_body = comment_body.replace(RUN_CI_KEYWORD, summary).strip()
+    comment_id = event.event_data["comment"]["id"]
+
+    response = requests.patch(
+        f"{GITHUB_API_URL}/repos/{event.repository}/issues/comments/{comment_id}",
+        json={"body": new_body},
+        headers=event.headers,
+    )
+
+    return response.status_code == 200
+
+
+def main(*args, **kwargs):
+    """Handles triggering workflows from PR comments."""
+    event = Action(*args, **kwargs)
+
+    # Only process new comments on PRs
+    if (
+        event.event_name != "issue_comment"
+        or "pull_request" not in event.event_data.get("issue", {})
+        or event.event_data.get("action") != "created"
+    ):
+        return
+
+    # Get comment info
+    comment_body = remove_html_comments(event.event_data["comment"].get("body", ""))
+    username = event.event_data["comment"]["user"]["login"]
+
+    # Check for trigger keyword and permissions
+    if RUN_CI_KEYWORD not in comment_body or not event.is_org_member(username):
+        return
+
+    # Get branch, trigger workflows, and update comment
+    branch = get_pr_branch(event)
+    print(f"Triggering workflows on branch: {branch}")
+
+    triggered_actions = trigger_and_get_workflow_info(event, branch)
+    success = update_comment(event, comment_body, triggered_actions, branch)
+
+    print(f"Comment update {'succeeded' if success else 'failed'}.")
+
+
+if __name__ == "__main__":
+    main()

actions/utils/common_utils.py

@@ -36,6 +36,15 @@ BAD_HTTP_CODES = frozenset(
         525,  # Cloudfare handshake error
     }
 )
+
+URL_ERROR_LIST = {  # automatically reject these URLs (important: replace spaces with '%20')
+    "https://blog.research.google/search/label/Spam%20and%20Abuse",
+    "https://blog.research.google/search/label/Adversarial%20Attacks",
+    "https://www.microsoft.com/en-us/security/business/ai-machine-learning-security",
+    "https://about.netflix.com/en/news/netflix-recommendations-beyond-the-5-stars-part-1",
+    "https://about.netflix.com/en/news/netflix-research-recommendations",
+}
+
 URL_IGNORE_LIST = {  # use a set (not frozenset) to update with possible private GitHub repos
     "localhost",
     "127.0.0",
@@ -51,10 +60,11 @@ URL_IGNORE_LIST = {  # use a set (not frozenset) to update with possible private
     "mailto:",
     "linkedin.com",
     "twitter.com",
-    "x.com",
+    "https://x.com",  # do not use just 'x' as this will catch other domains like netflix.com
     "storage.googleapis.com",  # private GCS buckets
     "{",  # possible Python fstring
     "(",  # breaks pattern matches
+    ")",
     "api.",  # ignore api endpoints
 }
 REDIRECT_START_IGNORE_LIST = frozenset(
@@ -69,9 +79,14 @@ REDIRECT_START_IGNORE_LIST = frozenset(
         "ultralytics.com/actions",
         "ultralytics.com/bilibili",
         "ultralytics.com/images",
+        "ultralytics.com/license",
+        "ultralytics.com/assets",
         "app.gong.io/call?",
         "docs.openvino.ai",
+        ".git",
+        "/raw/",  # GitHub images
     }
+    | URL_IGNORE_LIST
 )
 REDIRECT_END_IGNORE_LIST = frozenset(
     {
@@ -161,7 +176,7 @@ def is_url(url, session=None, check=True, max_attempts=3, timeout=3, return_url=
         # Check structure
         result = parse.urlparse(url)
         partition = result.netloc.partition(".")  # i.e. netloc = "github.com" -> ("github", ".", "com")
-        if not result.scheme or not partition[0] or not partition[2]:
+        if not result.scheme or not partition[0] or not partition[2] or (url in URL_ERROR_LIST):
             return (False, url) if return_url else False
 
         if check:

actions/utils/github_utils.py

@@ -50,6 +50,13 @@ class Action:
             print(f"Error parsing authenticated user response: {e}")
             return None
 
+    def is_org_member(self, username: str) -> bool:
+        """Checks if a user is a member of the organization using the GitHub API."""
+        org_name = self.repository.split("/")[0]
+        url = f"{GITHUB_API_URL}/orgs/{org_name}/members/{username}"
+        r = requests.get(url, headers=self.headers)
+        return r.status_code == 204  # 204 means the user is a member
+
     def get_pr_diff(self) -> str:
         """Retrieves the diff content for a specified pull request."""
         url = f"{GITHUB_API_URL}/repos/{self.repository}/pulls/{self.pr.get('number')}"

{ultralytics_actions-0.0.69.dist-info → ultralytics_actions-0.0.70.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ultralytics-actions
-Version: 0.0.69
+Version: 0.0.70
 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.70.dist-info/RECORD

@@ -0,0 +1,16 @@
+actions/__init__.py,sha256=jkNfCMkrM58RMv6ijJXC4tASCnP7q1VvKsh1yKtLQn4,742
+actions/dispatch_actions.py,sha256=plfDaFdfBPldqI1uUQPfgSeIDaRTKbkzkXnsYbGkmwI,4415
+actions/first_interaction.py,sha256=1_WvQHCi5RWaSfyi49ClF2Zk_3CKGjFnZqz6FlxPRAc,17868
+actions/summarize_pr.py,sha256=BKttOq-MGaanVaChLU5B1ewKUA8K6S05Cy3FQtyRmxU,11681
+actions/summarize_release.py,sha256=tov6qsYGC68lfobvkwVyoWZBGtJ598G0m097n4Ydzvo,8472
+actions/update_markdown_code_blocks.py,sha256=9PL7YIQfApRNAa0que2hYHv7umGZTZoHlblesB0xFj4,8587
+actions/utils/__init__.py,sha256=ZE0RmC9qOCt9TUhvORd6uVhbxOKVFWJDobR454v55_M,682
+actions/utils/common_utils.py,sha256=BCqB2noJRXxXLAH8rmZ7c4ss1WjFXDC9OPPPFX5OuRU,11758
+actions/utils/github_utils.py,sha256=EZIATz5OSrNgdBrLFxL-9KRxn0i_Ph5-ZXHhhcIrtPE,7495
+actions/utils/openai_utils.py,sha256=txbsEPQnIOieejatBuE6Yk7xR1fQ0erWOEs6cYgUQX4,2943
+ultralytics_actions-0.0.70.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
+ultralytics_actions-0.0.70.dist-info/METADATA,sha256=c9aZUeDQntfNp4gptnUlEVAxDVoEieK1h2qcGQ88jDs,10930
+ultralytics_actions-0.0.70.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
+ultralytics_actions-0.0.70.dist-info/entry_points.txt,sha256=GowvOFplj0C7JmsjbKcbpgLpdf2r921pcaOQkAHWZRA,378
+ultralytics_actions-0.0.70.dist-info/top_level.txt,sha256=5apM5x80QlJcGbACn1v3fkmIuL1-XQCKcItJre7w7Tw,8
+ultralytics_actions-0.0.70.dist-info/RECORD,,

{ultralytics_actions-0.0.69.dist-info → ultralytics_actions-0.0.70.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (78.1.0)
+Generator: setuptools (79.0.1)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

ultralytics_actions-0.0.69.dist-info/RECORD

@@ -1,15 +0,0 @@
-actions/__init__.py,sha256=jfTLYF5qlHNX7JutT_TeZ7oSc5jmqIYB-X87ISRWvzc,742
-actions/first_interaction.py,sha256=1_WvQHCi5RWaSfyi49ClF2Zk_3CKGjFnZqz6FlxPRAc,17868
-actions/summarize_pr.py,sha256=BKttOq-MGaanVaChLU5B1ewKUA8K6S05Cy3FQtyRmxU,11681
-actions/summarize_release.py,sha256=tov6qsYGC68lfobvkwVyoWZBGtJ598G0m097n4Ydzvo,8472
-actions/update_markdown_code_blocks.py,sha256=9PL7YIQfApRNAa0que2hYHv7umGZTZoHlblesB0xFj4,8587
-actions/utils/__init__.py,sha256=ZE0RmC9qOCt9TUhvORd6uVhbxOKVFWJDobR454v55_M,682
-actions/utils/common_utils.py,sha256=YRdEz8qluwzCZfWgqXNmyhKqNhdxNMpoHhGaHUD4AaM,11013
-actions/utils/github_utils.py,sha256=-F--JgxtXE0fSPMFEzakz7iZilp-vonzLiyXfg0b17Y,7117
-actions/utils/openai_utils.py,sha256=txbsEPQnIOieejatBuE6Yk7xR1fQ0erWOEs6cYgUQX4,2943
-ultralytics_actions-0.0.69.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
-ultralytics_actions-0.0.69.dist-info/METADATA,sha256=94Bmnj4tirzyhs7DAqvkEvjP0cPg_uIUXSLqlt1_fBI,10930
-ultralytics_actions-0.0.69.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-ultralytics_actions-0.0.69.dist-info/entry_points.txt,sha256=GowvOFplj0C7JmsjbKcbpgLpdf2r921pcaOQkAHWZRA,378
-ultralytics_actions-0.0.69.dist-info/top_level.txt,sha256=5apM5x80QlJcGbACn1v3fkmIuL1-XQCKcItJre7w7Tw,8
-ultralytics_actions-0.0.69.dist-info/RECORD,,