add helper

Author: srest2021

src/sentry/integrations/services/repository/impl.py

@@ -128,6 +128,37 @@ def update_repositories(self, *, organization_id: int, updates: list[RpcReposito
 
             Repository.objects.bulk_update(repositories, fields=list(fields_to_update))
 
+    def _cleanup_seer_project_repositories(
+        self,
+        organization_id: int,
+        repo_tuples: list[tuple[int, str, str]],
+    ) -> None:
+        """Delete SeerProjectRepository rows and call Seer API cleanup.
+
+        repo_tuples: list of (repo_id, external_id, provider).
+        """
+        try:
+            organization = Organization.objects.get_from_cache(id=organization_id)
+            if features.has("organizations:seer-project-settings-dual-write", organization):
+                SeerProjectRepository.objects.filter(
+                    repository_id__in=[repo_id for repo_id, _, _ in repo_tuples]
+                ).delete()
+        except Organization.DoesNotExist:
+            pass
+
+        repos_to_clean = [
+            {"repo_external_id": external_id, "repo_provider": repo_provider}
+            for _, external_id, repo_provider in repo_tuples
+            if external_id and repo_provider
+        ]
+        if repos_to_clean:
+            bulk_cleanup_seer_repository_preferences.apply_async(
+                kwargs={
+                    "organization_id": organization_id,
+                    "repos": repos_to_clean,
+                }
+            )
+
     def disable_repositories_for_integration(
         self, *, organization_id: int, integration_id: int, provider: str
     ) -> None:
@@ -143,26 +174,7 @@ def disable_repositories_for_integration(
 
             if repo_ids:
                 Repository.objects.filter(id__in=repo_ids).update(status=ObjectStatus.DISABLED)
-
-                try:
-                    organization = Organization.objects.get_from_cache(id=organization_id)
-                    if features.has("organizations:seer-project-settings-dual-write", organization):
-                        SeerProjectRepository.objects.filter(repository_id__in=repo_ids).delete()
-                except Organization.DoesNotExist:
-                    pass
-
-        repos_to_clean = [
-            {"repo_external_id": external_id, "repo_provider": repo_provider}
-            for _, external_id, repo_provider in repos
-            if external_id and repo_provider
-        ]
-        if repos_to_clean:
-            bulk_cleanup_seer_repository_preferences.apply_async(
-                kwargs={
-                    "organization_id": organization_id,
-                    "repos": repos_to_clean,
-                }
-            )
+                self._cleanup_seer_project_repositories(organization_id, repos)
 
     def disable_repositories_by_external_ids(
         self,
@@ -186,26 +198,7 @@ def disable_repositories_by_external_ids(
 
             if repo_ids:
                 Repository.objects.filter(id__in=repo_ids).update(status=ObjectStatus.DISABLED)
-
-                try:
-                    organization = Organization.objects.get_from_cache(id=organization_id)
-                    if features.has("organizations:seer-project-settings-dual-write", organization):
-                        SeerProjectRepository.objects.filter(repository_id__in=repo_ids).delete()
-                except Organization.DoesNotExist:
-                    pass
-
-        repos_to_clean = [
-            {"repo_external_id": external_id, "repo_provider": repo_provider}
-            for _, external_id, repo_provider in repos
-            if external_id and repo_provider
-        ]
-        if repos_to_clean:
-            bulk_cleanup_seer_repository_preferences.apply_async(
-                kwargs={
-                    "organization_id": organization_id,
-                    "repos": repos_to_clean,
-                }
-            )
+                self._cleanup_seer_project_repositories(organization_id, repos)
 
     def disassociate_organization_integration(
         self,
@@ -223,19 +216,8 @@ def disassociate_organization_integration(
             repo_ids = [repo_id for repo_id, _, _ in repos]
 
             if repo_ids:
-                # Disassociate repos from the organization integration being deleted
                 Repository.objects.filter(id__in=repo_ids).update(integration_id=None)
-
-                # Delete Seer project preferences for the affected repos.
-                # Organization may already be deleted if org deletion and integration
-                # uninstall overlap; skip SeerProjectRepository cleanup in that case
-                # since cascades will handle it.
-                try:
-                    organization = Organization.objects.get_from_cache(id=organization_id)
-                    if features.has("organizations:seer-project-settings-dual-write", organization):
-                        SeerProjectRepository.objects.filter(repository_id__in=repo_ids).delete()
-                except Organization.DoesNotExist:
-                    pass
+                self._cleanup_seer_project_repositories(organization_id, repos)
 
             # Delete Code Owners with a Code Mapping using the OrganizationIntegration
             ProjectCodeOwners.objects.filter(
@@ -248,17 +230,3 @@ def disassociate_organization_integration(
             RepositoryProjectPathConfig.objects.filter(
                 organization_integration_id=organization_integration_id
             ).delete()
-
-        # Delete Seer project preferences for the affected repos via Seer API
-        repos_to_clean = [
-            {"repo_external_id": external_id, "repo_provider": provider}
-            for _, external_id, provider in repos
-            if external_id and provider
-        ]
-        if repos_to_clean:
-            bulk_cleanup_seer_repository_preferences.apply_async(
-                kwargs={
-                    "organization_id": organization_id,
-                    "repos": repos_to_clean,
-                }
-            )