Add conditional based on upload type

codecov · Dec 9, 2024 · 09f1c66 · 09f1c66
1 parent 016eef6
commit 09f1c66
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 21 deletions.
diff --git a/shared/torngit/base.py b/shared/torngit/base.py
@@ -81,6 +81,7 @@ def __init__(
             "repo": {},
             "fallback_installations": None,
             "installation": None,
+            "additional_data": {}
         }
         self.verify_ssl = verify_ssl
         self.data.update(kwargs)

diff --git a/shared/torngit/github.py b/shared/torngit/github.py
@@ -35,7 +35,7 @@
 from shared.torngit.response_types import ProviderPull
 from shared.torngit.status import Status
 from shared.typings.oauth_token_types import OauthConsumerToken
-from shared.typings.torngit import GithubInstallationInfo
+from shared.typings.torngit import GithubInstallationInfo, UploadType
 from shared.utils.urls import url_concat
 
 log = logging.getLogger(__name__)
@@ -611,26 +611,27 @@ async def build_comment_request_body(
         self, body: dict, issueid: int | None = None
     ) -> dict:
         body = dict(body=body)
-        try:
-            ownerid = self.data["owner"].get("ownerid")
-            if (
-                issueid is not None
-                and await INCLUDE_GITHUB_COMMENT_ACTIONS_BY_OWNER.check_value_async(
-                    identifier=ownerid, default=False
-                )
-            ):
-                bot_name = get_config(
-                    "github", "comment_action_bot_name", default="sentry"
-                )
-                body["actions"] = [
-                    {
-                        "name": "Generate Unit Tests",
-                        "type": "copilot-chat",
-                        "prompt": f"@{bot_name} generate tests for this PR.",
-                    }
-                ]
-        except Exception:
-            pass
+        if self.data.get("additional_data", {}).get("upload_type") != UploadType.BUNDLE_ANALYSIS:
+            try:
+                ownerid = self.data["owner"].get("ownerid")
+                if (
+                    issueid is not None
+                    and await INCLUDE_GITHUB_COMMENT_ACTIONS_BY_OWNER.check_value_async(
+                        identifier=ownerid, default=False
+                    )
+                ):
+                    bot_name = get_config(
+                        "github", "comment_action_bot_name", default="sentry"
+                    )
+                    body["actions"] = [
+                        {
+                            "name": "Generate Unit Tests",
+                            "type": "copilot-chat",
+                            "prompt": f"@{bot_name} generate tests for this PR.",
+                        }
+                    ]
+            except Exception:
+                pass
 
         return body
 

diff --git a/shared/typings/torngit.py b/shared/typings/torngit.py
@@ -1,5 +1,10 @@
+from enum import Enum
 from typing import Dict, List, Optional, TypedDict, Union
 
+class UploadType(Enum):
+    COVERAGE = "coverage"
+    TEST_RESULTS = "test_results"
+    BUNDLE_ANALYSIS = "bundle_analysis"
 
 class OwnerInfo(TypedDict):
     service_id: str
@@ -25,9 +30,13 @@ class GithubInstallationInfo(TypedDict):
     app_id: Optional[int]
     pem_path: Optional[str]
 
+class AdditionalData(TypedDict):
+    upload_type: Optional[UploadType]
+
 
 class TorngitInstanceData(TypedDict):
     owner: Union[OwnerInfo, Dict]
     repo: Union[RepoInfo, Dict]
     fallback_installations: List[Optional[GithubInstallationInfo]] | None
     installation: Optional[GithubInstallationInfo]
+    additional_data: Optional[AdditionalData]