From 665be20505e3ef375373c8f8b0553b45e8dac788 Mon Sep 17 00:00:00 2001
From: Daniil Kalinin <Daniil.Kalinin@jetbrains.com>
Date: Mon, 26 Aug 2024 14:58:09 +0200
Subject: [PATCH] PY-75468 Disable Pycharm Pro suggestion for DataSpell

GitOrigin-RevId: 18afea616bcf511c8952af4ca9a819b4cbe0f824
---
 .../python/suggestions/PycharmProSuggestionProvider.kt      | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/python/src/com/jetbrains/python/suggestions/PycharmProSuggestionProvider.kt b/python/src/com/jetbrains/python/suggestions/PycharmProSuggestionProvider.kt
index cb965ab67c3d..705a44ee3f08 100644
--- a/python/src/com/jetbrains/python/suggestions/PycharmProSuggestionProvider.kt
+++ b/python/src/com/jetbrains/python/suggestions/PycharmProSuggestionProvider.kt
@@ -36,7 +36,7 @@ internal class PycharmProSuggestionProvider : PluginSuggestionProvider {
     if (!vFile.isPythonFile()
         || isDismissed()
         || tryUltimateIsDisabled()
-        || isPyCharmProOrIdeaUltimate()
+        || isUntargetedIDE()
         || FileIndexFacade.getInstance(project).isInLibraryClasses(vFile)) {
       return null
     }
@@ -63,9 +63,9 @@ internal class PycharmProSuggestionProvider : PluginSuggestionProvider {
     return PACKAGES_TO_ADVERTISE.intersect(imports).firstOrNull()
   }
 
-  private fun isPyCharmProOrIdeaUltimate(): Boolean {
+  private fun isUntargetedIDE(): Boolean {
     val productCode = ApplicationInfoImpl.getShadowInstanceImpl().build.productCode
-    return productCode == "PY" || productCode == "IU"
+    return productCode == "PY" || productCode == "IU" || productCode == "DS"
   }
 
   private fun isDismissed(): Boolean = PropertiesComponent.getInstance().isTrueValue(PYCHARM_PRO_SUGGESTION_DISMISSED_KEY)